Add unit tests for Project.get_services()

Signed-off-by: Chris Corbyn <chris@w3style.co.uk>
This commit is contained in:
d11wtq 2014-06-07 13:29:28 +00:00 committed by Chris Corbyn
parent dfc74e2a77
commit 22c531dea7
1 changed files with 26 additions and 0 deletions

View File

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