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

Unified Diff: infra_libs/ts_mon/common/test/targets_test.py

Issue 2213143002: Add infra_libs as a bootstrap dependency. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Removed the ugly import hack Created 4 years, 4 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 side-by-side diff with in-line comments
Download patch
Index: infra_libs/ts_mon/common/test/targets_test.py
diff --git a/infra_libs/ts_mon/common/test/targets_test.py b/infra_libs/ts_mon/common/test/targets_test.py
deleted file mode 100644
index e169b969b268de4da4491335ccf98aaa4c3bd715..0000000000000000000000000000000000000000
--- a/infra_libs/ts_mon/common/test/targets_test.py
+++ /dev/null
@@ -1,89 +0,0 @@
-# Copyright 2015 The Chromium Authors. All rights reserved.
-# Use of this source code is governed by a BSD-style license that can be
-# found in the LICENSE file.
-
-import unittest
-
-from infra_libs.ts_mon.common import targets
-from infra_libs.ts_mon.protos import metrics_pb2
-
-
-class DeviceTargetTest(unittest.TestCase):
-
- def test_populate_target(self):
- pb = metrics_pb2.MetricsData()
- t = targets.DeviceTarget('reg', 'role', 'net', 'host')
- t._populate_target_pb(pb)
- self.assertEquals(pb.network_device.metro, 'reg')
- self.assertEquals(pb.network_device.role, 'role')
- self.assertEquals(pb.network_device.hostgroup, 'net')
- self.assertEquals(pb.network_device.hostname, 'host')
- self.assertEquals(pb.network_device.realm, 'ACQ_CHROME')
- self.assertEquals(pb.network_device.alertable, True)
-
- def test_update_to_dict(self):
- target = targets.DeviceTarget('reg', 'role', 'net', 'host')
- self.assertEqual({
- 'region': 'reg',
- 'role': 'role',
- 'network': 'net',
- 'hostname': 'host'}, target.to_dict())
- target.update({'region': 'other', 'hostname': 'guest'})
- self.assertEqual({
- 'region': 'other',
- 'role': 'role',
- 'network': 'net',
- 'hostname': 'guest'}, target.to_dict())
-
- def test_update_private_field(self):
- target = targets.DeviceTarget('reg', 'role', 'net', 'host')
- with self.assertRaises(AttributeError):
- target.update({'realm': 'boo'})
-
- def test_update_nonexistent_field(self):
- target = targets.DeviceTarget('reg', 'role', 'net', 'host')
- # Simulate a bug: exporting a non-existent field.
- target._fields += ('bad',)
- with self.assertRaises(AttributeError):
- target.update({'bad': 'boo'})
-
-
-class TaskTargetTest(unittest.TestCase):
-
- def test_populate_target(self):
- pb = metrics_pb2.MetricsData()
- t = targets.TaskTarget('serv', 'job', 'reg', 'host')
- t._populate_target_pb(pb)
- self.assertEquals(pb.task.service_name, 'serv')
- self.assertEquals(pb.task.job_name, 'job')
- self.assertEquals(pb.task.data_center, 'reg')
- self.assertEquals(pb.task.host_name, 'host')
- self.assertEquals(pb.task.task_num, 0)
-
- def test_update_to_dict(self):
- target = targets.TaskTarget('serv', 'job', 'reg', 'host', 5)
- self.assertEqual({
- 'service_name': 'serv',
- 'job_name': 'job',
- 'region': 'reg',
- 'hostname': 'host',
- 'task_num': 5}, target.to_dict())
- target.update({'region': 'other', 'hostname': 'guest'})
- self.assertEqual({
- 'service_name': 'serv',
- 'job_name': 'job',
- 'region': 'other',
- 'hostname': 'guest',
- 'task_num': 5}, target.to_dict())
-
- def test_update_private_field(self):
- target = targets.TaskTarget('serv', 'job', 'reg', 'host')
- with self.assertRaises(AttributeError):
- target.update({'realm': 'boo'})
-
- def test_update_nonexistent_field(self):
- target = targets.TaskTarget('serv', 'job', 'reg', 'host')
- # Simulate a bug: exporting a non-existent field.
- target._fields += ('bad',)
- with self.assertRaises(AttributeError):
- target.update({'bad': 'boo'})

Powered by Google App Engine
This is Rietveld 408576698