Uploaded Test files
This commit is contained in:
parent
f584ad9d97
commit
2e81cb7d99
16627 changed files with 2065359 additions and 102444 deletions
203
venv/Lib/site-packages/win32/test/test_win32api.py
Normal file
203
venv/Lib/site-packages/win32/test/test_win32api.py
Normal file
|
|
@ -0,0 +1,203 @@
|
|||
# General test module for win32api - please add some :)
|
||||
|
||||
import unittest
|
||||
from pywin32_testutil import str2bytes
|
||||
|
||||
import win32api, win32con, win32event, winerror
|
||||
import sys, os
|
||||
import tempfile
|
||||
import datetime
|
||||
|
||||
class CurrentUserTestCase(unittest.TestCase):
|
||||
def testGetCurrentUser(self):
|
||||
name = "%s\\%s" % (win32api.GetDomainName(), win32api.GetUserName())
|
||||
self.failUnless(name == win32api.GetUserNameEx(win32api.NameSamCompatible))
|
||||
|
||||
class TestTime(unittest.TestCase):
|
||||
def testTimezone(self):
|
||||
# GetTimeZoneInformation
|
||||
rc, tzinfo = win32api.GetTimeZoneInformation()
|
||||
if rc == win32con.TIME_ZONE_ID_DAYLIGHT:
|
||||
tz_str = tzinfo[4]
|
||||
tz_time = tzinfo[5]
|
||||
else:
|
||||
tz_str = tzinfo[1]
|
||||
tz_time = tzinfo[2]
|
||||
# for the sake of code exercise but don't output
|
||||
tz_str.encode()
|
||||
if not isinstance(tz_time, datetime.datetime):
|
||||
tz_time.Format()
|
||||
def TestDateFormat(self):
|
||||
DATE_LONGDATE = 2
|
||||
date_flags = DATE_LONGDATE
|
||||
win32api.GetDateFormat(0, date_flags, None)
|
||||
win32api.GetDateFormat(0, date_flags, 0)
|
||||
win32api.GetDateFormat(0, date_flags, datetime.datetime.now())
|
||||
win32api.GetDateFormat(0, date_flags, time.time())
|
||||
def TestTimeFormat(self):
|
||||
win32api.GetTimeFormat(0, 0, None)
|
||||
win32api.GetTimeFormat(0, 0, 0)
|
||||
win32api.GetTimeFormat(0, 0, datetime.datetime.now())
|
||||
win32api.GetTimeFormat(0, 0, time.time())
|
||||
|
||||
|
||||
class Registry(unittest.TestCase):
|
||||
key_name = r'PythonTestHarness\Whatever'
|
||||
def test1(self):
|
||||
# This used to leave a stale exception behind.
|
||||
def reg_operation():
|
||||
hkey = win32api.RegCreateKey(win32con.HKEY_CURRENT_USER, self.key_name)
|
||||
x = 3/0 # or a statement like: raise 'error'
|
||||
# do the test
|
||||
try:
|
||||
try:
|
||||
try:
|
||||
reg_operation()
|
||||
except:
|
||||
1/0 # Force exception
|
||||
finally:
|
||||
win32api.RegDeleteKey(win32con.HKEY_CURRENT_USER, self.key_name)
|
||||
except ZeroDivisionError:
|
||||
pass
|
||||
|
||||
def testValues(self):
|
||||
key_name = r'PythonTestHarness\win32api'
|
||||
## tuples containing value name, value type, data
|
||||
values=(
|
||||
(None, win32con.REG_SZ, 'This is default unnamed value'),
|
||||
('REG_SZ', win32con.REG_SZ,'REG_SZ text data'),
|
||||
('REG_EXPAND_SZ', win32con.REG_EXPAND_SZ, '%systemdir%'),
|
||||
## REG_MULTI_SZ value needs to be a list since strings are returned as a list
|
||||
('REG_MULTI_SZ', win32con.REG_MULTI_SZ, ['string 1','string 2','string 3','string 4']),
|
||||
('REG_MULTI_SZ_empty', win32con.REG_MULTI_SZ, []),
|
||||
('REG_DWORD', win32con.REG_DWORD, 666),
|
||||
('REG_QWORD_INT', win32con.REG_QWORD, 99),
|
||||
('REG_QWORD', win32con.REG_QWORD, 2**33),
|
||||
('REG_BINARY', win32con.REG_BINARY, str2bytes('\x00\x01\x02\x03\x04\x05\x06\x07\x08\x01\x00')),
|
||||
)
|
||||
|
||||
hkey = win32api.RegCreateKey(win32con.HKEY_CURRENT_USER, key_name)
|
||||
for value_name, reg_type, data in values:
|
||||
win32api.RegSetValueEx(hkey, value_name, None, reg_type, data)
|
||||
|
||||
for value_name, orig_type, orig_data in values:
|
||||
data, typ=win32api.RegQueryValueEx(hkey, value_name)
|
||||
self.assertEqual(typ, orig_type)
|
||||
self.assertEqual(data, orig_data)
|
||||
|
||||
def testNotifyChange(self):
|
||||
def change():
|
||||
hkey = win32api.RegCreateKey(win32con.HKEY_CURRENT_USER, self.key_name)
|
||||
try:
|
||||
win32api.RegSetValue(hkey, None, win32con.REG_SZ, "foo")
|
||||
finally:
|
||||
win32api.RegDeleteKey(win32con.HKEY_CURRENT_USER, self.key_name)
|
||||
|
||||
evt = win32event.CreateEvent(None,0,0,None)
|
||||
## REG_NOTIFY_CHANGE_LAST_SET - values
|
||||
## REG_CHANGE_NOTIFY_NAME - keys
|
||||
## REG_NOTIFY_CHANGE_SECURITY - security descriptor
|
||||
## REG_NOTIFY_CHANGE_ATTRIBUTES
|
||||
win32api.RegNotifyChangeKeyValue(win32con.HKEY_CURRENT_USER,1,win32api.REG_NOTIFY_CHANGE_LAST_SET,evt,True)
|
||||
ret_code=win32event.WaitForSingleObject(evt,0)
|
||||
# Should be no change.
|
||||
self.failUnless(ret_code==win32con.WAIT_TIMEOUT)
|
||||
change()
|
||||
# Our event should now be in a signalled state.
|
||||
ret_code=win32event.WaitForSingleObject(evt,0)
|
||||
self.failUnless(ret_code==win32con.WAIT_OBJECT_0)
|
||||
|
||||
class FileNames(unittest.TestCase):
|
||||
def testShortLongPathNames(self):
|
||||
try:
|
||||
me = __file__
|
||||
except NameError:
|
||||
me = sys.argv[0]
|
||||
fname = os.path.abspath(me).lower()
|
||||
short_name = win32api.GetShortPathName(fname).lower()
|
||||
long_name = win32api.GetLongPathName(short_name).lower()
|
||||
self.failUnless(long_name==fname, \
|
||||
"Expected long name ('%s') to be original name ('%s')" % (long_name, fname))
|
||||
self.failUnlessEqual(long_name, win32api.GetLongPathNameW(short_name).lower())
|
||||
long_name = win32api.GetLongPathNameW(short_name).lower()
|
||||
self.failUnless(type(long_name)==str, "GetLongPathNameW returned type '%s'" % (type(long_name),))
|
||||
self.failUnless(long_name==fname, \
|
||||
"Expected long name ('%s') to be original name ('%s')" % (long_name, fname))
|
||||
|
||||
def testShortUnicodeNames(self):
|
||||
try:
|
||||
me = __file__
|
||||
except NameError:
|
||||
me = sys.argv[0]
|
||||
fname = os.path.abspath(me).lower()
|
||||
# passing unicode should cause GetShortPathNameW to be called.
|
||||
short_name = win32api.GetShortPathName(str(fname)).lower()
|
||||
self.failUnless(isinstance(short_name, str))
|
||||
long_name = win32api.GetLongPathName(short_name).lower()
|
||||
self.failUnless(long_name==fname, \
|
||||
"Expected long name ('%s') to be original name ('%s')" % (long_name, fname))
|
||||
self.failUnlessEqual(long_name, win32api.GetLongPathNameW(short_name).lower())
|
||||
long_name = win32api.GetLongPathNameW(short_name).lower()
|
||||
self.failUnless(type(long_name)==str, "GetLongPathNameW returned type '%s'" % (type(long_name),))
|
||||
self.failUnless(long_name==fname, \
|
||||
"Expected long name ('%s') to be original name ('%s')" % (long_name, fname))
|
||||
|
||||
def testLongLongPathNames(self):
|
||||
# We need filename where the FQN is > 256 - simplest way is to create a
|
||||
# 250 character directory in the cwd (except - cwd may be on a drive
|
||||
# not supporting \\\\?\\ (eg, network share) - so use temp.
|
||||
import win32file
|
||||
basename = "a" * 250
|
||||
# but we need to ensure we use the 'long' version of the
|
||||
# temp dir for later comparison.
|
||||
long_temp_dir = win32api.GetLongPathNameW(tempfile.gettempdir())
|
||||
fname = "\\\\?\\" + os.path.join(long_temp_dir, basename)
|
||||
try:
|
||||
win32file.CreateDirectoryW(fname, None)
|
||||
except win32api.error as details:
|
||||
if details.winerror!=winerror.ERROR_ALREADY_EXISTS:
|
||||
raise
|
||||
try:
|
||||
# GetFileAttributes automatically calls GetFileAttributesW when
|
||||
# passed unicode
|
||||
try:
|
||||
attr = win32api.GetFileAttributes(fname)
|
||||
except win32api.error as details:
|
||||
if details.winerror != winerror.ERROR_FILENAME_EXCED_RANGE:
|
||||
raise
|
||||
|
||||
attr = win32api.GetFileAttributes(str(fname))
|
||||
self.failUnless(attr & win32con.FILE_ATTRIBUTE_DIRECTORY, attr)
|
||||
|
||||
long_name = win32api.GetLongPathNameW(fname)
|
||||
self.failUnlessEqual(long_name.lower(), fname.lower())
|
||||
finally:
|
||||
win32file.RemoveDirectory(fname)
|
||||
|
||||
class FormatMessage(unittest.TestCase):
|
||||
def test_FromString(self):
|
||||
msg = "Hello %1, how are you %2?"
|
||||
inserts = ["Mark", "today"]
|
||||
result = win32api.FormatMessage(win32con.FORMAT_MESSAGE_FROM_STRING,
|
||||
msg, # source
|
||||
0, # ID
|
||||
0, # LangID
|
||||
inserts)
|
||||
self.assertEqual(result, "Hello Mark, how are you today?")
|
||||
|
||||
class Misc(unittest.TestCase):
|
||||
def test_last_error(self):
|
||||
for x in (0, 1, -1, winerror.TRUST_E_PROVIDER_UNKNOWN):
|
||||
win32api.SetLastError(x)
|
||||
self.failUnlessEqual(x, win32api.GetLastError())
|
||||
|
||||
def testVkKeyScan(self):
|
||||
# hopefully ' ' doesn't depend on the locale!
|
||||
self.failUnlessEqual(win32api.VkKeyScan(' '), 32)
|
||||
|
||||
def testVkKeyScanEx(self):
|
||||
# hopefully ' ' doesn't depend on the locale!
|
||||
self.failUnlessEqual(win32api.VkKeyScanEx(' ', 0), 32)
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Loading…
Add table
Add a link
Reference in a new issue