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])