Add test for ServiceCollection.get

This commit is contained in:
Ben Firshman 2013-12-09 18:43:10 +00:00
parent 2199b62783
commit f0768d4dca

View File

@ -16,12 +16,10 @@ class ServiceCollectionTest(ServiceTestCase):
} }
]) ])
self.assertEqual(len(collection), 2) self.assertEqual(len(collection), 2)
web = [s for s in collection if s.name == 'web'][0] self.assertEqual(collection.get('web').name, 'web')
self.assertEqual(web.name, 'web') self.assertEqual(collection.get('web').image, 'ubuntu')
self.assertEqual(web.image, 'ubuntu') self.assertEqual(collection.get('db').name, 'db')
db = [s for s in collection if s.name == 'db'][0] self.assertEqual(collection.get('db').image, 'ubuntu')
self.assertEqual(db.name, 'db')
self.assertEqual(db.image, 'ubuntu')
def test_from_dict_sorts_in_dependency_order(self): def test_from_dict_sorts_in_dependency_order(self):
collection = ServiceCollection.from_dicts(None, [ collection = ServiceCollection.from_dicts(None, [
@ -39,6 +37,11 @@ class ServiceCollectionTest(ServiceTestCase):
self.assertEqual(collection[0].name, 'db') self.assertEqual(collection[0].name, 'db')
self.assertEqual(collection[1].name, 'web') self.assertEqual(collection[1].name, 'web')
def test_get(self):
web = self.create_service('web')
collection = ServiceCollection([web])
self.assertEqual(collection.get('web'), web)
def test_start_stop(self): def test_start_stop(self):
collection = ServiceCollection([ collection = ServiceCollection([
self.create_service('web'), self.create_service('web'),