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()
