def test__get_snapshot_to_send__restrict_schedule():
assert get_snapshots_to_send(
["2018-09-02_17-45", "2018-09-02_17-46", "2018-09-02_17-47"],
["2018-09-02_17-45"],
Mock(periodic_snapshot_tasks=[Mock(naming_schema="%Y-%m-%d_%H-%M")],
also_include_naming_schema=[],
restrict_schedule=CronSchedule("*/2", "*", "*", "*", "*", time(0, 0), time(23, 59)),
only_matching_schedule=False,
retention_policy=Mock(calculate_delete_snapshots=Mock(return_value=[]))),
) == ("2018-09-02_17-45", ["2018-09-02_17-46"])
def test__get_snapshot_to_send__multiple_tasks():
assert get_snapshots_to_send(
["1w-2018-09-02_00-00", "2d-2018-09-02_00-00", "2d-2018-09-02_12-00",
"1w-2018-09-03_00-00", "2d-2018-09-03_12-00"],
["1w-2018-09-02_00-00", "2d-2018-09-02_00-00"],
Mock(periodic_snapshot_tasks=[Mock(naming_schema="1w-%Y-%m-%d_%H-%M"),
Mock(naming_schema="2d-%Y-%m-%d_%H-%M")],
also_include_naming_schema=[],
restrict_schedule=None,
only_matching_schedule=False,
retention_policy=Mock(calculate_delete_snapshots=Mock(return_value=[]))),
) == ("2d-2018-09-02_00-00", ["2d-2018-09-02_12-00", "1w-2018-09-03_00-00", "2d-2018-09-03_12-00"])
def test__get_snapshot_to_send__multiple_tasks_retention_policy():
retention_policy = lambda now, src_snapshots, dst_snapshots: [dst_snapshots[1]] # 1w-2018-09-03_00-00
assert get_snapshots_to_send(
["1w-2018-09-02_00-00", "2d-2018-09-02_00-00", "2d-2018-09-02_12-00",
"1w-2018-09-03_00-00", "2d-2018-09-03_12-00"],
["1w-2018-09-02_00-00", "2d-2018-09-02_00-00"],
Mock(periodic_snapshot_tasks=[Mock(naming_schema="1w-%Y-%m-%d_%H-%M"),
Mock(naming_schema="2d-%Y-%m-%d_%H-%M")],
also_include_naming_schema=[],
restrict_schedule=None,
only_matching_schedule=False,
retention_policy=Mock(calculate_delete_snapshots=Mock(side_effect=retention_policy))),
) == ("2d-2018-09-02_00-00", ["2d-2018-09-02_12-00", "2d-2018-09-03_12-00"])
