def testGetActiveShows(self, getStubMock): show1Name = 'first-show' show2Name = 'second-show' stubMock = mock.Mock() stubMock.GetActiveShows.return_value = show_pb2.ShowGetActiveShowsResponse( shows=show_pb2.ShowSeq( shows=[show_pb2.Show(name=show1Name), show_pb2.Show(name=show2Name)])) getStubMock.return_value = stubMock showList = opencue.api.getActiveShows() stubMock.GetActiveShows.assert_called_with( show_pb2.ShowGetActiveShowsRequest(), timeout=mock.ANY) self.assertEqual([show1Name, show2Name], [show.name() for show in showList])
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 testDelete(self, getStubMock): stubMock = mock.Mock() stubMock.Delete.return_value = show_pb2.ShowDeleteResponse() getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) show.delete() stubMock.Delete.assert_called_with( show_pb2.ShowDeleteRequest(show=show.data), timeout=mock.ANY)
def testGetShows(self, getStubMock): stubMock = mock.Mock() stubMock.GetShows.return_value = show_pb2.ShowGetShowsResponse( shows=show_pb2.ShowSeq(shows=[show_pb2.Show(name=TEST_SHOW_NAME)])) getStubMock.return_value = stubMock showList = opencue.api.getShows() stubMock.GetShows.assert_called_with(show_pb2.ShowGetShowsRequest(), timeout=mock.ANY) self.assertEqual([TEST_SHOW_NAME], [show.name() for show in showList])
def testSetActive(self, getStubMock): stubMock = mock.Mock() stubMock.SetActive.return_value = show_pb2.ShowSetActiveResponse() getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) show.setActive(TEST_ENABLE_VALUE) stubMock.SetActive.assert_called_with(show_pb2.ShowSetActiveRequest( show=show.data, value=TEST_ENABLE_VALUE), timeout=mock.ANY)
def testCreateShow(self, getStubMock): stubMock = mock.Mock() stubMock.CreateShow.return_value = show_pb2.ShowCreateShowResponse( show=show_pb2.Show(name=TEST_SHOW_NAME)) getStubMock.return_value = stubMock newShow = opencue.api.createShow(TEST_SHOW_NAME) stubMock.CreateShow.assert_called_with( show_pb2.ShowCreateShowRequest(name=TEST_SHOW_NAME), timeout=mock.ANY) self.assertEqual(TEST_SHOW_NAME, newShow.name())
def testEnableDispatching(self, getStubMock): stubMock = mock.Mock() stubMock.EnableDispatching.return_value = show_pb2.ShowEnableDispatchingResponse() getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) show.enableDispatching(TEST_ENABLE_VALUE) stubMock.EnableDispatching.assert_called_with( show_pb2.ShowEnableDispatchingRequest(show=show.data, enabled=TEST_ENABLE_VALUE), timeout=mock.ANY)
def testSetDefaultMinCores(self, getStubMock): stubMock = mock.Mock() stubMock.SetDefaultMinCores.return_value = show_pb2.ShowSetDefaultMinCoresResponse() getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) show.setDefaultMinCores(TEST_MIN_CORES) stubMock.SetDefaultMinCores.assert_called_with( show_pb2.ShowSetDefaultMinCoresRequest(show=show.data, min_cores=TEST_MIN_CORES), timeout=mock.ANY)
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 testDeleteShow(self, getStubMock): arbitraryId = '00000000-0000-0000-0000-012345678980' showToDelete = show_pb2.Show(id=arbitraryId) stubMock = mock.Mock() stubMock.FindShow.return_value = show_pb2.ShowFindShowResponse(show=showToDelete) stubMock.Delete.return_value = show_pb2.ShowDeleteResponse() getStubMock.return_value = stubMock opencue.api.deleteShow(arbitraryId) stubMock.Delete.assert_called_with( show_pb2.ShowDeleteRequest(show=showToDelete), timeout=mock.ANY)
def testSetDefaultMaxGpus(self, getStubMock): stubMock = mock.Mock() stubMock.SetDefaultMaxGpus.return_value = show_pb2.ShowSetDefaultMaxGpusResponse( ) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) show.setDefaultMaxGpus(TEST_MAX_GPUS) stubMock.SetDefaultMaxGpus.assert_called_with( show_pb2.ShowSetDefaultMaxGpusRequest(show=show.data, max_gpus=TEST_MAX_GPUS), timeout=mock.ANY)
def testCreateOwner(self, getStubMock): stubMock = mock.Mock() stubMock.CreateOwner.return_value = show_pb2.ShowCreateOwnerResponse( owner=host_pb2.Owner(name=TEST_OWNER_NAME)) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) owner = show.createOwner(TEST_OWNER_NAME) stubMock.CreateOwner.assert_called_with( show_pb2.ShowCreateOwnerRequest(show=show.data, name=TEST_OWNER_NAME), timeout=mock.ANY) self.assertEqual(owner.name, TEST_OWNER_NAME)
def testGetFilters(self, getStubMock): stubMock = mock.Mock() stubMock.GetFilters.return_value = show_pb2.ShowGetFiltersResponse( filters=filter_pb2.FilterSeq(filters=[filter_pb2.Filter(name=TEST_FILTER_NAME)])) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) filters = show.getFilters() stubMock.GetFilters.assert_called_with( show_pb2.ShowGetFiltersRequest(show=show.data), timeout=mock.ANY) self.assertEqual(len(filters), 1) self.assertEqual(filters[0].name(), TEST_FILTER_NAME)
def testFindFilter(self, getStubMock): stubMock = mock.Mock() stubMock.FindFilter.return_value = show_pb2.ShowFindFilterResponse( filter=filter_pb2.Filter(name=TEST_FILTER_NAME)) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) filter_found = show.findFilter(TEST_FILTER_NAME) stubMock.FindFilter.assert_called_with(show_pb2.ShowFindFilterRequest( show=show.data, name=TEST_FILTER_NAME), timeout=mock.ANY) self.assertEqual(filter_found.name(), TEST_FILTER_NAME)
def testGetRootGroup(self, getStubMock): stubMock = mock.Mock() stubMock.GetRootGroup.return_value = show_pb2.ShowGetRootGroupResponse( group=job_pb2.Group(name=TEST_GROUP_NAME, department=TEST_GROUP_DEPT)) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) rootGroup = show.getRootGroup() stubMock.GetRootGroup.assert_called_with( show_pb2.ShowGetRootGroupRequest(show=show.data), timeout=mock.ANY) self.assertEqual(rootGroup.name(), TEST_GROUP_NAME) self.assertEqual(rootGroup.department(), TEST_GROUP_DEPT)
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)
def testGetJobWhiteboard(self, getStubMock): stubMock = mock.Mock() stubMock.GetJobWhiteboard.return_value = show_pb2.ShowGetJobWhiteboardResponse( whiteboard=job_pb2.NestedGroup(name=TEST_GROUP_NAME, department=TEST_GROUP_DEPT)) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) whiteboard = show.getJobWhiteboard() stubMock.GetJobWhiteboard.assert_called_with( show_pb2.ShowGetJobWhiteboardRequest(show=show.data), timeout=mock.ANY) self.assertEqual(whiteboard.name, TEST_GROUP_NAME) self.assertEqual(whiteboard.department, TEST_GROUP_DEPT)
def testGetServiceOverrides(self, getStubMock): stubMock = mock.Mock() stubMock.GetServiceOverrides.return_value = show_pb2.ShowGetServiceOverridesResponse( service_overrides=service_pb2.ServiceOverrideSeq( service_overrides=[ service_pb2.ServiceOverride(data=service_pb2.Service(name=TEST_SERVICE_NAME)) ])) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) overrides = show.getServiceOverrides() stubMock.GetServiceOverrides.assert_called_with( show_pb2.ShowGetServiceOverridesRequest(show=show.data), timeout=mock.ANY) self.assertEqual(overrides[0].data.name, TEST_SERVICE_NAME)
def testGetGroups(self, getStubMock): stubMock = mock.Mock() stubMock.GetGroups.return_value = show_pb2.ShowGetGroupsResponse( groups=job_pb2.GroupSeq(groups=[ job_pb2.Group(name=TEST_GROUP_NAME, department=TEST_GROUP_DEPT) ])) getStubMock.return_value = stubMock show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME)) groups = show.getGroups() stubMock.GetGroups.assert_called_with( show_pb2.ShowGetGroupsRequest(show=show.data), timeout=mock.ANY) self.assertEqual(len(groups), 1) self.assertEqual(groups[0].name(), TEST_GROUP_NAME) self.assertEqual(groups[0].department(), TEST_GROUP_DEPT)