def testDelete(self, getStubMock): stubMock = mock.Mock() stubMock.Delete.return_value = host_pb2.DeedDeleteResponse() getStubMock.return_value = stubMock deed = opencue.wrappers.deed.Deed(host_pb2.Deed(id=TEST_DEED_ID)) deed.delete() stubMock.Delete.assert_called_with( host_pb2.DeedDeleteRequest(deed=deed.data), timeout=mock.ANY)
def testGetOwner(self, getStubMock): stubMock = mock.Mock() stubMock.GetOwner.return_value = host_pb2.DeedGetOwnerResponse( owner=host_pb2.Owner(name=TEST_DEED_OWNER)) getStubMock.return_value = stubMock deed = opencue.wrappers.deed.Deed(host_pb2.Deed(id=TEST_DEED_ID)) owner = deed.getOwner() stubMock.GetOwner.assert_called_with( host_pb2.DeedGetOwnerRequest(deed=deed.data), timeout=mock.ANY) self.assertEqual(owner.name(), TEST_DEED_OWNER)
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 testSetBlackoutTimeEnabled(self, getStubMock): stubMock = mock.Mock() stubMock.SetBlackoutTimeEnabled.return_value = host_pb2.DeedSetBlackoutTimeEnabledResponse() getStubMock.return_value = stubMock testBlackoutEnabled = True deed = opencue.wrappers.deed.Deed(host_pb2.Deed(id=TEST_DEED_ID)) deed.setBlackoutTimeEnabled(testBlackoutEnabled) stubMock.SetBlackoutTimeEnabled.assert_called_with( host_pb2.DeedSetBlackoutTimeEnabledRequest(deed=deed.data, enabled=testBlackoutEnabled), timeout=mock.ANY)
def testGetDeeds(self, getStubMock): stubMock = mock.Mock() stubMock.GetDeeds.return_value = host_pb2.OwnerGetDeedsResponse( deeds=host_pb2.DeedSeq(deeds=[host_pb2.Deed(id=TEST_DEED_ID)])) getStubMock.return_value = stubMock owner = opencue.wrappers.owner.Owner( host_pb2.Owner(id=TEST_OWNER_ID, name=TEST_OWNER_NAME)) deeds = owner.getDeeds() stubMock.GetDeeds.assert_called_with( host_pb2.OwnerGetDeedsRequest(owner=owner.data), timeout=mock.ANY) self.assertEqual(len(deeds), 1) self.assertEqual(deeds[0].id(), TEST_DEED_ID)
def testSetBlackoutTime(self, getStubMock): stubMock = mock.Mock() stubMock.SetBlackoutTime.return_value = host_pb2.DeedSetBlackoutTimeResponse() getStubMock.return_value = stubMock testStartTime = 100 testStopTime = 200 deed = opencue.wrappers.deed.Deed(host_pb2.Deed(id=TEST_DEED_ID)) deed.setBlackoutTime(testStartTime, testStopTime) stubMock.SetBlackoutTime.assert_called_with( host_pb2.DeedSetBlackoutTimeRequest(deed=deed.data, start_time=testStartTime, stop_time=testStopTime), timeout=mock.ANY)