"""Handlers to automate managing test data."""

from flask import g
from owsresponse import response
from owsresponse.adaptors.flask import flaskify

from product_review.api import app
from product_review.testing_accommodations import art_relations, integration_test_utils


@app.route("/dbsync", methods=["GET"])
def sync_db():
    """Update product statuses to fix possible mismatches."""
    setattr(g.request_context, "is_db_resync", True)
    art_relations.sync_db()
    return flaskify(response.Response())


@app.route("/qa-only-reset-test-product/<int:product_id>", methods=["PUT"])
def qa_only_reset_test_product(product_id):
    """Reset db state relied on by tests."""
    result = integration_test_utils.reset_integration_test_db_records(product_id)
    return flaskify(response.Response(result))