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)
