def clear_db(): models.clear_db() data = {} data['message'] = 'Database has been cleared' data['return_url'] = url_for('data_index') data['return_url_text'] = 'Back to Data' return render_template('message.html',**data)
def test_view(self): """ test getting one server """ clear_db() s1 = Server.create(server_id=None, tenant_id=1, name="name1", status="ready") Server.create(server_id=None, tenant_id=1, name="name2", status="scheduled") Server.create(server_id=None, tenant_id=2, name="name3", status="ready") request = HTTPRequest( self.get_url("/1/servers/" + str(s1.server_id)), method="GET", headers=HTTPHeaders({"X-AUTH-TOKEN": X_AUTH_TOKEN}), ) self.http_client.fetch(request, self.stop) response = self.wait() message = json.loads(response.body) assert message == { "status": s1.status, "id": s1.server_id, "name": s1.name, "date_created": str(s1.date_created), }
def test_create_server(self): """ try to create server with too short name """ clear_db() server_name = "server1" request = HTTPRequest( self.get_url("/1/servers/create"), method="POST", headers=HTTPHeaders({"X-AUTH-TOKEN": X_AUTH_TOKEN}), body="name=%s" % server_name, ) self.http_client.fetch(request, self.stop) response = self.wait() message = json.loads(response.body) assert message["status"] == "pending" assert message["name"] == "server1" server_id = message["id"] sleep(1) for status in ("scheduled", "processing", "ready"): sleep(10) request = HTTPRequest( self.get_url("/1/servers/" + str(server_id)), method="GET", headers=HTTPHeaders({"X-AUTH-TOKEN": X_AUTH_TOKEN}), ) self.http_client.fetch(request, self.stop) response = self.wait() message = json.loads(response.body) assert message["status"] == status
def test_delete_not_found(self): """ try to delete non existing server """ clear_db() request = HTTPRequest( self.get_url("/1/servers/1"), method="DELETE", headers=HTTPHeaders({"X-AUTH-TOKEN": X_AUTH_TOKEN}) ) self.http_client.fetch(request, self.stop) response = self.wait() message = json.loads(response.body) assert message == {"status": "not_found"}
def sample_data(): models.clear_db() with open('data/authors.txt','r') as f: author_data = f.read() cnt = models.load_author_data(author_data) with open('data/students.txt','r') as f: student_data = f.read() cnt = models.load_student_data(student_data) data = {} data['message'] = 'Sample data loaded.' data['return_url'] = url_for('index') data['return_url_text'] = 'Return to Home.' return render_template('message.html',**data)
def test_delete(self): """ delete existing server """ clear_db() s1 = Server.create(server_id=None, tenant_id=1, name="name1", status="ready") request = HTTPRequest( self.get_url("/1/servers/" + str(s1.server_id)), method="DELETE", headers=HTTPHeaders({"X-AUTH-TOKEN": X_AUTH_TOKEN}), ) self.http_client.fetch(request, self.stop) response = self.wait() message = json.loads(response.body) assert message == {"status": "deleted", "id": s1.server_id}
def test_create_invalid_name(self): """ try to create server with too short name """ clear_db() request = HTTPRequest( self.get_url("/1/servers/create"), method="POST", headers=HTTPHeaders({"X-AUTH-TOKEN": X_AUTH_TOKEN}), body="name=test", ) self.http_client.fetch(request, self.stop) response = self.wait() message = json.loads(response.body) assert message == {"status": "invalid_name"}
def test_list(self): """ list of servers """ clear_db() s1 = Server.create(server_id=None, tenant_id=1, name="name1", status="ready") s2 = Server.create(server_id=None, tenant_id=1, name="name2", status="scheduled") Server.create(server_id=None, tenant_id=2, name="name3", status="ready") request = HTTPRequest( self.get_url("/1/servers/"), method="GET", headers=HTTPHeaders({"X-AUTH-TOKEN": X_AUTH_TOKEN}) ) self.http_client.fetch(request, self.stop) response = self.wait() message = json.loads(response.body) assert message == { "servers": [ {"status": s1.status, "id": s1.server_id, "name": s1.name}, {"status": s2.status, "id": s2.server_id, "name": s2.name}, ] }
def refresh_db(): download_history() clear_db() populate_db()