test: add unit tests for utils, custom_types, clean_sql, date_service

Adds 97 new unit tests covering pure-logic modules that had zero coverage:
- testgen/utils: to_int, str_to_timestamp, is_uuid4, try_json, make_json_safe,
  chunk_queries, score, friendly_score, to_dataframe, log_and_swallow_exception
- common/models/custom_types: YNString, NullIfEmptyString, ZeroIfEmptyInteger,
  UpdateTimestamp, EncryptedBytea/EncryptedJson roundtrip
- common/clean_sql: CleanSQL comment/whitespace normalization, concat_columns
- common/date_service: as_iso_timestamp, get_now_as_iso_timestamp

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: rboni-dk

tests/unit/common/models/test_custom_types.py

@@ -0,0 +1,132 @@
+from datetime import UTC, datetime
+from unittest.mock import patch
+
+import pytest
+
+from testgen.common.models.custom_types import (
+    EncryptedBytea,
+    EncryptedJson,
+    NullIfEmptyString,
+    UpdateTimestamp,
+    YNString,
+    ZeroIfEmptyInteger,
+)
+
+pytestmark = pytest.mark.unit
+
+
+# --- NullIfEmptyString ---
+
+@pytest.mark.parametrize(
+    "value, expected",
+    [
+        ("", None),
+        ("hello", "hello"),
+        (None, None),
+    ],
+)
+def test_null_if_empty_string(value, expected):
+    t = NullIfEmptyString()
+    assert t.process_bind_param(value, None) == expected
+
+
+# --- YNString ---
+
+@pytest.mark.parametrize(
+    "value, expected",
+    [
+        (True, "Y"),
+        (False, "N"),
+        ("Y", "Y"),
+        ("N", "N"),
+        (None, None),
+    ],
+)
+def test_yn_string_bind(value, expected):
+    t = YNString()
+    assert t.process_bind_param(value, None) == expected
+
+
+@pytest.mark.parametrize(
+    "value, expected",
+    [
+        ("Y", True),
+        ("N", False),
+        (None, None),
+    ],
+)
+def test_yn_string_result(value, expected):
+    t = YNString()
+    assert t.process_result_value(value, None) == expected
+
+
+# --- ZeroIfEmptyInteger ---
+
+@pytest.mark.parametrize(
+    "value, expected",
+    [
+        (5, 5),
+        (0, 0),
+        ("", 0),
+        (None, 0),
+    ],
+)
+def test_zero_if_empty_integer(value, expected):
+    t = ZeroIfEmptyInteger()
+    assert t.process_bind_param(value, None) == expected
+
+
+# --- UpdateTimestamp ---
+
+def test_update_timestamp():
+    t = UpdateTimestamp()
+    before = datetime.now(UTC)
+    result = t.process_bind_param("ignored", None)
+    after = datetime.now(UTC)
+    assert before <= result <= after
+
+
+# --- EncryptedBytea roundtrip ---
+
+@patch("testgen.common.encrypt.settings")
+def test_encrypted_bytea_roundtrip(mock_settings):
+    mock_settings.APP_ENCRYPTION_SALT = "testsalt12345678"
+    mock_settings.APP_ENCRYPTION_SECRET = "testsecret123456"
+
+    t = EncryptedBytea()
+    original = "sensitive data"
+
+    encrypted = t.process_bind_param(original, None)
+    assert encrypted != original.encode()
+
+    decrypted = t.process_result_value(encrypted, None)
+    assert decrypted == original
+
+
+@patch("testgen.common.encrypt.settings")
+def test_encrypted_bytea_none(mock_settings):
+    t = EncryptedBytea()
+    assert t.process_bind_param(None, None) is None
+    assert t.process_result_value(None, None) is None
+
+
+# --- EncryptedJson roundtrip ---
+
+@patch("testgen.common.encrypt.settings")
+def test_encrypted_json_roundtrip(mock_settings):
+    mock_settings.APP_ENCRYPTION_SALT = "testsalt12345678"
+    mock_settings.APP_ENCRYPTION_SECRET = "testsecret123456"
+
+    t = EncryptedJson()
+    original = {"key": "value", "num": 42, "list": [1, 2, 3]}
+
+    encrypted = t.process_bind_param(original, None)
+    decrypted = t.process_result_value(encrypted, None)
+    assert decrypted == original
+
+
+@patch("testgen.common.encrypt.settings")
+def test_encrypted_json_none(mock_settings):
+    t = EncryptedJson()
+    assert t.process_bind_param(None, None) is None
+    assert t.process_result_value(None, None) is None

