From 22c531dea7cf0e9e7de686907033bd1d07c3caef Mon Sep 17 00:00:00 2001 From: d11wtq Date: Sat, 7 Jun 2014 13:29:28 +0000 Subject: [PATCH] Add unit tests for Project.get_services() Signed-off-by: Chris Corbyn --- tests/unit/project_test.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/tests/unit/project_test.py b/tests/unit/project_test.py index f5bacc654..7c0ee4cd9 100644 --- a/tests/unit/project_test.py +++ b/tests/unit/project_test.py @@ -67,3 +67,29 @@ class ProjectTest(unittest.TestCase): ) project = Project('test', [web], None) self.assertEqual(project.get_service('web'), web) + + def test_get_services_returns_all_auto_started_without_args(self): + web = Service( + project='figtest', + name='web', + ) + console = Service( + project='figtest', + name='console', + auto_start=False + ) + project = Project('test', [web, console], None) + self.assertEqual(project.get_services(), [web]) + + def test_get_services_returns_listed_services_with_args(self): + web = Service( + project='figtest', + name='web', + ) + console = Service( + project='figtest', + name='console', + auto_start=False + ) + project = Project('test', [web, console], None) + self.assertEqual(project.get_services(['console']), [console])