Source Code for Module tests.test_lockable

 1  #!/usr/bin/env python 
 2   
 3  # This file is part of PARPG. 
 4  #  
 5  # PARPG is free software: you can redistribute it and/or modify 
 6  # it under the terms of the GNU General Public License as published by 
 7  # the Free Software Foundation, either version 3 of the License, or 
 8  # (at your option) any later version. 
 9  #  
10  # PARPG is distributed in the hope that it will be useful, 
11  # but WITHOUT ANY WARRANTY; without even the implied warranty of 
12  # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 
13  # GNU General Public License for more details. 
14  #  
15  # You should have received a copy of the GNU General Public License 
16  # along with PARPG.  If not, see <http://www.gnu.org/licenses/>. 
17   
18  import unittest 
19  from parpg.objects.base import Lockable, GameObject 
20   
21   
22 -class  TestLockable(unittest.TestCase): 
23   
24 -    class LockableObject (GameObject, Lockable): 
25 -        def __init__ (self, ID, **kwargs): 
26              GameObject.__init__(self, ID, **kwargs) 
27              Lockable.__init__(self, **kwargs) 
28   
29 -    def testConstructor(self): 
30          """ Test Lockable mixin constructor """ 
31          lockable = self.LockableObject(4) 
32          self.assertEqual(lockable.locked,False) 
33          self.assertEqual(lockable.is_open,True) 
34          lockable = self.LockableObject(4,locked=False,is_open=False) 
35          self.assertEqual(lockable.locked, False) 
36          self.assertEqual(lockable.is_open, False) 
37          lockable = self.LockableObject(4,locked=True) 
38          self.assertEqual(lockable.locked, True) 
39          self.assertEqual(lockable.is_open, False) 
40   
41 -    def testLockUnlock(self): 
42          """ Test Lockable mixin locking and unlocking """ 
43          lockable = self.LockableObject(4) 
44          lockable.open() 
45          self.assertEqual(lockable.is_open, True) 
46          lockable.lock() 
47          self.assertEqual(lockable.locked, True) 
48          self.assertEqual(lockable.is_open,False) 
49          self.assertRaises(ValueError, lockable.open) 
50          lockable.unlock() 
51          self.assertEqual(lockable.locked, False) 
52          self.assertEqual(lockable.is_open,False)     
53   
54  if __name__ == '__main__': 
55      unittest.main() 
56