48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
# coding=utf-8
|
|
from __future__ import unicode_literals
|
|
from os import sys, path
|
|
sys.path.append(path.dirname(path.dirname(path.abspath(__file__))))
|
|
|
|
import json
|
|
import unittest
|
|
import requests
|
|
|
|
|
|
from client.Python.client import JudgeServerClient, JudgeServerClientError
|
|
|
|
|
|
class JudgeServerClientForTokenHeaderTest(JudgeServerClient):
|
|
def _request(self, url, data=None):
|
|
kwargs = {"headers": {"Content-Type": "application/json"}}
|
|
if data:
|
|
kwargs["data"] = json.dumps(data)
|
|
try:
|
|
return requests.post(url, **kwargs).json()
|
|
except Exception as e:
|
|
raise JudgeServerClientError(e.message)
|
|
|
|
|
|
class JudgeServerTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.token = "YOUR_TOKEN_HERE"
|
|
self.server_base_url = "http://127.0.0.1:12358"
|
|
self.client = JudgeServerClient(token=self.token, server_base_url=self.server_base_url)
|
|
|
|
def test_success(self):
|
|
data = self.client.ping()
|
|
self.assertEqual(data["err"], None)
|
|
self.assertEqual(data["data"]["action"], "pong")
|
|
|
|
def test_invalid_token(self):
|
|
client = JudgeServerClient(token="wrong token", server_base_url=self.server_base_url)
|
|
data = client.ping()
|
|
self.assertEqual(data["err"], "TokenVerificationFailed")
|
|
|
|
def test_no_token_header(self):
|
|
client = JudgeServerClientForTokenHeaderTest(token="wrong token", server_base_url=self.server_base_url)
|
|
data = client.ping()
|
|
self.assertEqual(data["err"], "TokenVerificationFailed")
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|