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)