mirror of
https://github.com/makeplane/plane
synced 2024-06-14 14:31:34 +00:00
35 lines
913 B
Python
35 lines
913 B
Python
# Third party imports
|
|
from rest_framework.test import APITestCase, APIClient
|
|
|
|
# Module imports
|
|
from plane.db.models import User
|
|
from plane.api.views.authentication import get_tokens_for_user
|
|
|
|
|
|
class BaseAPITest(APITestCase):
|
|
def setUp(self):
|
|
self.client = APIClient(HTTP_USER_AGENT="plane/test", REMOTE_ADDR="10.10.10.10")
|
|
|
|
|
|
class AuthenticatedAPITest(BaseAPITest):
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
## Create Dummy User
|
|
self.email = "user@plane.so"
|
|
user = User.objects.create(email=self.email)
|
|
user.set_password("user@123")
|
|
user.save()
|
|
|
|
# Set user
|
|
self.user = user
|
|
|
|
# Set Up User ID
|
|
self.user_id = user.id
|
|
|
|
access_token, _ = get_tokens_for_user(user)
|
|
self.access_token = access_token
|
|
|
|
# Set Up Authentication Token
|
|
self.client.credentials(HTTP_AUTHORIZATION="Bearer " + access_token)
|