"""SQLAlchemy model to use in tests."""

from abacus_common_logic.models import base


class ModelForTests(base.BaseModel):
    """Test model class."""

    id = base.db.Column(base.db.INTEGER, primary_key=True)  # noqa
    name = base.db.Column(base.db.String(255), nullable=True)
    last_name = base.db.Column(base.db.String(255), nullable=True)


class SoftDeletableModel(ModelForTests, base.BaseSoftDeleteModel):
    """Test model class with soft delete."""

    ...


def create_test_object(**kwargs):
    """Create new row in model_for_tests table."""
    test_object = ModelForTests(**kwargs)
    base.db.session.add(test_object)
    base.db.session.commit()
    return test_object


def clear_test_model():
    """Delete all rows from model_for_tests."""
    base.db.session.query(ModelForTests).delete()
    base.db.session.query(SoftDeletableModel).delete()
    base.db.session.commit()