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 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)