def test_get_resources_not_found(self):
for locator in [Locator(vm=VmLocator("1234567")),
Locator(disk=DiskLocator("891"))]:
request = GetResourcesRequest([locator])
response = rpc_call(self.host_client.get_resources, request)
assert_that(response.result, is_(GetResourcesResultCode.OK))
assert_that(response.resources, is_([]))
def get_disk(self, disk_id, expect_found=True):
request = GetResourcesRequest([Locator(disk=DiskLocator(disk_id))])
response = rpc_call(self.host_client.get_resources, request)
assert_that(response.result, equal_to(GetResourcesResultCode.OK))
if expect_found:
assert_that(len(response.resources), is_(1))
return response.resources[0].disks[0]
else:
assert_that(len(response.resources), is_(0))
return None
def test_find_response(self):
response = FindResponse(FindResultCode.OK)
scheduler = MagicMock()
scheduler.find.return_value = response
handler = SchedulerHandler()
handler._schedulers["foo"] = scheduler
actual_response = handler.find(FindRequest(Locator(), "foo"))
assert_that(actual_response, is_(same_instance(response)))
def find_disk(self, disk_id, expect=FindResultCode.OK):
request = FindRequest(Locator(disk=DiskLocator(disk_id)))
response = rpc_call(self.host_client.find, request)
assert_that(response.result, equal_to(expect))
return response.datastore
def find(self, expect=FindResultCode.OK):
request = FindRequest(Locator(VmLocator(self.id)))
response = rpc_call(self.host_client.find, request)
assert_that(response.result, equal_to(expect))
return response
def test_find_missing_scheduler(self):
handler = SchedulerHandler()
response = handler.find(FindRequest(Locator()))
assert_that(response.result, is_(FindResultCode.INVALID_SCHEDULER))