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