import json

import pytest
from tests.utils import make_project


@pytest.mark.django_db
def test_has_lock(business_client):
    project = make_project({}, business_client.user, use_ml_backend=False)

    tasks = [
        {'data': {'location': 'London', 'text': 'text A'}},
    ]
    r = business_client.post(
        f'/api/projects/{project.id}/tasks/bulk', data=json.dumps(tasks), content_type='application/json'
    )
    assert r.status_code == 201

    task = project.tasks.first()

    annotation_data = {
        'task': task.id,
        'result': json.dumps(
            [{'from_name': 'text_class', 'to_name': 'text', 'value': {'labels': ['class_A'], 'start': 0, 'end': 1}}]
        ),
    }
    r = business_client.post('/api/tasks/{}/annotations/'.format(task.id), data=annotation_data)
    assert r.status_code == 201
    r = business_client.post('/api/tasks/{}/annotations/'.format(task.id), data=annotation_data)
    assert r.status_code == 201

    task.refresh_from_db()

    assert task.is_labeled is True

    task.is_labeled = False
    task.save()

    task.has_lock()
    task.refresh_from_db()

    assert task.is_labeled is True