def getSubscription(uniq):
"""Returns a Subscription object from a unique identifier.
:type uniq: str
:param uniq: a unique subscription identifier
:rtype: Subscription
:return: a Subscription object"""
return Subscription(Cuebot.getStub('subscription').Get(
subscription_pb2.SubscriptionGetRequest(id=uniq), timeout=Cuebot.Timeout).subscription)
def testGetSubscription(self, getStubMock):
arbitraryId = '00000000-0000-0000-0000-012345678980'
stubMock = mock.Mock()
stubMock.Get.return_value = subscription_pb2.SubscriptionGetResponse(
subscription=subscription_pb2.Subscription(id=arbitraryId))
getStubMock.return_value = stubMock
sub = opencue.api.getSubscription(arbitraryId)
stubMock.Get.assert_called_with(
subscription_pb2.SubscriptionGetRequest(id=arbitraryId), timeout=mock.ANY)
self.assertEqual(arbitraryId, sub.id())
def get(self, subscription_id):
"""Returns a subscription by id.
:type subscription_id: str
:param subscription_id: id of subscription to get
:rtype: opencue.wrappers.subscription.Subscription
:return: the subscription of the id
"""
response = self.stub.Get(
subscription_pb2.SubscriptionGetRequest(id=subscription_id),
timeout=Cuebot.Timeout)
return Subscription(response.subscription)
def testGet(self, getStubMock):
stubMock = mock.Mock()
stubMock.Get.return_value = subscription_pb2.SubscriptionGetResponse(
subscription=subscription_pb2.Subscription(
id=TEST_SUBSCRIPTION_ID, name=TEST_SUBSCRIPTION_NAME))
getStubMock.return_value = stubMock
wrapper = opencue.wrappers.subscription.Subscription()
subscription = wrapper.get(TEST_SUBSCRIPTION_ID)
stubMock.Get.assert_called_with(
subscription_pb2.SubscriptionGetRequest(id=TEST_SUBSCRIPTION_ID),
timeout=mock.ANY)
self.assertEqual(subscription.id(), TEST_SUBSCRIPTION_ID)
def get(self, id):
response = self.stub.Get(
subscription_pb2.SubscriptionGetRequest(id=id),
timeout=Cuebot.Timeout)
return Subscription(response.subscription)
