def findSubscription(name):
"""Returns the subscription object that matches the name.
:type name: str
:param name: the name of the subscription
:rtype: Subscription
:return: a Subscription object"""
return Subscription(Cuebot.getStub('subscription').Find(
subscription_pb2.SubscriptionFindRequest(name=name), timeout=Cuebot.Timeout).subscription)
def testFindSubscription(self, getStubMock):
stubMock = mock.Mock()
stubMock.Find.return_value = subscription_pb2.SubscriptionFindResponse(
subscription=subscription_pb2.Subscription(name=TEST_SUB_NAME))
getStubMock.return_value = stubMock
sub = opencue.api.findSubscription(TEST_SUB_NAME)
stubMock.Find.assert_called_with(
subscription_pb2.SubscriptionFindRequest(name=TEST_SUB_NAME), timeout=mock.ANY)
self.assertEqual(TEST_SUB_NAME, sub.name())
def find(self, name):
"""Returns a subscription by name.
:type name: str
:param name: name of subscription to find
:rtype: opencue.wrappers.subscription.Subscription
:return: the named subscription
"""
response = self.stub.Find(
subscription_pb2.SubscriptionFindRequest(name=name),
timeout=Cuebot.Timeout)
return Subscription(response.subscription)
def testFindSubscription(self, getStubMock):
stubMock = mock.Mock()
stubMock.Find.return_value = subscription_pb2.SubscriptionFindResponse(
subscription=subscription_pb2.Subscription(name=TEST_SUBSCRIPTION_NAME))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
subscription = show.findSubscription(TEST_SUBSCRIPTION_NAME)
stubMock.Find.assert_called_with(
subscription_pb2.SubscriptionFindRequest(name=TEST_SUBSCRIPTION_NAME), timeout=mock.ANY)
self.assertEqual(subscription.name(), TEST_SUBSCRIPTION_NAME)
def testFind(self, getStubMock):
stubMock = mock.Mock()
stubMock.Find.return_value = subscription_pb2.SubscriptionFindResponse(
subscription=subscription_pb2.Subscription(
name=TEST_SUBSCRIPTION_NAME))
getStubMock.return_value = stubMock
expected = opencue.wrappers.subscription.Subscription(
subscription_pb2.Subscription(name=TEST_SUBSCRIPTION_NAME))
wrapper = opencue.wrappers.subscription.Subscription()
subscription = wrapper.find(name=TEST_SUBSCRIPTION_NAME)
stubMock.Find.assert_called_with(
subscription_pb2.SubscriptionFindRequest(
name=TEST_SUBSCRIPTION_NAME),
timeout=mock.ANY)
self.assertEqual(expected.name(), subscription.name())
def find(self, name):
response = self.stub.Find(
subscription_pb2.SubscriptionFindRequest(name=name),
timeout=Cuebot.Timeout)
return Subscription(response.subscription)
