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()