Răsfoiți Sursa

Add unit tests for Project.get_services()

Signed-off-by: Chris Corbyn <[email protected]>
d11wtq 11 ani în urmă
părinte
comite
22c531dea7
1 a modificat fișierele cu 26 adăugiri și 0 ștergeri
  1. 26 0
      tests/unit/project_test.py

+ 26 - 0
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])