"""DB test utilities."""

from datetime import datetime, timezone

from product_review.api import db
from product_review.models.account_note import AccountNote
from product_review.models.approval import Approval
from product_review.models.canned_response import CannedResponse
from product_review.models.canned_response_category import CannedResponseCategory
from product_review.models.canned_response_history import CannedResponseHistory
from product_review.models.canned_response_note import CannedResponseNote
from product_review.models.move_target import MoveTarget
from product_review.models.queue_move_history import QueueMoveHistory
from product_review.models.rejection import Rejection
from product_review.models.review_queue import ReviewQueue
from product_review.models.review_validation_history import ReviewValidationHistory
from product_review.models.score_history import ScoreHistory
from product_review.models.subaccount_note import SubaccountNote
from product_review.models.target_group import TargetGroup
from product_review.models.unsubmit import Unsubmit


def create_test_review_queue_item(**kwargs):
    """Create new row in review_queue table."""
    test_object = ReviewQueue(**kwargs)
    test_object.created_datetime = test_object.created_datetime or datetime.now(
        timezone.utc
    )
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_unsubmit(**kwargs):
    """Create new row in unsubmit table."""
    test_object = Unsubmit(**kwargs)
    test_object.user_id = test_object.user_id or "abc"
    test_object.reason = test_object.reason or "reason"
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_queue_move_history_item(**kwargs):
    """Create new row in queue_move_history table."""
    test_object = QueueMoveHistory(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_approval(**kwargs):
    """Create new row in approval table."""
    test_object = Approval(**kwargs)
    test_object.created_datetime = test_object.created_datetime or datetime.now(
        timezone.utc
    )
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_rejection(**kwargs):
    """Create new row in rejection table."""
    test_object = Rejection(**kwargs)
    test_object.created_datetime = test_object.created_datetime or datetime.now(
        timezone.utc
    )
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_review_validation_history(**kwargs):
    """Create new row in review_validation_history table."""
    test_object = ReviewValidationHistory(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_move_target(**kwargs):
    """Create new row in move_target table."""
    test_object = MoveTarget(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_target_group(**kwargs):
    """Create new row in target_group table."""
    test_object = TargetGroup(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_canned_response_item(**kwargs):
    """Create new row in canned_response table."""
    test_object = CannedResponse(**kwargs)
    test_object.last_updated = test_object.last_updated or datetime.now(timezone.utc)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_canned_response_note(**kwargs):
    """Create new row in canned_response_note table."""
    test_object = CannedResponseNote(**kwargs)
    test_object.last_updated = test_object.last_updated or datetime.now(timezone.utc)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_canned_response_category(**kwargs):
    """Create new row in canned_response_category table."""
    test_object = CannedResponseCategory(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_canned_response_history(**kwargs):
    """Create new row in canned_response_history table."""
    test_object = CannedResponseHistory(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_account_note(**kwargs):
    """Create new row in account_note table."""
    test_object = AccountNote(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_subaccount_note(**kwargs):
    """Create new row in subaccount_note table."""
    test_object = SubaccountNote(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object


def create_test_score_history_item(**kwargs):
    """Create new row in score_history table."""
    test_object = ScoreHistory(**kwargs)
    db.session.add(test_object)
    db.session.commit()
    return test_object