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)