-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathoverload_unittest.py
More file actions
43 lines (34 loc) · 1.14 KB
/
overload_unittest.py
File metadata and controls
43 lines (34 loc) · 1.14 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#!/urs/bin/env python3
#import requests
import unittest
import argparse
import hashlib
import sys
import os
def fetchRiakFile(url):
os.system('curl -k -o remoteRiakFile %s' % url)
md5Remote = hashlib.md5(open('remoteRiakFile', 'rb').read()).hexdigest()
return md5Remote
def fetchLocalFile(riakFile):
md5Local = hashlib.md5(open(riakFile, 'rb').read()).hexdigest()
return md5Local
class TestOne(unittest.TestCase):
def __init__(self, md5remote, md5local):
super(TestOne, self).__init__('testone')
self.md5remote = md5remote
self.md5local = md5local
def testone(self):
self.assertEqual(self.md5remote, self.md5local)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('-u', '--url', required=True)
parser.add_argument('-f', '--localFile', required=True)
args = parser.parse_args()
suite = unittest.TestSuite()
try:
md5remote = fetchRiakFile(args.url)
md5local = fetchLocalFile(args.localFile)
except Exception as e:
print(e)
suite.addTest(TestOne(md5remote, md5local))
unittest.TextTestRunner().run(suite)