def test_with_data(self):
ds_1 = object()
ds_2 = object()
ds_3 = object()
provider = MockDataSourceProvider({
'domain1': [ds_1, ds_2],
'domain2': [ds_3],
})
self.assertEqual({ds_1, ds_2, ds_3},
set(provider.get_all_data_sources()))
self.assertEqual([ds_1, ds_2], provider.by_domain('domain1'))
self.assertEqual([ds_3], provider.by_domain('domain2'))
self.assertEqual([], provider.by_domain('domain3'))
def test_empty(self):
provider = MockDataSourceProvider()
self.assertEqual([], provider.get_all_data_sources())
self.assertEqual([], provider.by_domain('foo'))
self.assertEqual([],
provider.get_data_sources_modified_since(
datetime.datetime.utcnow()))