mirror of
https://github.com/TurTaskProject/TurTaskWeb.git
synced 2025-12-19 14:04:07 +01:00
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
from rest_framework.test import APIClient
|
|
|
|
from users.models import CustomUser
|
|
from ..models import Task
|
|
|
|
|
|
def create_test_user(email="testusertestuser@example.com", username="testusertestuser",
|
|
first_name="Test", password="testpassword",):
|
|
"""create predifined user for testing"""
|
|
return CustomUser.objects.create_user(
|
|
email=email,
|
|
username=username,
|
|
first_name=first_name,
|
|
password=password,
|
|
)
|
|
|
|
|
|
def login_user(user):
|
|
"""Login a user to API client."""
|
|
|
|
client = APIClient()
|
|
client.force_authenticate(user=user)
|
|
return client
|
|
|
|
|
|
def create_task_json(user, **kwargs):
|
|
"""Create task JSON data to use with the API."""
|
|
defaults = {
|
|
"title": "Test Task",
|
|
"type": "habit",
|
|
"notes": "This is a test task created via the API.",
|
|
"exp": 10,
|
|
"priority": 1.5,
|
|
"difficulty": 1,
|
|
"attribute": "str",
|
|
"challenge": False,
|
|
"reminders": False,
|
|
"fromSystem": False,
|
|
"creation_date": None,
|
|
"last_update": None,
|
|
}
|
|
|
|
task_attributes = {**defaults, **kwargs}
|
|
task_attributes["user"] = user
|
|
|
|
return task_attributes
|
|
|
|
|
|
def create_test_task(user, **kwargs):
|
|
"""Create a test task and associate it with the given user."""
|
|
defaults = {
|
|
'title': "Test Task",
|
|
'task_type': 'habit',
|
|
'notes': "This is a test task created via the API.",
|
|
'exp': 10,
|
|
'priority': 1.5,
|
|
'difficulty': 1,
|
|
'attribute': 'str',
|
|
'challenge': False,
|
|
'reminders': False,
|
|
'fromSystem': False,
|
|
}
|
|
|
|
task_attributes = {**defaults, **kwargs}
|
|
|
|
return Task.objects.create(user=user, **task_attributes) |