def byOptions(cls, **options):
criteria = cls.criteriaFromOptions(**options)
return [
opencue.wrappers.host.Host(host) for host in Cuebot.getStub(
'host').GetHosts(host_pb2.HostGetHostsRequest(r=criteria),
timeout=Cuebot.Timeout).hosts.hosts
]
def getHost(uniq):
"""Returns a Host object from a unique identifier
@type uniq: a unique identifier.
@param uniq: an id
@rtype: Host
@return: A Host object"""
return Host(
Cuebot.getStub('host').GetHost(host_pb2.HostGetHostsRequest(id=uniq),
timeout=Cuebot.Timeout).host)
def testBaseSearchHost(self, getStubMock):
stubMock = mock.Mock()
getStubMock.return_value = stubMock
hostSearch = opencue.search.HostSearch(substr=['unittest_host'])
hostSearch.search()
stubMock.GetHosts.assert_called_with(host_pb2.HostGetHostsRequest(
r=host_pb2.HostSearchCriteria(substr=['unittest_host'])),
timeout=mock.ANY)
def testGetHosts(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetHosts.return_value = host_pb2.HostGetHostsResponse(
hosts=host_pb2.HostSeq(hosts=[host_pb2.Host(name=TEST_HOST_NAME)]))
getStubMock.return_value = stubMock
hosts = opencue.api.getHosts(name=[TEST_HOST_NAME])
stubMock.GetHosts.assert_called_with(
host_pb2.HostGetHostsRequest(r=host_pb2.HostSearchCriteria(hosts=[TEST_HOST_NAME])),
timeout=mock.ANY)
self.assertEqual(1, len(hosts))
self.assertEqual(TEST_HOST_NAME, hosts[0].name())
def byOptions(cls, **options):
criteria = cls.criteriaFromOptions(**options)
return Cuebot.getStub('host').GetHosts(
host_pb2.HostGetHostsRequest(r=criteria), timeout=Cuebot.Timeout)