From df50662b97ebfceef7ea5f800f3b7f79e27a172f Mon Sep 17 00:00:00 2001 From: virusdefender Date: Sat, 9 Dec 2017 22:42:54 +0800 Subject: [PATCH] fix tests --- conf/tests.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/conf/tests.py b/conf/tests.py index cd834446..a899a7c1 100644 --- a/conf/tests.py +++ b/conf/tests.py @@ -77,9 +77,10 @@ class JudgeServerHeartbeatTest(APITestCase): self.token = "test" self.hashed_token = hashlib.sha256(self.token.encode("utf-8")).hexdigest() 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): - 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) server = JudgeServer.objects.first() 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" data = self.data 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) server = JudgeServer.objects.first() self.assertEqual(server.service_url, service_url) @@ -98,7 +99,7 @@ class JudgeServerHeartbeatTest(APITestCase): self.test_new_heartbeat() data = self.data 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.assertEqual(JudgeServer.objects.get(hostname=self.data["hostname"]).judger_version, data["judger_version"])