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)