def setUp(self):
project = 'test-project'
instance = 'test-instance'
options = {'project': project,
'zone': 'us-central1-f',
'instance_id': instance,
'config_dir': '/notfound',
'fix_stackdriver_labels_unsafe': True}
self.mockStub = mock.create_autospec(['projects'])
self.mockProjects = mock.create_autospec(
['metricDescriptors', 'timeSeries'])
self.mockMetricDescriptors = mock.create_autospec(
['create', 'delete', 'get'])
self.mockTimeSeries = mock.create_autospec(['create'])
self.mockStub.projects = Mock(return_value=self.mockProjects)
self.mockProjects.metricDescriptors = Mock(
return_value=self.mockMetricDescriptors)
self.mockProjects.timeSeries = Mock(return_value=self.mockTimeSeries)
# pylint: disable=invalid-name
self.mockCreateTimeSeries = Mock(spec=['execute'])
self.mockCreateDescriptor = Mock(spec=['execute'])
self.mockGetDescriptor = Mock(spec=['execute'])
self.mockDeleteDescriptor = Mock(spec=['execute'])
self.mockMetricDescriptors.create = Mock(
return_value=self.mockCreateDescriptor)
self.mockMetricDescriptors.delete = Mock(
return_value=self.mockDeleteDescriptor)
self.mockMetricDescriptors.get = Mock(
return_value=self.mockGetDescriptor)
self.mockTimeSeries.create = Mock(return_value=self.mockCreateTimeSeries)
self.service = StackdriverMetricsService(lambda: self.mockStub, options)
def setUp(self):
options = {
'project': 'PROJECT',
'source_path': 'IgnoreThis',
'config_dir': '/notfound',
'output_path': None
}
mockStub = mock.create_autospec(['projects', 'new_batch_http_request'])
mockStub.new_batch_http_request = Mock()
self.fake_batch_list = []
stackdriver = StackdriverMetricsService(lambda: mockStub, options)
self.upsertHandler = stackdriver_handlers.UpsertCustomDescriptorsHandler(
None, options, None)
self.mockProjects = mock.create_autospec(['metricDescriptors'])
self.mockMetricDescriptors = mock.create_autospec(
['create', 'delete', 'list'])
self.mockStub = mockStub
self.mockStub.projects = Mock(return_value=self.mockProjects)
self.mockProjects.metricDescriptors = Mock(
return_value=self.mockMetricDescriptors)
self.mockCreateDescriptor = Mock(spec=['execute'])
self.mockDeleteDescriptor = Mock(spec=['execute'])
self.mockListDescriptors = Mock(spec=['execute'])
self.mockMetricDescriptors.create = Mock(
return_value=self.mockCreateDescriptor)
self.mockMetricDescriptors.delete = Mock(
return_value=self.mockDeleteDescriptor)
self.mockMetricDescriptors.list = Mock(
return_value=self.mockListDescriptors)
self.mockMetricDescriptors.list_next = Mock(return_value=None)
self.options = options
self.project = options['project']
self.stackdriver = stackdriver
