def setUp(self):
target = targets.TaskTarget('test_service', 'test_job', 'test_region',
'test_host')
self.mock_state = interface.State(target=target)
self.state_patcher = mock.patch(
'infra_libs.ts_mon.common.interface.state', new=self.mock_state)
self.state_patcher.start()
def setUp(self):
super(TSMonJSHandlerTest, self).setUp()
config.reset_for_unittest()
target = targets.TaskTarget('test_service', 'test_job', 'test_region',
'test_host')
self.mock_state = interface.State(target=target)
mock.patch('infra_libs.ts_mon.common.interface.state',
new=self.mock_state).start()
self.request = webapp2.Request.blank('/_/ts_mon_js')
self.response = webapp2.Response()
self.ts_mon_handler = handlers.TSMonJSHandler(request=self.request,
response=self.response)
self.ts_mon_handler.register_metrics([
metrics.BooleanMetric(
'frontend/boolean_test',
'Boolean metric test',
field_spec=[metrics.StringField('client_id')]),
])
self.ts_mon_handler.xsrf_is_valid = mock.Mock(return_value=True)
self.mock_timestamp = 1537821859
def time_fn():
return self.mock_timestamp
self.ts_mon_handler.time_fn = time_fn
def tearDown(self):
# It's important to call close() before un-setting the mock state object,
# because any FlushThread started by the test is stored in that mock state
# and needs to be stopped before running any other tests.
interface.close()
# This should probably live in interface.close()
interface.state = interface.State()
super(GlobalsTest, self).tearDown()
def setUp(self):
interface.state = interface.State()
interface.state.metric_name_prefix = '/infra/test/'
interface.state.target = targets.TaskTarget(
service_name='service', job_name='job', region='region',
hostname='hostname', task_num=0)
interface.state.store = metric_store.InProcessMetricStore(
interface.state, time.time)
def setUp(self):
super(HandlersTest, self).setUp()
config.reset_for_unittest()
target = targets.TaskTarget('test_service', 'test_job', 'test_region',
'test_host')
self.mock_state = interface.State(target=target)
mock.patch('infra_libs.ts_mon.common.interface.state',
new=self.mock_state).start()
def setUp(self):
interface.state = interface.State()
interface.state.metric_name_prefix = '/infra/test/'
interface.state.target = targets.TaskTarget(service_name='service',
job_name='job',
region='region',
hostname='hostname',
task_num=0)
self.time_fn = mock.create_autospec(time.time, spec_set=True)
interface.state.store = metric_store.InProcessMetricStore(
interface.state, self.time_fn)
def setUp(self):
super(InitializeTest, self).setUp()
config.reset_for_unittest()
target = targets.TaskTarget('test_service', 'test_job', 'test_region',
'test_host')
self.mock_state = interface.State(target=target)
self.mock_state.metrics = copy.copy(interface.state.metrics)
mock.patch('infra_libs.ts_mon.common.interface.state',
new=self.mock_state).start()
mock.patch('infra_libs.ts_mon.common.monitors.HttpsMonitor',
autospec=True).start()
def setUp(self):
super(MetricStoreTestBase, self).setUp()
self.mock_time = mock.create_autospec(time.time, spec_set=True)
self.mock_time.return_value = 1234
target = targets.TaskTarget('test_service', 'test_job', 'test_region',
'test_host')
self.state = interface.State(store_ctor=self.create_store,
target=target)
mock.patch('infra_libs.ts_mon.common.interface.state',
new=self.state).start()
self.store = self.state.store
self.metric = metrics.Metric('foo', 'desc', None)
def setUp(self):
super(HandlersTest, self).setUp()
config.reset_for_unittest()
target = targets.TaskTarget('test_service', 'test_job', 'test_region',
'test_host')
self.mock_state = interface.State(target=target)
mock.patch('infra_libs.ts_mon.common.interface.state',
new=self.mock_state).start()
# Workaround the fact that 'system' module is not mocked.
class _memory_usage(object):
def current(self):
return 10.0
self.mock(runtime, 'memory_usage', _memory_usage)
def setUp(self):
super(TestBase, self).setUp()
target = targets.TaskTarget('test_service', 'test_job',
'test_region', 'test_host')
self.mock_state = interface.State(target=target)
self.state_patcher = mock.patch('infra_libs.ts_mon.common.interface.state',
new=self.mock_state)
self.state_patcher.start()
self.mock_state.target = targets.TaskTarget(
service_name='service', job_name='job', region='region',
hostname='hostname', task_num=0)
self.time_fn = mock.create_autospec(time.time, spec_set=True)
self.mock_state.store = metric_store.InProcessMetricStore(
self.mock_state, self.time_fn)
def setUp(self):
super(GlobalsTest, self).setUp()
interface.state = interface.State()
self.mock(config, 'load_machine_config', lambda x: {})
def setUp(self):
interface.state = interface.State()
interface.state.reset_for_unittest()
interface.state.target = targets.TaskTarget('test_service', 'test_job',
'test_region', 'test_host')
ts_mon.reset_for_unittest()
