def test_return_projects(self):
fake_driver = MagicMock()
fake_driver.name = factory.make_name("pod")
projects = [
{
"name": "p1",
"description": "Project 1"
},
{
"name": "p2",
"description": "Project 2"
},
]
fake_driver.discover_projects.return_value = succeed(projects)
self.patch(PodDriverRegistry, "get_item").return_value = fake_driver
result = yield pods.discover_pod_projects(fake_driver.name, {})
self.assertEqual(
result,
{
"projects": [
DiscoveredPodProject(name="p1", description="Project 1"),
DiscoveredPodProject(name="p2", description="Project 2"),
]
},
)
def test_converts_exceptions(self):
fake_driver = MagicMock()
fake_driver.name = factory.make_name("pod")
fake_driver.discover_projects.return_value = fail(Exception("fail!"))
self.patch(PodDriverRegistry, "get_item").return_value = fake_driver
with ExpectedException(exceptions.PodActionFail):
yield pods.discover_pod_projects(fake_driver.name, {})
def test_unknown_pod_raises_UnknownPodType(self):
unknown_type = factory.make_name("unknown")
with ExpectedException(exceptions.UnknownPodType):
yield pods.discover_pod_projects(unknown_type, {})