import pytest

from label_studio.tests.sdk.common import LABEL_CONFIG_AND_TASKS

pytestmark = pytest.mark.django_db
from label_studio_sdk import Client


def test_annotation_create_and_update(django_live_url, business_client):
    ls = Client(url=django_live_url, api_key=business_client.api_key)
    p = ls.start_project(title='New Project', label_config=LABEL_CONFIG_AND_TASKS['label_config'])

    task_data = [{'data': {'my_text': 'Test task'}}]
    p.import_tasks(task_data)

    tasks = p.get_tasks()
    assert tasks[0]['data'] == task_data[0]['data']

    task_id = tasks[0]['id']
    annotation_data = {
        'result': [{'from_name': 'label', 'to_name': 'my_text', 'type': 'choices', 'value': {'choices': ['Positive']}}]
    }
    new_annotation = p.create_annotation(task_id, **annotation_data)
    assert (annotation_id := new_annotation['id'])
    assert new_annotation['result'] == annotation_data['result']

    p.update_annotation(
        annotation_id,
        result=[{'from_name': 'label', 'to_name': 'my_text', 'type': 'choices', 'value': {'choices': ['Negative']}}],
    )

    updated_annotation = p.get_tasks()[0]['annotations'][0]
    assert updated_annotation['result'][0]['value'] == {'choices': ['Negative']}


def test_annotation_marks_task_as_labeled(django_live_url, business_client):
    ls = Client(url=django_live_url, api_key=business_client.api_key)
    p = ls.start_project(
        title='New Project',
        label_config=LABEL_CONFIG_AND_TASKS['label_config'],
    )

    task_data = [{'data': {'my_text': 'Test task'}}, {'data': {'my_text': 'Test task 2'}}]
    p.import_tasks(task_data)
    tasks = p.get_tasks()

    assert p.get_labeled_tasks() == []

    task_id = tasks[0]['id']
    annotation_data = {
        'result': [{'from_name': 'label', 'to_name': 'my_text', 'type': 'choices', 'value': {'choices': ['Positive']}}]
    }
    p.create_annotation(task_id, **annotation_data)

    assert len(labeled_tasks := p.get_labeled_tasks()) == 1
    assert labeled_tasks[0]['data'] == task_data[0]['data']