def reparentHosts(self, hosts):
"""Moves the given hosts to the allocation
@type hosts: list<HostInterfacePrx or Host or id or str hostname>
@param hosts: The hosts to move to this allocation
"""
hostSeq = host_pb2.HostSeq()
hostSeq.hosts.extend(hosts)
self.stub.ReparentHosts(facility_pb2.AllocReparentHostsRequest(
allocation=self.data, hosts=hostSeq),
timeout=Cuebot.Timeout)
def reparentHosts(self, hosts):
"""Moves the given hosts to the allocation
@type hosts: list<opencue.wrappers.host.Host>
@param hosts: The hosts to move to this allocation
"""
hostSeq = host_pb2.HostSeq()
hostSeq.hosts.extend([host.data for host in hosts])
self.stub.ReparentHosts(
facility_pb2.AllocReparentHostsRequest(allocation=self.data, hosts=hostSeq),
timeout=Cuebot.Timeout
)
def reparentHosts(self, hosts):
"""Moves the given hosts into the allocation.
:type hosts: list<opencue.wrappers.host.Host>
:param hosts: the hosts to move to this allocation
"""
hostSeq = host_pb2.HostSeq()
# pylint: disable=no-member
hostSeq.hosts.extend([host.data for host in hosts])
# pylint: enable=no-member
self.stub.ReparentHosts(facility_pb2.AllocReparentHostsRequest(
allocation=self.data, hosts=hostSeq),
timeout=Cuebot.Timeout)
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 testGetHosts(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetHosts.return_value = host_pb2.OwnerGetHostsResponse(
hosts=host_pb2.HostSeq(hosts=[host_pb2.Host(id=TEST_HOST_ID)]))
getStubMock.return_value = stubMock
owner = opencue.wrappers.owner.Owner(
host_pb2.Owner(id=TEST_OWNER_ID, name=TEST_OWNER_NAME))
hosts = owner.getHosts()
stubMock.GetHosts.assert_called_with(
host_pb2.OwnerGetHostsRequest(owner=owner.data), timeout=mock.ANY)
self.assertEqual(len(hosts), 1)
self.assertEqual(hosts[0].id(), TEST_HOST_ID)
def testReparentHostIds(self, getStubMock):
stubMock = mock.Mock()
stubMock.ReparentHosts.return_value = facility_pb2.AllocReparentHostsResponse()
getStubMock.return_value = stubMock
alloc = opencue.wrappers.allocation.Allocation(
facility_pb2.Allocation(name=TEST_ALLOC_NAME))
hostIds = [TEST_HOST_ID]
alloc.reparentHostIds(hostIds)
hosts = [host_pb2.Host(id=TEST_HOST_ID)]
stubMock.ReparentHosts.assert_called_with(
facility_pb2.AllocReparentHostsRequest(
allocation=alloc.data, hosts=host_pb2.HostSeq(hosts=hosts)), timeout=mock.ANY)
def testGetHosts(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetHosts.return_value = facility_pb2.AllocGetHostsResponse(
hosts=host_pb2.HostSeq(hosts=[host_pb2.Host(name=TEST_HOST_NAME)]))
getStubMock.return_value = stubMock
alloc = opencue.wrappers.allocation.Allocation(
facility_pb2.Allocation(name=TEST_ALLOC_NAME))
hosts = alloc.getHosts()
stubMock.GetHosts.assert_called_with(
facility_pb2.AllocGetHostsRequest(allocation=alloc.data),
timeout=mock.ANY)
self.assertEqual(len(hosts), 1)
self.assertEqual(hosts[0].name(), TEST_HOST_NAME)
