"""Test Unsubmit logic."""

from product_review.api import db
from product_review.logic import unsubmit


def test_unsubmit_success(app_context, test_review_queue_models):
    """Test unsubmit success."""
    app_context.g.request_context.identity_id = "foo"
    db.session.add(test_review_queue_models[0])
    db.session.commit()

    result = unsubmit.unsubmit_product(test_review_queue_models[0].product_id, "bar")
    assert result.reason == "bar"
    assert result.user_id == "foo"
    assert result.review_queue_id == test_review_queue_models[0].review_queue_id

    test_review_queue_models[0].query.delete()


def test_unsubmit_not_found():
    """Test submit not found."""
    result = unsubmit.unsubmit_product(1, "foo")
    assert result.status == 404