test: add test to verify missing oid lookups in string formats

diff --git a/tests/types/test_numeric.py b/tests/types/test_numeric.py
index cc471e8c717bd3cf42ba9ab259c08450d9ce93da..a27bc849c0cc98070942089d98195a724e3ecf8d 100644 (file)
--- a/tests/types/test_numeric.py
+++ b/tests/types/test_numeric.py
@@ -612,3 +612,14 @@ def test_repr_wrapper(conn, wrapper, fmt_in):
     cur = conn.execute(f"select pg_typeof(%{fmt_in.value})::oid", [wrapper(0)])
     oid = cur.fetchone()[0]
     assert oid == psycopg.postgres.types[wrapper.__name__.lower()].oid
+
+
+@pytest.mark.parametrize("fmt_out", pq.Format)
+@pytest.mark.parametrize(
+    "typename",
+    "integer int2 int4 int8 float4 float8 numeric".split() + ["double precision"],
+)
+def test_oid_lookup(conn, typename, fmt_out):
+    dumper = conn.adapters.get_dumper_by_oid(conn.adapters.types[typename].oid, fmt_out)
+    assert dumper.oid == conn.adapters.types[typename].oid
+    assert dumper.format == fmt_out

diff --git a/tests/types/test_string.py b/tests/types/test_string.py
index 8499959e13f3c373651d07da76afb8794fd4f39b..d23e5e03b3ea18e0403f62d1e68032c6632608c4 100644 (file)
--- a/tests/types/test_string.py
+++ b/tests/types/test_string.py
@@ -241,6 +241,14 @@ def test_text_array_ascii(conn, fmt_in, fmt_out):
     assert res == exp
 
 
+@pytest.mark.parametrize("fmt_out", pq.Format)
+@pytest.mark.parametrize("typename", ["text", "varchar", "name"])
+def test_oid_lookup(conn, typename, fmt_out):
+    dumper = conn.adapters.get_dumper_by_oid(conn.adapters.types[typename].oid, fmt_out)
+    assert dumper.oid == conn.adapters.types[typename].oid
+    assert dumper.format == fmt_out
+
+
 #
 # tests with bytea
 #