54 lines
1.5 KiB
Python
Executable File
54 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# Copyright 2014 The Chromium OS Authors. All rights reserved.
|
|
# 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"""
|
|
|
|
from __future__ import print_function
|
|
|
|
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()
|