48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from datetime import timedelta
|
|
|
|
import pytest
|
|
from django.utils import timezone
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
from apps.time_entries.services.time_entries import create_time_entry, stop_time_entry
|
|
from apps.users.models import User
|
|
from apps.workspaces.models import Workspace
|
|
|
|
|
|
@pytest.fixture
|
|
def workspace_owner(db):
|
|
user = User.objects.create_user(mobile="09121111111", password="secret123")
|
|
workspace = Workspace.objects.create(name="Core", owner=user)
|
|
return user, workspace
|
|
|
|
|
|
def test_create_time_entry_allows_only_one_running_timer_per_workspace(workspace_owner):
|
|
user, workspace = workspace_owner
|
|
|
|
create_time_entry(
|
|
user=user,
|
|
workspace_id=workspace.id,
|
|
start_time=timezone.now(),
|
|
)
|
|
|
|
with pytest.raises(ValidationError):
|
|
create_time_entry(
|
|
user=user,
|
|
workspace_id=workspace.id,
|
|
start_time=timezone.now() + timedelta(minutes=5),
|
|
)
|
|
|
|
|
|
def test_stop_time_entry_sets_end_time_and_duration(workspace_owner):
|
|
user, workspace = workspace_owner
|
|
entry = create_time_entry(
|
|
user=user,
|
|
workspace_id=workspace.id,
|
|
start_time=timezone.now() - timedelta(hours=1),
|
|
)
|
|
|
|
stopped_entry = stop_time_entry(entry, end_time=timezone.now())
|
|
|
|
assert stopped_entry.end_time is not None
|
|
assert stopped_entry.duration is not None
|