def test_all_data_html_report(self, list_dbs_mock, query_mock):
query_mock.return_value = {1: [2]}
list_dbs_mock.return_value = ["ns_serverEastdefault"]
params = {"snapshot": "all_data", "cluster": "East",
"hash": hash(time.time())}
request = self.factory.get("/reports/html/", params)
self.response = views.html_report(request)
expected = 'src="/media/EastEastdefaultdisk_write_queue.png"'
self.assertIn(expected, self.response.content)
map(lambda f: os.remove(f), glob.glob("webapp/media/*.png"))
def test_base_html_report(self, list_dbs_mock, query_mock):
query_mock.return_value = {1: [2]}
list_dbs_mock.return_value = ["ns_serverEastdefault"]
params = {"snapshot": "run-1_access-phase_vperf-reb_2.0.0-1976",
"hash": hash(time.time())}
request = self.factory.get("/reports/html/", params)
self.response = views.html_report(request)
expected = 'src="/media/run-1_access-phase_vperf-reb_2.0.0-1976Eastdefaultdisk_write_queue.png"'
self.assertIn(expected, self.response.content)
map(lambda f: os.remove(f), glob.glob("webapp/media/*.png"))
def test_not_existing_snapshot(self):
params = {"snapshot": "test"}
request = self.factory.get("/reports/html/", params)
response = views.html_report(request)
self.assertEqual(response.status_code, 400)
