| #!/usr/bin/env python3 |
| # -*- coding: utf-8 -*- |
| # Copyright 2014 The ChromiumOS Authors |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """Unit tests for config.py""" |
| |
| |
| import unittest |
| |
| import config |
| |
| |
| class ConfigTestCase(unittest.TestCase): |
| """Class for the config unit tests.""" |
| |
| def test_config(self): |
| # Verify that config exists, that it's a dictionary, and that it's |
| # empty. |
| self.assertTrue(isinstance(config.config, dict)) |
| self.assertEqual(len(config.config), 0) |
| |
| # Verify that attempting to get a non-existant key out of the |
| # dictionary returns None. |
| self.assertIsNone(config.GetConfig("rabbit")) |
| self.assertIsNone(config.GetConfig("key1")) |
| |
| config.AddConfig("key1", 16) |
| config.AddConfig("key2", 32) |
| config.AddConfig("key3", "third value") |
| |
| # Verify that after 3 calls to AddConfig we have 3 values in the |
| # dictionary. |
| self.assertEqual(len(config.config), 3) |
| |
| # Verify that GetConfig works and gets the expected values. |
| self.assertIs(config.GetConfig("key2"), 32) |
| self.assertIs(config.GetConfig("key3"), "third value") |
| self.assertIs(config.GetConfig("key1"), 16) |
| |
| # Re-set config. |
| config.config.clear() |
| |
| # Verify that config exists, that it's a dictionary, and that it's |
| # empty. |
| self.assertTrue(isinstance(config.config, dict)) |
| self.assertEqual(len(config.config), 0) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |