def testCreateSubscription(self, getStubMock): stubMock = mock.Mock() stubMock.CreateSubscription.return_value = show_pb2.ShowCreateSubscriptionResponse( subscription=subscription_pb2.Subscription( name=TEST_SUBSCRIPTION_NAME, size=TEST_SUBSCRIPTION_SIZE, burst=TEST_SUBSCRIPTION_BURST)) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) allocation = opencue.wrappers.allocation.Allocation( facility_pb2.Allocation(id=TEST_ALLOCATION_ID)) subscription = show.createSubscription(allocation, TEST_SUBSCRIPTION_SIZE, TEST_SUBSCRIPTION_BURST) stubMock.CreateSubscription.assert_called_with( show_pb2.ShowCreateSubscriptionRequest( show=show.data, allocation_id=TEST_ALLOCATION_ID, size=TEST_SUBSCRIPTION_SIZE, burst=TEST_SUBSCRIPTION_BURST), timeout=mock.ANY) self.assertEqual(subscription.size(), TEST_SUBSCRIPTION_SIZE) self.assertEqual(subscription.burst(), TEST_SUBSCRIPTION_BURST)
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 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 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 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 testSetSize(self, getStubMock): stubMock = mock.Mock() stubMock.SetSize.return_value = subscription_pb2.SubscriptionSetSizeResponse( ) getStubMock.return_value = stubMock subscription = opencue.wrappers.subscription.Subscription( subscription_pb2.Subscription(name='testSubscription')) subscription.setSize(16) stubMock.SetSize.assert_called_with( subscription_pb2.SubscriptionSetSizeRequest( subscription=subscription.data, new_size=16), timeout=mock.ANY)
def testSetBurst(self, getStubMock): stubMock = mock.Mock() stubMock.SetBurst.return_value = subscription_pb2.SubscriptionSetBurstResponse( ) getStubMock.return_value = stubMock subscription = opencue.wrappers.subscription.Subscription( subscription_pb2.Subscription(name='testSubscription')) subscription.setBurst(15) stubMock.SetBurst.assert_called_with( subscription_pb2.SubscriptionSetBurstRequest( subscription=subscription.data, burst=15), timeout=mock.ANY)
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 testDelete(self, getStubMock): stubMock = mock.Mock() stubMock.Delete.return_value = subscription_pb2.SubscriptionDeleteResponse( ) getStubMock.return_value = stubMock subscription = opencue.wrappers.subscription.Subscription( subscription_pb2.Subscription(name=TEST_SUBSCRIPTION_NAME)) subscription.delete() stubMock.Delete.assert_called_with( subscription_pb2.SubscriptionDeleteRequest( subscription=subscription.data), timeout=mock.ANY)
def testGetSubscriptions(self, getStubMock): stubMock = mock.Mock() stubMock.GetSubscriptions.return_value = facility_pb2.AllocGetSubscriptionsResponse( subscriptions=subscription_pb2.SubscriptionSeq( subscriptions=[subscription_pb2.Subscription(name=TEST_SUBSCRIPTION_NAME)])) getStubMock.return_value = stubMock alloc = opencue.wrappers.allocation.Allocation( facility_pb2.Allocation(name=TEST_ALLOC_NAME)) subscriptions = alloc.getSubscriptions() stubMock.GetSubscriptions.assert_called_with( facility_pb2.AllocGetSubscriptionsRequest(allocation=alloc.data), timeout=mock.ANY) self.assertEqual(len(subscriptions), 1) self.assertEqual(subscriptions[0].name(), TEST_SUBSCRIPTION_NAME)
def testGetSubscriptions(self, getStubMock): stubMock = mock.Mock() stubMock.GetSubscriptions.return_value = show_pb2.ShowGetSubscriptionResponse( subscriptions=subscription_pb2.SubscriptionSeq( subscriptions=[subscription_pb2.Subscription(name=TEST_SUBSCRIPTION_NAME)]) ) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) subscriptions = show.getSubscriptions() stubMock.GetSubscriptions.assert_called_with( show_pb2.ShowGetSubscriptionRequest(show=show.data), timeout=mock.ANY) self.assertEqual(len(subscriptions), 1) self.assertEqual(subscriptions[0].name(), TEST_SUBSCRIPTION_NAME)