"""Test submission count."""

from tests.utils.db_utils import create_test_rejection, create_test_review_queue_item


def test_submission_count_re_submission(request_engine, fixture_client):
    """Test re_submission submission count."""
    request_engine["ows-product"].add_spec(
        "GET",
        "/product/9",
        status=200,
        response={"status": "transfer_to_content"},
    )
    create_test_review_queue_item(review_queue_id=111, product_id=9, status="complete")
    create_test_rejection(review_queue_id=111, user_id="333", note="foo")
    create_test_review_queue_item(review_queue_id=112, product_id=9, status="complete")
    create_test_review_queue_item(review_queue_id=113, product_id=9, status="complete")
    create_test_rejection(review_queue_id=113, user_id="333", note="foo")
    result = fixture_client.post("/review-queue/product/9")
    assert result.status_code == 201
    assert result.json.get("submission_count") == 3


def test_submission_count_revision(request_engine, fixture_client):
    """Test revision submission count."""
    request_engine["ows-product"].add_spec(
        "GET",
        "/product/9",
        status=200,
        response={"status": "in_content"},
    )
    request_engine["ows-product-workflow"].add_spec(
        "GET",
        "/product/9/correction",
        status=200,
        response={"release_correction_id": 3, "status": "submitted"},
    )
    create_test_review_queue_item(
        review_queue_id=111, product_id=9, error_correction_id=2, status="complete"
    )
    create_test_rejection(review_queue_id=111, user_id="333", note="foo")
    create_test_review_queue_item(
        review_queue_id=112, product_id=9, error_correction_id=3, status="complete"
    )
    create_test_review_queue_item(
        review_queue_id=113, product_id=9, error_correction_id=3, status="complete"
    )
    create_test_rejection(review_queue_id=113, user_id="333", note="foo")
    result = fixture_client.post("/review-queue/product/9")
    assert result.status_code == 201
    assert result.json.get("submission_count") == 2