def findHost(name):
"""Returns the host for the matching hostname.
:type name: str
:param name: the unique name of a host
:rtype: Host
:return: The matching host object"""
return Host(Cuebot.getStub('host').FindHost(
host_pb2.HostFindHostRequest(name=name), timeout=Cuebot.Timeout).host)
def testFindHost(self, getStubMock):
stubMock = mock.Mock()
stubMock.FindHost.return_value = host_pb2.HostFindHostResponse(
host=host_pb2.Host(name=TEST_HOST_NAME))
getStubMock.return_value = stubMock
host = opencue.api.findHost(TEST_HOST_NAME)
stubMock.FindHost.assert_called_with(
host_pb2.HostFindHostRequest(name=TEST_HOST_NAME), timeout=mock.ANY)
self.assertEqual(TEST_HOST_NAME, host.name())
