def get(self, requests):
try:
filePath = requests.GET['filePath']
if not filePath:
raise Exception("Invalid Request Parameter")
except:
return JsonResponse({'error': 'Invalid Request Parameter'},
status=400)
return Utility.downloadPMML(filePath)
def test_3_downloadPMMLFile(self):
logging.info("Test Case : Test download file for pmml file.")
with open('testUseCase/supportdata/test.pmml', 'w') as ff:
ff.write('<?xml version="1.0" encoding="UTF-8"?>')
filePath = os.path.abspath('testUseCase/supportdata/test.pmml')
result = Utility.downloadPMML(filePath)
self.assertEqual(result.status_code, 200)
self.assertEqual(result.__dict__['_container'][0].decode('utf-8'),
'<?xml version="1.0" encoding="UTF-8"?>')
self.assertEqual(result.__dict__['_headers']['content-type'][1],
'application/xml')
os.remove('testUseCase/supportdata/test.pmml')
logging.info("PASSED")
def test_2_downloadCSVFile(self):
logging.info("Test Case : Test download file for csv file.")
with open('testUseCase/supportdata/test.csv', 'w') as ff:
ff.write('This,is,a,Test,File')
filePath = os.path.abspath('testUseCase/supportdata/test.csv')
result = Utility.downloadPMML(filePath)
self.assertEqual(result.status_code, 200)
self.assertEqual(result.__dict__['_container'][0].decode('utf-8'),
'This,is,a,Test,File')
self.assertEqual(result.__dict__['_headers']['content-type'][1],
'text/csv')
os.remove('testUseCase/supportdata/test.csv')
logging.info("PASSED")
def test_1_downloadJSONFile(self):
logging.info("Test Case : Test download file for json file.")
content = {'content': 'This is a Test File'}
with open('testUseCase/supportdata/test.json', 'w') as ff:
ff.write(json.dumps(content))
filePath = os.path.abspath('testUseCase/supportdata/test.json')
result = Utility.downloadPMML(filePath)
self.assertEqual(result.status_code, 200)
self.assertEqual(
json.loads(result.__dict__['_container'][0])['content'],
'This is a Test File')
self.assertEqual(result.__dict__['_headers']['content-type'][1],
'application/json')
os.remove('testUseCase/supportdata/test.json')
logging.info("PASSED")