def testDelete(self, getStubMock):
stubMock = mock.Mock()
stubMock.Delete.return_value = host_pb2.OwnerDeleteResponse()
getStubMock.return_value = stubMock
owner = opencue.wrappers.owner.Owner(
host_pb2.Owner(id=TEST_OWNER_ID, name=TEST_OWNER_NAME))
owner.delete()
stubMock.Delete.assert_called_with(
host_pb2.OwnerDeleteRequest(owner=owner.data), timeout=mock.ANY)
def testGetOwner(self, getStubMock):
ownerName = 'arbitrary-name'
stubMock = mock.Mock()
stubMock.GetOwner.return_value = host_pb2.OwnerGetOwnerResponse(
owner=host_pb2.Owner(name=ownerName))
getStubMock.return_value = stubMock
owner = opencue.api.getOwner(ownerName)
stubMock.GetOwner.assert_called_with(
host_pb2.OwnerGetOwnerRequest(name=ownerName), timeout=mock.ANY)
self.assertEqual(ownerName, owner.name())
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 testSetShow(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetShow.return_value = host_pb2.OwnerSetShowResponse()
getStubMock.return_value = stubMock
owner = opencue.wrappers.owner.Owner(
host_pb2.Owner(id=TEST_OWNER_ID, name=TEST_OWNER_NAME))
owner.setShow(TEST_SHOW_NAME)
stubMock.SetShow.assert_called_with(
host_pb2.OwnerSetShowRequest(owner=owner.data, show=TEST_SHOW_NAME),
timeout=mock.ANY)
def testTakeOwnership(self, getStubMock):
stubMock = mock.Mock()
stubMock.TakeOwnership.return_value = host_pb2.OwnerTakeOwnershipResponse()
getStubMock.return_value = stubMock
owner = opencue.wrappers.owner.Owner(
host_pb2.Owner(id=TEST_OWNER_ID, name=TEST_OWNER_NAME))
owner.takeOwnership(TEST_HOST_ID)
stubMock.TakeOwnership.assert_called_with(
host_pb2.OwnerTakeOwnershipRequest(owner=owner.data, host=TEST_HOST_ID),
timeout=mock.ANY)
def testGetOwner(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetOwner.return_value = host_pb2.OwnerGetOwnerResponse(
owner=host_pb2.Owner(id=TEST_OWNER_ID, name=TEST_OWNER_NAME))
getStubMock.return_value = stubMock
owner = opencue.wrappers.owner.Owner()
response = owner.getOwner(TEST_OWNER_NAME)
stubMock.GetOwner.assert_called_with(
host_pb2.OwnerGetOwnerRequest(name=TEST_OWNER_NAME), timeout=mock.ANY)
self.assertEqual(response.id(), TEST_OWNER_ID)
self.assertEqual(response.name(), TEST_OWNER_NAME)
def testCreateOwner(self, getStubMock):
stubMock = mock.Mock()
stubMock.CreateOwner.return_value = show_pb2.ShowCreateOwnerResponse(
owner=host_pb2.Owner(name=TEST_OWNER_NAME))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
owner = show.createOwner(TEST_OWNER_NAME)
stubMock.CreateOwner.assert_called_with(
show_pb2.ShowCreateOwnerRequest(show=show.data, name=TEST_OWNER_NAME),
timeout=mock.ANY)
self.assertEqual(owner.name, TEST_OWNER_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)
