1 #!/usr/bin/env python 2 3 import unittest 4 import os 5 from test import test_support 6 7 # Skip this test if the _tkinter module wasn't built. 8 _tkinter = test_support.import_module('_tkinter') 9 10 from Tkinter import Tcl 11 from _tkinter import TclError 12 13 14 class TkinterTest(unittest.TestCase): 15 16 def testFlattenLen(self): 17 # flatten(<object with no length>) 18 self.assertRaises(TypeError, _tkinter._flatten, True) 19 20 21 class TclTest(unittest.TestCase): 22 23 def setUp(self): 24 self.interp = Tcl() 25 26 def testEval(self): 27 tcl = self.interp 28 tcl.eval('set a 1') 29 self.assertEqual(tcl.eval('set a'),'1') 30 31 def testEvalException(self): 32 tcl = self.interp 33 self.assertRaises(TclError,tcl.eval,'set a') 34 35 def testEvalException2(self): 36 tcl = self.interp 37 self.assertRaises(TclError,tcl.eval,'this is wrong') 38 39 def testCall(self): 40 tcl = self.interp 41 tcl.call('set','a','1') 42 self.assertEqual(tcl.call('set','a'),'1') 43 44 def testCallException(self): 45 tcl = self.interp 46 self.assertRaises(TclError,tcl.call,'set','a') 47 48 def testCallException2(self): 49 tcl = self.interp 50 self.assertRaises(TclError,tcl.call,'this','is','wrong') 51 52 def testSetVar(self): 53 tcl = self.interp 54 tcl.setvar('a','1') 55 self.assertEqual(tcl.eval('set a'),'1') 56 57 def testSetVarArray(self): 58 tcl = self.interp 59 tcl.setvar('a(1)','1') 60 self.assertEqual(tcl.eval('set a(1)'),'1') 61 62 def testGetVar(self): 63 tcl = self.interp 64 tcl.eval('set a 1') 65 self.assertEqual(tcl.getvar('a'),'1') 66 67 def testGetVarArray(self): 68 tcl = self.interp 69 tcl.eval('set a(1) 1') 70 self.assertEqual(tcl.getvar('a(1)'),'1') 71 72 def testGetVarException(self): 73 tcl = self.interp 74 self.assertRaises(TclError,tcl.getvar,'a') 75 76 def testGetVarArrayException(self): 77 tcl = self.interp 78 self.assertRaises(TclError,tcl.getvar,'a(1)') 79 80 def testUnsetVar(self): 81 tcl = self.interp 82 tcl.setvar('a',1) 83 self.assertEqual(tcl.eval('info exists a'),'1') 84 tcl.unsetvar('a') 85 self.assertEqual(tcl.eval('info exists a'),'0') 86 87 def testUnsetVarArray(self): 88 tcl = self.interp 89 tcl.setvar('a(1)',1) 90 tcl.setvar('a(2)',2) 91 self.assertEqual(tcl.eval('info exists a(1)'),'1') 92 self.assertEqual(tcl.eval('info exists a(2)'),'1') 93 tcl.unsetvar('a(1)') 94 self.assertEqual(tcl.eval('info exists a(1)'),'0') 95 self.assertEqual(tcl.eval('info exists a(2)'),'1') 96 97 def testUnsetVarException(self): 98 tcl = self.interp 99 self.assertRaises(TclError,tcl.unsetvar,'a') 100 101 def testEvalFile(self): 102 tcl = self.interp 103 filename = "testEvalFile.tcl" 104 fd = open(filename,'w') 105 script = """set a 1 106 set b 2 107 set c [ expr $a + $b ] 108 """ 109 fd.write(script) 110 fd.close() 111 tcl.evalfile(filename) 112 os.remove(filename) 113 self.assertEqual(tcl.eval('set a'),'1') 114 self.assertEqual(tcl.eval('set b'),'2') 115 self.assertEqual(tcl.eval('set c'),'3') 116 117 def testEvalFileException(self): 118 tcl = self.interp 119 filename = "doesnotexists" 120 try: 121 os.remove(filename) 122 except Exception,e: 123 pass 124 self.assertRaises(TclError,tcl.evalfile,filename) 125 126 def testPackageRequireException(self): 127 tcl = self.interp 128 self.assertRaises(TclError,tcl.eval,'package require DNE') 129 130 def testLoadWithUNC(self): 131 import sys 132 if sys.platform != 'win32': 133 return 134 135 # Build a UNC path from the regular path. 136 # Something like 137 # \\%COMPUTERNAME%\c$\python27\python.exe 138 139 fullname = os.path.abspath(sys.executable) 140 if fullname[1] != ':': 141 return 142 unc_name = r'\\%s\%s$\%s' % (os.environ['COMPUTERNAME'], 143 fullname[0], 144 fullname[3:]) 145 146 with test_support.EnvironmentVarGuard() as env: 147 env.unset("TCL_LIBRARY") 148 f = os.popen('%s -c "import Tkinter; print Tkinter"' % (unc_name,)) 149 150 self.assertTrue('Tkinter.py' in f.read()) 151 # exit code must be zero 152 self.assertEqual(f.close(), None) 153 154 155 156 def test_main(): 157 test_support.run_unittest(TclTest, TkinterTest) 158 159 if __name__ == "__main__": 160 test_main() 161
