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 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): 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(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.PubSubMonitor', autospec=True).start()
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 tearDown(self): config.reset_for_unittest() mock.patch.stopall() super(InstrumentEndpointTest, self).tearDown()
def tearDown(self): config.reset_for_unittest() mock.patch.stopall() super(InitializeTest, self).tearDown()
def tearDown(self): mock.patch.stopall() config.reset_for_unittest() super(HandlersTest, self).tearDown()
def tearDown(self): config.reset_for_unittest() self.assertEqual([], list(shared.global_metrics_callbacks)) mock.patch.stopall() super(InstrumentEndpointTest, self).tearDown()