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)
