kenneth.skovhede@gmail.com
20a6757d56
git-svn-id: https://duplicati.googlecode.com/svn/trunk@11 59da171f-624f-0410-aa54-27559c288bec
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from win32inet import *
|
|
from win32inetcon import *
|
|
|
|
import unittest
|
|
|
|
class CookieTests(unittest.TestCase):
|
|
def testCookies(self):
|
|
data = "TestData=Test"
|
|
InternetSetCookie("http://www.python.org", None, data)
|
|
got = InternetGetCookie("http://www.python.org", None)
|
|
self.assertEqual(got, data)
|
|
|
|
class UrlTests(unittest.TestCase):
|
|
def testSimpleCanonicalize(self):
|
|
ret = InternetCanonicalizeUrl("foo bar")
|
|
self.assertEqual(ret, "foo%20bar")
|
|
|
|
def testLongCanonicalize(self):
|
|
# a 4k URL causes the underlying API to request a bigger buffer"
|
|
big = "x" * 2048
|
|
ret = InternetCanonicalizeUrl(big + " " + big)
|
|
self.assertEqual(ret, big + "%20" + big)
|
|
|
|
class TestNetwork(unittest.TestCase):
|
|
def setUp(self):
|
|
self.hi = InternetOpen("test", INTERNET_OPEN_TYPE_DIRECT, None, None, 0)
|
|
def tearDown(self):
|
|
self.hi.Close()
|
|
def testPythonDotOrg(self):
|
|
hdl = InternetOpenUrl(self.hi, "http://www.python.org", None,
|
|
INTERNET_FLAG_EXISTING_CONNECT)
|
|
chunks = []
|
|
while 1:
|
|
chunk = InternetReadFile(hdl, 1024)
|
|
if not chunk:
|
|
break
|
|
chunks.append(chunk)
|
|
data = ''.join(chunks)
|
|
assert data.find("Python")>0, repr(data) # This must appear somewhere on the main page!
|
|
|
|
def testFtpCommand(self):
|
|
hcon = InternetConnect(self.hi, "ftp.python.org", INTERNET_INVALID_PORT_NUMBER,
|
|
None, None, # username/password
|
|
INTERNET_SERVICE_FTP, 0, 0)
|
|
try:
|
|
try:
|
|
hftp = FtpCommand(hcon, True, FTP_TRANSFER_TYPE_ASCII, 'NLST', 0)
|
|
except error:
|
|
print "Error info is", InternetGetLastResponseInfo()
|
|
InternetReadFile(hftp, 2048)
|
|
hftp.Close()
|
|
finally:
|
|
hcon.Close()
|
|
|
|
if __name__=='__main__':
|
|
unittest.main()
|