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 createSubscription(self, allocation, size, burst): """Creates a new subscription @type allocation: Allocation @param allocation: Allocation object @type size: float @param size: Allocation size @type burst: float @param burst: Allocation burst @rtype: Subscription @return: The created subscription object """ response = self.stub.CreateSubscription(show_pb2.ShowCreateSubscriptionRequest( show=self.data, allocation_id=allocation.id, size=size, burst=burst), timeout=Cuebot.Timeout) return opencue.wrappers.subscription.Subscription(response.subscription)
def createSubscription(self, allocation, size, burst): """Creates a new subscription for the show. A subscription links a show to an allocation, and determines how many cores the show can utilize within that allocation. :type allocation: opencue.wrappers.allocation.Allocation :param allocation: allocation to subscribe to :type size: float :param size: number of cores the show is allowed to use consistently :type burst: float :param burst: number of cores the show is allowed to burst to :rtype: opencue.wrappers.subscription.Subscription :return: the created subscription object """ response = self.stub.CreateSubscription(show_pb2.ShowCreateSubscriptionRequest( show=self.data, allocation_id=allocation.id(), size=size, burst=burst), timeout=Cuebot.Timeout) return opencue.wrappers.subscription.Subscription(response.subscription)