crosperf/config_unittest.py

#!/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()