def test__run_periodic_snapshot_tasks__recursive(): with patch("zettarepl.zettarepl.create_snapshot") as create_snapshot: with patch("zettarepl.zettarepl.get_empty_snapshots_for_deletion", Mock(return_value=[])): zettarepl = Zettarepl(Mock(), Mock()) zettarepl._run_periodic_snapshot_tasks( datetime(2018, 9, 1, 15, 11), [ Mock(dataset="data", recursive=False, naming_schema="snap-%Y-%m-%d_%H-%M"), Mock(dataset="data", recursive=True, naming_schema="snap-%Y-%m-%d_%H-%M"), ] ) create_snapshot.assert_called_once_with(ANY, Snapshot("data", "snap-2018-09-01_15-11"), True, ANY)
def test__run_periodic_snapshot_tasks__alphabetical(): with patch("zettarepl.zettarepl.create_snapshot") as create_snapshot: with patch("zettarepl.zettarepl.get_empty_snapshots_for_deletion", Mock(return_value=[])): zettarepl = Zettarepl(Mock(), Mock()) zettarepl._run_periodic_snapshot_tasks( datetime(2018, 9, 1, 15, 11), [ Mock(dataset="data", recursive=False, naming_schema="snap-%Y-%m-%d_%H-%M-2d"), Mock(dataset="data", recursive=False, naming_schema="snap-%Y-%m-%d_%H-%M-1w"), ] ) assert create_snapshot.call_count == 2 create_snapshot.assert_has_calls([ call(ANY, Snapshot("data", "snap-2018-09-01_15-11-1w"), False, ANY), call(ANY, Snapshot("data", "snap-2018-09-01_15-11-2d"), False, ANY), ])