Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(212)

Side by Side Diff: infra_libs/ts_mon/test/config_test.py

Issue 2119933003: Add Configurable metric_name_prefix (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Add trailing slash Created 4 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « infra_libs/ts_mon/config.py ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 # Copyright 2015 The Chromium Authors. All rights reserved. 1 # Copyright 2015 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be 2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file. 3 # found in the LICENSE file.
4 4
5 import argparse 5 import argparse
6 import json 6 import json
7 import os 7 import os
8 import requests 8 import requests
9 import tempfile 9 import tempfile
10 import unittest 10 import unittest
(...skipping 286 matching lines...) Expand 10 before | Expand all | Expand 10 after
297 config.add_argparse_options(p) 297 config.add_argparse_options(p)
298 args = p.parse_args(['--ts-mon-credentials', '/path/to/creds.p8.json', 298 args = p.parse_args(['--ts-mon-credentials', '/path/to/creds.p8.json',
299 '--ts-mon-target-type', 'task', 299 '--ts-mon-target-type', 'task',
300 '--ts-mon-task-service-name', 'serv', 300 '--ts-mon-task-service-name', 'serv',
301 '--ts-mon-task-region', 'reg', 301 '--ts-mon-task-region', 'reg',
302 '--ts-mon-task-hostname', 'host', 302 '--ts-mon-task-hostname', 'host',
303 '--ts-mon-task-number', '1']) 303 '--ts-mon-task-number', '1'])
304 with self.assertRaises(SystemExit): 304 with self.assertRaises(SystemExit):
305 config.process_argparse_options(args) 305 config.process_argparse_options(args)
306 306
307 def test_metric_name_prefix(self):
308 p = argparse.ArgumentParser()
309 config.add_argparse_options(p)
310 args = p.parse_args(['--ts-mon-metric-name-prefix', '/test/random/'])
311 config.process_argparse_options(args)
312 self.assertEqual('/test/random/', interface.state.metric_name_prefix)
313
307 @mock.patch('infra_libs.ts_mon.common.monitors.NullMonitor', autospec=True) 314 @mock.patch('infra_libs.ts_mon.common.monitors.NullMonitor', autospec=True)
308 def test_no_args(self, fake_monitor): 315 def test_no_args(self, fake_monitor):
309 singleton = mock.Mock() 316 singleton = mock.Mock()
310 fake_monitor.return_value = singleton 317 fake_monitor.return_value = singleton
311 p = argparse.ArgumentParser() 318 p = argparse.ArgumentParser()
312 config.add_argparse_options(p) 319 config.add_argparse_options(p)
313 args = p.parse_args([]) 320 args = p.parse_args([])
314 config.process_argparse_options(args) 321 config.process_argparse_options(args)
315 self.assertEqual(1, len(fake_monitor.mock_calls)) 322 self.assertEqual(1, len(fake_monitor.mock_calls))
323 self.assertEqual('/chrome/infra/', interface.state.metric_name_prefix)
316 self.assertIs(interface.state.global_monitor, singleton) 324 self.assertIs(interface.state.global_monitor, singleton)
317 325
318 @mock.patch('requests.get', autospec=True) 326 @mock.patch('requests.get', autospec=True)
319 def test_gce_region(self, mock_get): 327 def test_gce_region(self, mock_get):
320 r = mock_get.return_value 328 r = mock_get.return_value
321 r.status_code = 200 329 r.status_code = 200
322 r.text = 'projects/182615506979/zones/us-central1-f' 330 r.text = 'projects/182615506979/zones/us-central1-f'
323 331
324 self.assertEquals('us-central1-f', config._default_region('foo.golo')) 332 self.assertEquals('us-central1-f', config._default_region('foo.golo'))
325 333
(...skipping 24 matching lines...) Expand all
350 with infra_libs.temporary_directory() as temp_dir: 358 with infra_libs.temporary_directory() as temp_dir:
351 filename = os.path.join(temp_dir, 'config') 359 filename = os.path.join(temp_dir, 'config')
352 with open(filename, 'w') as fh: 360 with open(filename, 'w') as fh:
353 fh.write('not a json file') 361 fh.write('not a json file')
354 362
355 with self.assertRaises(ValueError): 363 with self.assertRaises(ValueError):
356 config.load_machine_config(filename) 364 config.load_machine_config(filename)
357 365
358 def test_load_machine_config_not_exists(self): 366 def test_load_machine_config_not_exists(self):
359 self.assertEquals({}, config.load_machine_config('does not exist')) 367 self.assertEquals({}, config.load_machine_config('does not exist'))
OLDNEW
« no previous file with comments | « infra_libs/ts_mon/config.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698