def testBackOff(self):
# no support for ``with assertRaises`` in python 2.6
if sys.version_info < (2,7): return
def mypost(url, data, headers):
self.assertEqual(url, 'h://c:4/p')
self.assertEqual(data.decode('utf-8'), """\
<?xml version='1.0' encoding='utf-8'?>
<galileo-client version="2.0"><client-info><client-id>%(id)s</client-id><client-version>%(version)s</client-version><client-mode>status</client-mode></client-info></galileo-client>""" % {
'id': RemoteXMLDatabase.ID,
'version': __version__})
self.assertEqual(headers['Content-Type'], 'text/xml')
return requestResponse("""
<back-off>
<min>1800000</min>
<max>3600000</max>
</back-off>
<ui-request action="login">
<client-display height="450" width="650" minDisplayTimeMs="20000" containsForm="false">
Server is in maintenance mode. We'll be back soon!
</client-display>
</ui-request>""", '')
mod.requests.post = mypost
gc = RemoteXMLDatabase('h', 'c', 'p', 4)
with self.assertRaises(BackOffException) as cm:
gc.requestStatus()
e = cm.exception
self.assertEqual(e.min, 1800000)
self.assertEqual(e.max, 3600000)
val = e.getAValue()
self.assertTrue(e.min <= val <= e.max)
def testOk(self):
def mypost(url, data, headers):
self.assertEqual(url, 'scheme://host:8888/path/to/stuff')
self.assertEqual(data.decode('utf-8'), """\
<?xml version='1.0' encoding='utf-8'?>
<galileo-client version="2.0"><client-info><client-id>%(id)s</client-id><client-version>%(version)s</client-version><client-mode>status</client-mode></client-info></galileo-client>""" % {
'id': RemoteXMLDatabase.ID, 'version': __version__})
self.assertEqual(headers['Content-Type'], 'text/xml')
return requestResponse('')
mod.requests.post = mypost
gc = RemoteXMLDatabase('scheme', 'host', 'path/to/stuff', 8888)
gc.requestStatus()
def testHTTPError(self): # issue147
def mypost(url, data, headers):
class Response(object): status_code=500
if mod.requests.__build__ > 0x020000:
# Only newer requests exceptions inherit from IOError
e = mod.requests.exceptions.HTTPError('bad', response=Response())
else:
# older inherit from RuntimeError (no kwargs)
e = mod.requests.exceptions.HTTPError('bad')
raise e
mod.requests.post = mypost
T_ID = 'abcd'
D = MyDongle(0, 0)
d = MyMegaDump('YWJjZA==')
gc = RemoteXMLDatabase('a', 'b', 'c', 0)
with self.assertRaises(SyncError) as cm:
gc.sync(D, T_ID, d)
self.assertEqual(cm.exception.errorstring, 'HTTPError: bad (500)')
def testConnectionError(self):
T_ID = 'abcd'
D = MyDongle(0, 0)
d = MyMegaDump('YWJjZA==')
def mypost(url, data, headers):
class Reason(object):
class Error(object): strerror = ''
reason = Error()
raise mod.requests.exceptions.ConnectionError(Reason())
mod.requests.post = mypost
gc = RemoteXMLDatabase('a', 'b', 'c', 0)
self.assertRaises(SyncError, gc.sync,D, T_ID, d)
def testOk(self):
T_ID = 'abcd'
D = MyDongle(0, 0)
d = MyMegaDump('YWJjZA==')
def mypost(url, data, headers):
self.assertEqual(url, 'a://b:0/c')
self.assertEqual(data.decode('utf-8'), """\
<?xml version='1.0' encoding='utf-8'?>
<galileo-client version="2.0"><client-info><client-id>%(id)s</client-id><client-version>%(version)s</client-version><client-mode>sync</client-mode><dongle-version major="%(M)d" minor="%(m)d" /></client-info><tracker tracker-id="%(t_id)s"><data>%(b64dump)s</data></tracker></galileo-client>""" % {
'id': RemoteXMLDatabase.ID,
'version': __version__,
'M': D.major,
'm': D.minor,
't_id': T_ID,
'b64dump': d.toBase64()})
self.assertEqual(headers['Content-Type'], 'text/xml')
return requestResponse('<tracker tracker-id="abcd" type="megadumpresponse"><data>ZWZnaA==</data></tracker>')
mod.requests.post = mypost
gc = RemoteXMLDatabase('a', 'b', 'c', 0)
self.assertEqual(gc.sync(D, T_ID, d),
[101, 102, 103, 104])
def testError(self):
def mypost(url, data, headers):
self.assertEqual(url, 'h://c:8/p')
self.assertEqual(data.decode('utf-8'), """\
<?xml version='1.0' encoding='utf-8'?>
<galileo-client version="2.0"><client-info><client-id>%(id)s</client-id><client-version>%(version)s</client-version><client-mode>status</client-mode></client-info></galileo-client>""" % {
'id': RemoteXMLDatabase.ID,
'version': __version__})
self.assertEqual(headers['Content-Type'], 'text/xml')
return requestResponse('<error>Something is Wrong</error>')
mod.requests.post = mypost
gc = RemoteXMLDatabase('h', 'c', 'p', 8)
self.assertRaises(SyncError, gc.requestStatus)
def testNotData(self):
T_ID = 'aaaabbbbccccdddd'
D = MyDongle(-2, 42)
d = MyMegaDump('base64Dump')
def mypost(url, data, headers):
self.assertEqual(url, 'rsync://ssh:22/a/b/c')
self.assertEqual(data.decode('utf-8'), """\
<?xml version='1.0' encoding='utf-8'?>
<galileo-client version="2.0"><client-info><client-id>%(id)s</client-id><client-version>%(version)s</client-version><client-mode>sync</client-mode><dongle-version major="%(M)d" minor="%(m)d" /></client-info><tracker tracker-id="%(t_id)s"><data>%(b64dump)s</data></tracker></galileo-client>""" % {
'id': RemoteXMLDatabase.ID,
'version': __version__,
'M': D.major,
'm': D.minor,
't_id': T_ID,
'b64dump': d.toBase64()})
self.assertEqual(headers['Content-Type'], 'text/xml')
return requestResponse('<tracker tracker-id="abcd" type="megadumpresponse"><not_data /></tracker>')
mod.requests.post = mypost
gc = RemoteXMLDatabase('rsync', 'ssh', 'a/b/c', 22)
self.assertRaises(SyncError, gc.sync, D, T_ID, d)
def testNoTracker(self):
T_ID = 'aaaabbbb'
D = MyDongle(34, 88)
d = MyMegaDump('base64Dump')
def mypost(url, data, headers):
self.assertEqual(url, 'z://y:42/u')
self.assertEqual(data.decode('utf-8'), """\
<?xml version='1.0' encoding='utf-8'?>
<galileo-client version="2.0"><client-info><client-id>%(id)s</client-id><client-version>%(version)s</client-version><client-mode>sync</client-mode><dongle-version major="%(M)d" minor="%(m)d" /></client-info><tracker tracker-id="%(t_id)s"><data>%(b64dump)s</data></tracker></galileo-client>""" % {
'id': RemoteXMLDatabase.ID,
'version': __version__,
'M': D.major,
'm': D.minor,
't_id': T_ID,
'b64dump': d.toBase64()})
self.assertEqual(headers['Content-Type'], 'text/xml')
return requestResponse('')
mod.requests.post = mypost
gc = RemoteXMLDatabase('z', 'y', 'u', 42)
self.assertRaises(SyncError, gc.sync, D, T_ID, d)
def testUnknownPort(self):
# no support for ``with assertRaises`` in python 2.6
if sys.version_info < (2,7): return
gc = RemoteXMLDatabase('blah', 'h', 'a')
with self.assertRaises(KeyError):
gc.url
def testHTTPSPort(self):
gc = RemoteXMLDatabase('https', 'h', 'a/b/c')
self.assertEqual(gc.url, 'https://h:443/a/b/c')
def testHTTPPort(self):
gc = RemoteXMLDatabase('http', 'h', 'a/b/c')
self.assertEqual(gc.url, 'http://h:80/a/b/c')
def testWithPort(self):
gc = RemoteXMLDatabase('scheme', 'host', 'path/to/stuff', 8000)
self.assertEqual(gc.url, 'scheme://host:8000/path/to/stuff')