Stop table reset of better persistent tables. (#373)

tests/conftest.py

@@ -183,6 +183,9 @@ def full_db(empty_db: Session, finance_statements: list[dict[str, Any]]) -> Sess
             for finance_statement in finance_statements
         ]
     )
+    empty_db.add(
+        entities.MissingCompany(name="Some company missing", zip_code="", city="")
+    )
     empty_db.commit()
     # print(pd.read_sql_table("company", empty_db.bind).to_string())
     return empty_db

tests/utils/sql/rest_sql_test.py

@@ -0,0 +1,38 @@
+"""Tests for sql rests."""
+import sys
+
+import pandas as pd
+import pytest
+from _pytest.monkeypatch import MonkeyPatch
+from sqlalchemy.orm import Session
+
+from aki_prj23_transparenzregister.utils.sql import entities, reset_sql
+
+
+def test_reset_sql_all(full_db: Session) -> None:
+    """Tests if all sql tables are reset."""
+    reset_sql.reset_tables(all_tables=True, db=full_db)
+    assert pd.read_sql_table(
+        entities.MissingCompany.__tablename__, con=full_db.bind  # type:ignore
+    ).empty
+    assert pd.read_sql_table(
+        entities.Company.__tablename__, con=full_db.bind  # type:ignore
+    ).empty
+
+
+def test_reset_sql(full_db: Session) -> None:
+    """Tests if only most sql tables are reset."""
+    reset_sql.reset_tables(all_tables=False, db=full_db)
+    assert pd.read_sql_table(
+        entities.Company.__tablename__, con=full_db.bind  # type:ignore
+    ).empty
+    assert not pd.read_sql_table(
+        entities.MissingCompany.__tablename__, con=full_db.bind  # type:ignore
+    ).empty
+
+
+def test_reset_help(monkeypatch: MonkeyPatch) -> None:
+    """Tests if all sql tables are reset."""
+    monkeypatch.setattr(sys, "argv", [sys.argv[0], "-h"])
+    with pytest.raises(SystemExit):
+        reset_sql.cli()