def reparentHostIds(self, hostIds):
"""Moves the given hosts to the allocation
@type hostIds: list<str>
@param hostIds: The host ids to move to this allocation
"""
hosts = [opencue.wrappers.host.Host(host_pb2.Host(id=hostId)) for hostId in hostIds]
self.reparentHosts(hosts)
def testDelete(self, getStubMock):
stubMock = mock.Mock()
stubMock.Delete.return_value = host_pb2.HostDeleteResponse()
getStubMock.return_value = stubMock
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.delete()
stubMock.Delete.assert_called_with(
host_pb2.HostDeleteRequest(host=host.data), timeout=mock.ANY)
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())
def testSetOs(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetOs.return_value = host_pb2.HostSetOsResponse()
getStubMock.return_value = stubMock
osName = 'linux'
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.setOs(osName)
stubMock.SetOs.assert_called_with(host_pb2.HostSetOsRequest(
host=host.data, os=osName),
timeout=mock.ANY)
def testGetHost(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetHost.return_value = host_pb2.ProcGetHostResponse(
host=host_pb2.Host(name=TEST_HOST_NAME))
getStubMock.return_value = stubMock
proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME))
host = proc.getHost()
stubMock.GetHost.assert_called_with(
host_pb2.ProcGetHostRequest(proc=proc.data), timeout=mock.ANY)
self.assertEqual(host.name(), TEST_HOST_NAME)
def testRemoveTags(self, getStubMock):
stubMock = mock.Mock()
stubMock.RemoveTags.return_value = host_pb2.HostRemoveTagsResponse()
getStubMock.return_value = stubMock
tags = ['tags', 'are', 'fun']
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.removeTags(tags)
stubMock.RemoveTags.assert_called_with(host_pb2.HostRemoveTagsRequest(
host=host.data, tags=tags),
timeout=mock.ANY)
def testGetHost(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetHost.return_value = host_pb2.DeedGetHostResponse(
host=host_pb2.Host(id=TEST_HOST_ID))
getStubMock.return_value = stubMock
deed = opencue.wrappers.deed.Deed(host_pb2.Deed(id=TEST_DEED_ID))
host = deed.getHost()
stubMock.GetHost.assert_called_with(
host_pb2.DeedGetHostRequest(deed=deed.data), timeout=mock.ANY)
self.assertEqual(host.id(), TEST_HOST_ID)
def testGetHost(self, getStubMock):
arbitraryId = '00000000-0000-0000-0000-012345678980'
stubMock = mock.Mock()
stubMock.GetHost.return_value = host_pb2.HostGetHostResponse(
host=host_pb2.Host(id=arbitraryId))
getStubMock.return_value = stubMock
host = opencue.api.getHost(arbitraryId)
stubMock.GetHost.assert_called_with(
host_pb2.HostGetHostRequest(id=arbitraryId), timeout=mock.ANY)
self.assertEqual(arbitraryId, host.id())
def testSetHardwareState(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetHardwareState.return_value = host_pb2.HostSetHardwareStateResponse(
)
getStubMock.return_value = stubMock
state = host_pb2.REBOOTING
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.setHardwareState(state)
stubMock.SetHardwareState.assert_called_with(
host_pb2.HostSetHardwareStateRequest(host=host.data, state=state),
timeout=mock.ANY)
def testRenameTag(self, getStubMock):
stubMock = mock.Mock()
stubMock.RenameTag.return_value = host_pb2.HostRenameTagResponse()
getStubMock.return_value = stubMock
oldTag = 'sad'
newTag = 'happy'
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.renameTag(oldTag, newTag)
stubMock.RenameTag.assert_called_with(host_pb2.HostRenameTagRequest(
host=host.data, old_tag=oldTag, new_tag=newTag),
timeout=mock.ANY)
def testSetThreadMode(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetThreadMode.return_value = host_pb2.HostSetThreadModeResponse(
)
getStubMock.return_value = stubMock
mode = host_pb2.VARIABLE
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.setThreadMode(mode)
stubMock.SetThreadMode.assert_called_with(
host_pb2.HostSetThreadModeRequest(host=host.data, mode=mode),
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 testSetAllocation(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetAllocation.return_value = host_pb2.HostSetAllocationResponse()
getStubMock.return_value = stubMock
allocId = 'aaa-aaaa-aaa'
alloc = opencue.wrappers.allocation.Allocation(facility_pb2.Allocation(id=allocId))
host = opencue.wrappers.host.Host(
host_pb2.Host(name=TEST_HOST_NAME))
host.setAllocation(alloc)
stubMock.SetAllocation.assert_called_with(
host_pb2.HostSetAllocationRequest(host=host.data, allocation_id=alloc.id()),
timeout=mock.ANY)
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 = 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 testGetProcs(self, getStubMock):
procName = 'testProc'
stubMock = mock.Mock()
stubMock.GetProcs.return_value = host_pb2.HostGetProcsResponse(
procs=host_pb2.ProcSeq(procs=[host_pb2.Proc(name=procName)]))
getStubMock.return_value = stubMock
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
procs = host.getProcs()
stubMock.GetProcs.assert_called_with(
host_pb2.HostGetProcsRequest(host=host.data), timeout=mock.ANY)
self.assertEqual(len(procs), 1)
self.assertEqual(procs[0].name(), procName)
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)
def testGetComments(self, getStubMock):
message = 'this is a test.'
stubMock = mock.Mock()
stubMock.GetComments.return_value = host_pb2.HostGetCommentsResponse(
comments=comment_pb2.CommentSeq(
comments=[comment_pb2.Comment(message=message)]))
getStubMock.return_value = stubMock
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
comments = host.getComments()
stubMock.GetComments.assert_called_with(
host_pb2.HostGetCommentsRequest(host=host.data), timeout=mock.ANY)
self.assertEqual(len(comments), 1)
self.assertEqual(comments[0].message(), message)
def testAddComment(self, getStubMock):
stubMock = mock.Mock()
stubMock.AddComment.return_value = host_pb2.HostAddCommentResponse()
getStubMock.return_value = stubMock
subject = 'test'
message = 'this is a test.'
comment = comment_pb2.Comment(user=os.getenv("USER", "unknown"),
subject=subject,
message=message,
timestamp=0)
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.addComment(subject, message)
stubMock.AddComment.assert_called_with(host_pb2.HostAddCommentRequest(
host=host.data, new_comment=comment),
timeout=mock.ANY)
def testGetRenderPartitions(self, getStubMock):
renderPartId = 'rpr-rprp-rpr'
stubMock = mock.Mock()
stubMock.GetRenderPartitions.return_value = host_pb2.HostGetRenderPartitionsResponse(
render_partitions=renderPartition_pb2.RenderPartitionSeq(
render_partitions=[renderPartition_pb2.RenderPartition(id=renderPartId)]))
getStubMock.return_value = stubMock
host = opencue.wrappers.host.Host(
host_pb2.Host(name=TEST_HOST_NAME))
renderParts = host.getRenderPartitions()
stubMock.GetRenderPartitions.assert_called_with(
host_pb2.HostGetRenderPartitionsRequest(host=host.data),
timeout=mock.ANY)
self.assertEqual(len(renderParts), 1)
self.assertEqual(renderParts[0].id, renderPartId)
