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)
