def createOwner(self, user):
"""Creates a new owner
@type user: str
@param user: user name
@rtype: Owner
@return: The created owner object
"""
response = self.stub.CreateOwner(show_pb2.ShowCreateOwnerRequest(show=self.data, name=user),
timeout=Cuebot.Timeout)
return response.owner
def createOwner(self, user):
"""Creates a new owner for the show.
:type user: str
:param user: user name
:rtype: host_pb2.Owner
:return: the created owner object
"""
response = self.stub.CreateOwner(show_pb2.ShowCreateOwnerRequest(show=self.data, name=user),
timeout=Cuebot.Timeout)
return response.owner
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)