tests/unit/common/test_clean_sql.py

@@ -0,0 +1,74 @@
+import pytest
+
+from testgen.common.clean_sql import CleanSQL, concat_columns
+
+pytestmark = pytest.mark.unit
+
+
+# --- CleanSQL ---
+
+def test_clean_sql_block_comments():
+    assert CleanSQL("SELECT /* comment */ 1") == "SELECT 1"
+
+
+def test_clean_sql_multiline_block_comments():
+    sql = """SELECT /*
+    multi-line
+    comment
+    */ 1"""
+    assert CleanSQL(sql) == "SELECT 1"
+
+
+def test_clean_sql_line_comments():
+    sql = "SELECT 1 -- this is a comment\nFROM t"
+    assert CleanSQL(sql) == "SELECT 1 FROM t"
+
+
+def test_clean_sql_tabs_and_extra_spaces():
+    sql = "SELECT\t  1\t\tFROM   t"
+    assert CleanSQL(sql) == "SELECT 1 FROM t"
+
+
+def test_clean_sql_preserves_quoted_strings():
+    sql = "SELECT '  spaces  ' FROM t"
+    result = CleanSQL(sql)
+    assert "'  spaces  '" in result
+
+
+def test_clean_sql_preserves_double_quoted_strings():
+    sql = 'SELECT "  col  name  " FROM t'
+    result = CleanSQL(sql)
+    assert '"  col  name  "' in result
+
+
+def test_clean_sql_combined():
+    sql = """
+    SELECT /* get all */
+        col1,   col2
+    FROM   table1 -- main table
+    WHERE  col1 = 'hello  world'
+    """
+    result = CleanSQL(sql)
+    assert "/* get all */" not in result
+    assert "-- main table" not in result
+    assert "'hello  world'" in result
+    assert "col1, col2" in result
+
+
+# --- concat_columns ---
+
+def test_concat_columns_multiple():
+    result = concat_columns("col1, col2, col3", "NULL")
+    assert result == "CONCAT(COALESCE(col1, 'NULL'), COALESCE(col2, 'NULL'), COALESCE(col3, 'NULL'))"
+
+
+def test_concat_columns_single():
+    assert concat_columns("col1", "NULL") == "col1"
+
+
+def test_concat_columns_empty():
+    assert concat_columns("", "NULL") == ""
+
+
+def test_concat_columns_none():
+    assert concat_columns(None, "NULL") == ""

tests/unit/common/test_date_service.py

@@ -0,0 +1,29 @@
+from datetime import UTC, datetime
+from unittest.mock import patch
+
+import pytest
+
+from testgen.common.date_service import as_iso_timestamp, get_now_as_iso_timestamp
+
+pytestmark = pytest.mark.unit
+
+
+@pytest.mark.parametrize(
+    "value, expected",
+    [
+        (datetime(2024, 3, 15, 10, 30, 45), "2024-03-15T10:30:45Z"),
+        (datetime(2024, 1, 1, 0, 0, 0), "2024-01-01T00:00:00Z"),
+        (None, None),
+    ],
+)
+def test_as_iso_timestamp(value, expected):
+    assert as_iso_timestamp(value) == expected
+
+
+def test_get_now_as_iso_timestamp():
+    with patch("testgen.common.date_service.datetime") as mock_dt:
+        mock_dt.now.return_value = datetime(2024, 6, 15, 12, 0, 0, tzinfo=UTC)
+        mock_dt.strftime = datetime.strftime
+        result = get_now_as_iso_timestamp()
+
+    assert result == "2024-06-15T12:00:00Z"