xuyue/OnlineJudge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
virusdefender
2017-12-09 22:42:54 +08:00
parent c4889d7085
commit df50662b97
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
conf/tests.py
View File

@@ -77,9 +77,10 @@ class JudgeServerHeartbeatTest(APITestCase):
self.token = "test" self.token = "test"
self.hashed_token = hashlib.sha256(self.token.encode("utf-8")).hexdigest() self.hashed_token = hashlib.sha256(self.token.encode("utf-8")).hexdigest()
SysOptions.judge_server_token = self.token SysOptions.judge_server_token = self.token
self.headers = {"HTTP_X_JUDGE_SERVER_TOKEN": self.hashed_token, "HTTP_X_REAL_IP": "1.2.3.4"}
def test_new_heartbeat(self): def test_new_heartbeat(self):
resp = self.client.post(self.url, data=self.data, **{"HTTP_X_JUDGE_SERVER_TOKEN": self.hashed_token}) resp = self.client.post(self.url, data=self.data, **self.headers)
self.assertSuccess(resp) self.assertSuccess(resp)
server = JudgeServer.objects.first() server = JudgeServer.objects.first()
self.assertEqual(server.ip, "127.0.0.1") self.assertEqual(server.ip, "127.0.0.1")
@@ -89,7 +90,7 @@ class JudgeServerHeartbeatTest(APITestCase):
service_url = "http://1.2.3.4:8000/api/judge" service_url = "http://1.2.3.4:8000/api/judge"
data = self.data data = self.data
data["service_url"] = service_url data["service_url"] = service_url
resp = self.client.post(self.url, data=self.data, **{"HTTP_X_JUDGE_SERVER_TOKEN": self.hashed_token}) resp = self.client.post(self.url, data=self.data, **self.headers)
self.assertSuccess(resp) self.assertSuccess(resp)
server = JudgeServer.objects.first() server = JudgeServer.objects.first()
self.assertEqual(server.service_url, service_url) self.assertEqual(server.service_url, service_url)
@@ -98,7 +99,7 @@ class JudgeServerHeartbeatTest(APITestCase):
self.test_new_heartbeat() self.test_new_heartbeat()
data = self.data data = self.data
data["judger_version"] = "2.0.0" data["judger_version"] = "2.0.0"
resp = self.client.post(self.url, data=data, **{"HTTP_X_JUDGE_SERVER_TOKEN": self.hashed_token}) resp = self.client.post(self.url, data=data, **self.headers)
self.assertSuccess(resp) self.assertSuccess(resp)
self.assertEqual(JudgeServer.objects.get(hostname=self.data["hostname"]).judger_version, data["judger_version"]) self.assertEqual(JudgeServer.objects.get(hostname=self.data["hostname"]).judger_version, data["judger_version"])