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

Unified Diff: infra_libs/test/experiments_test.py

Issue 1365583002: Have service_manager start a cloudtail attached to each service it starts. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master
Patch Set: Just use the default project ID Created 5 years, 3 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
« infra_libs/experiments.py ('K') | « infra_libs/experiments.py ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: infra_libs/test/experiments_test.py
diff --git a/infra_libs/test/experiments_test.py b/infra_libs/test/experiments_test.py
new file mode 100644
index 0000000000000000000000000000000000000000..4f38a775cfbe1bc89a518bdeee6d5e14f1fa2745
--- /dev/null
+++ b/infra_libs/test/experiments_test.py
@@ -0,0 +1,46 @@
+# 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
+
+import mock
+
+from infra_libs import experiments
+
+
+class ExperimentsTest(unittest.TestCase):
+ def setUp(self):
+ self.mock_getfqdn = mock.patch('socket.getfqdn').start()
+
+ def tearDown(self):
+ mock.patch.stopall()
+
+ def test_is_active_for_host(self):
+ self.mock_getfqdn.return_value = 'vm123-m1'
+
+ self.assertFalse(experiments.is_active_for_host('f1', 0))
+ self.assertFalse(experiments.is_active_for_host('f2', 0))
+ self.assertFalse(experiments.is_active_for_host('f3', 0))
+
+ self.assertTrue(experiments.is_active_for_host('f1', 50))
+ self.assertFalse(experiments.is_active_for_host('f2', 50))
+ self.assertTrue(experiments.is_active_for_host('f3', 50))
+
+ self.assertTrue(experiments.is_active_for_host('f1', 100))
+ self.assertTrue(experiments.is_active_for_host('f2', 100))
+ self.assertTrue(experiments.is_active_for_host('f3', 100))
+
+ self.mock_getfqdn.return_value = 'vm124-m1'
+
+ self.assertFalse(experiments.is_active_for_host('f1', 0))
+ self.assertFalse(experiments.is_active_for_host('f2', 0))
+ self.assertFalse(experiments.is_active_for_host('f3', 0))
+
+ self.assertTrue(experiments.is_active_for_host('f1', 50))
+ self.assertTrue(experiments.is_active_for_host('f2', 50))
+ self.assertFalse(experiments.is_active_for_host('f3', 50))
+
+ self.assertTrue(experiments.is_active_for_host('f1', 100))
+ self.assertTrue(experiments.is_active_for_host('f2', 100))
+ self.assertTrue(experiments.is_active_for_host('f3', 100))
« infra_libs/experiments.py ('K') | « infra_libs/experiments.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698