def test_should_generate_prefixes_for_one_day(): assert _s3_key_prefixes(prefix="some-prefix/", account_ids=["000"], regions=["some-region-1"], from_date=datetime.datetime(2017, 1, 1, tzinfo=pytz.utc), to_date=datetime.datetime(2017, 1, 1, tzinfo=pytz.utc)) == \ ["some-prefix/AWSLogs/000/CloudTrail/some-region-1/2017/01/01"]
def test_should_generate_prefixes_one_organisation(): assert _s3_key_prefixes(prefix="some-prefix/", org_ids=["o-s0mething"], account_ids=["000"], regions=["some-region-1"], from_date=datetime.datetime(2017, 1, 1, tzinfo=pytz.utc), to_date=datetime.datetime(2017, 1, 1, tzinfo=pytz.utc)) == \ ["some-prefix/AWSLogs/o-s0mething/000/CloudTrail/some-region-1/2017/01/01"]