Skip to content
Snippets Groups Projects
Commit c2d0ef52 authored by Jedrzej Rybicki's avatar Jedrzej Rybicki
Browse files

pytest compatible testing

parent 322f1ad8
Branches
Tags
No related merge requests found
Pipeline #69392 passed
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from apiserver.main import app, my_user, my_auth from context import apiserver, storage
#from apiserver import app, my_user, my_auth
from unittest import TestCase from unittest import TestCase
from apiserver.security.user import User from apiserver.security.user import User
...@@ -9,26 +11,25 @@ def myfunc(): ...@@ -9,26 +11,25 @@ def myfunc():
class UserTests(TestCase): class UserTests(TestCase):
def setUp(self): def setUp(self):
self.client = TestClient(app) apiserver.app.dependency_overrides[apiserver.main.my_auth] = myfunc
app.dependency_overrides={} apiserver.app.dependency_overrides[apiserver.main.my_user] = myfunc
self.client = TestClient(apiserver.app)
def tearDown(self): def tearDown(self):
app.dependency_overrides={} apiserver.app.dependency_overrides={}
def test_me(self): def test_me(self):
app.dependency_overrides[my_user] = myfunc
resp = self.client.get('/me') resp = self.client.get('/me')
self.assertEquals(resp.json(), {'username': 'foo', 'email': 'bar'}) self.assertEqual(resp.json(), {'username': 'foo', 'email': 'bar'})
def test_token(self): def test_token(self):
app.dependency_overrides[my_auth] = myfunc
rsp = self.client.post('/token').json() rsp = self.client.post('/token').json()
self.assertEqual(rsp["token_type"], "bearer") self.assertEqual(rsp["token_type"], "bearer")
self.assertTrue('access_token' in rsp) self.assertTrue('access_token' in rsp)
def test_create(self): def test_create(self):
app.dependency_overrides[my_user] = myfunc
my_data = { my_data = {
'name': 'some dataset', 'name': 'some dataset',
'url': 'http://loc.me/1', 'url': 'http://loc.me/1',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment