def testSetNumInstances_CurrentModule(self): self.request_data.get_dispatcher().AndReturn(self.dispatcher) self.request_data.get_module(None).AndReturn('default') self.dispatcher.set_num_instances('default', '1', 2) self.mox.ReplayAll() modules.set_num_instances(version='1', instances=2) self.mox.VerifyAll()
def testSetNumInstances_CurrentVersionDifferentModule(self): self.request_data.get_dispatcher().AndReturn(self.dispatcher) self.request_data.get_version(None).AndReturn('1') self.dispatcher.get_versions('other').AndReturn(['1']) self.dispatcher.set_num_instances('other', '1', 2) self.mox.ReplayAll() modules.set_num_instances(module='other', instances=2) self.mox.VerifyAll()
def testSetNumInstances_AllDefaults(self): """Test we return the expected results when no args are passed.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.instances = 15 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(15)
def testSetNumInstances_NoModule(self): """Test we return the expected results when no module is passed.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.version = 'v1' expected_request.instances = 14 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(14, version='v1')
def testSetNumInstances_NoVersion(self): """Test we return the expected results when no version is passed.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.module = 'module1' expected_request.instances = 13 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(13, 'module1')
def testSetNumInstances(self): """Test we return the expected results.""" expected_request = modules_service_pb2.SetNumInstancesRequest() expected_request.module = 'module1' expected_request.version = 'v1' expected_request.instances = 12 service_response = modules_service_pb2.SetNumInstancesResponse() self.SetSuccessExpectations('SetNumInstances', expected_request, service_response) modules.set_num_instances(12, 'module1', 'v1')
def testSetNumInstances(self): self.request_data.get_dispatcher().AndReturn(self.dispatcher) self.dispatcher.set_num_instances('default', '1', 2) self.mox.ReplayAll() modules.set_num_instances(2, 'default', '1') self.mox.VerifyAll()