46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
#!/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.
|
||
|
|
|
||
|
|
"""The unittest of flags."""
|
||
|
|
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import test_flag
|
||
|
|
|
||
|
|
|
||
|
|
class FlagTestCase(unittest.TestCase):
|
||
|
|
"""The unittest class."""
|
||
|
|
|
||
|
|
def test_test_flag(self):
|
||
|
|
# Verify that test_flag.is_test exists, that it is a list,
|
||
|
|
# and that it contains 1 element.
|
||
|
|
self.assertTrue(isinstance(test_flag.is_test, list))
|
||
|
|
self.assertEqual(len(test_flag.is_test), 1)
|
||
|
|
|
||
|
|
# Verify that the getting the flag works and that the flag
|
||
|
|
# contains False, its starting value.
|
||
|
|
save_flag = test_flag.GetTestMode()
|
||
|
|
self.assertFalse(save_flag)
|
||
|
|
|
||
|
|
# Verify that setting the flat to True, then getting it, works.
|
||
|
|
test_flag.SetTestMode(True)
|
||
|
|
self.assertTrue(test_flag.GetTestMode())
|
||
|
|
|
||
|
|
# Verify that setting the flag to False, then getting it, works.
|
||
|
|
test_flag.SetTestMode(save_flag)
|
||
|
|
self.assertFalse(test_flag.GetTestMode())
|
||
|
|
|
||
|
|
# Verify that test_flag.is_test still exists, that it still is a
|
||
|
|
# list, and that it still contains 1 element.
|
||
|
|
self.assertTrue(isinstance(test_flag.is_test, list))
|
||
|
|
self.assertEqual(len(test_flag.is_test), 1)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|