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