Archived
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
# -- coding:utf-8 --
|
|
# @Author: markushammered@gmail.com
|
|
# @Development Tool: Pycharm
|
|
# @Create Time: 2022/5/2
|
|
# @File Name: test_auth.py
|
|
|
|
|
|
import random
|
|
import requests
|
|
import unittest
|
|
|
|
|
|
class TestAuth(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.api = 'http://127.0.0.1:5000/api/v1/'
|
|
|
|
@property
|
|
def r_pwd(self):
|
|
letters = 'abcdefghijklmnopqrstuvwxyz'
|
|
r_pwd = ''.join(random.choice(letters) for _ in range(16))
|
|
return r_pwd
|
|
|
|
def test_api_system_header(self):
|
|
res = requests.post(self.api + 'system', headers={'Authorization': 'bearer unittest'})
|
|
self.assertEqual(res.status_code, 200)
|
|
|
|
def test_api_system_query(self):
|
|
res = requests.post(self.api + 'system', params={'key': 'unittest'})
|
|
self.assertEqual(res.status_code, 200)
|
|
|
|
def test_api_system_header_failed(self):
|
|
res = requests.post(self.api + 'system', headers={'Authorization': f'bearer {self.r_pwd}'})
|
|
self.assertEqual(res.status_code, 403)
|
|
|
|
def test_api_system_query_failed(self):
|
|
res = requests.post(self.api + 'system', params={'key': self.r_pwd})
|
|
self.assertEqual(res.status_code, 403)
|