1*760c253cSXin Li#!/usr/bin/env python3 2*760c253cSXin Li# -*- coding: utf-8 -*- 3*760c253cSXin Li# Copyright 2014 The ChromiumOS Authors 4*760c253cSXin Li# Use of this source code is governed by a BSD-style license that can be 5*760c253cSXin Li# found in the LICENSE file. 6*760c253cSXin Li 7*760c253cSXin Li"""The unittest of flags.""" 8*760c253cSXin Li 9*760c253cSXin Li 10*760c253cSXin Liimport unittest 11*760c253cSXin Li 12*760c253cSXin Liimport test_flag 13*760c253cSXin Li 14*760c253cSXin Li 15*760c253cSXin Liclass FlagTestCase(unittest.TestCase): 16*760c253cSXin Li """The unittest class.""" 17*760c253cSXin Li 18*760c253cSXin Li def test_test_flag(self): 19*760c253cSXin Li # Verify that test_flag.is_test exists, that it is a list, 20*760c253cSXin Li # and that it contains 1 element. 21*760c253cSXin Li self.assertTrue(isinstance(test_flag.is_test, list)) 22*760c253cSXin Li self.assertEqual(len(test_flag.is_test), 1) 23*760c253cSXin Li 24*760c253cSXin Li # Verify that the getting the flag works and that the flag 25*760c253cSXin Li # contains False, its starting value. 26*760c253cSXin Li save_flag = test_flag.GetTestMode() 27*760c253cSXin Li self.assertFalse(save_flag) 28*760c253cSXin Li 29*760c253cSXin Li # Verify that setting the flat to True, then getting it, works. 30*760c253cSXin Li test_flag.SetTestMode(True) 31*760c253cSXin Li self.assertTrue(test_flag.GetTestMode()) 32*760c253cSXin Li 33*760c253cSXin Li # Verify that setting the flag to False, then getting it, works. 34*760c253cSXin Li test_flag.SetTestMode(save_flag) 35*760c253cSXin Li self.assertFalse(test_flag.GetTestMode()) 36*760c253cSXin Li 37*760c253cSXin Li # Verify that test_flag.is_test still exists, that it still is a 38*760c253cSXin Li # list, and that it still contains 1 element. 39*760c253cSXin Li self.assertTrue(isinstance(test_flag.is_test, list)) 40*760c253cSXin Li self.assertEqual(len(test_flag.is_test), 1) 41*760c253cSXin Li 42*760c253cSXin Li 43*760c253cSXin Liif __name__ == "__main__": 44*760c253cSXin Li unittest.main() 45