From 20a5aca1ec3c9543ba0466820add5b70496e6d3b Mon Sep 17 00:00:00 2001
From: Inada Naoki <songofacandy@gmail.com>
Date: Mon, 13 Mar 2023 16:55:43 +0900
Subject: [PATCH] Add tests for #535

---
 tests/test_errors.py | 56 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 tests/test_errors.py

diff --git a/tests/test_errors.py b/tests/test_errors.py
new file mode 100644
index 00000000..3a9fecaa
--- /dev/null
+++ b/tests/test_errors.py
@@ -0,0 +1,56 @@
+import pytest
+import MySQLdb.cursors
+from configdb import connection_factory
+
+
+_conns = []
+_tables = []
+
+
+def connect(**kwargs):
+    conn = connection_factory(**kwargs)
+    _conns.append(conn)
+    return conn
+
+
+def teardown_function(function):
+    if _tables:
+        c = _conns[0]
+        cur = c.cursor()
+        for t in _tables:
+            cur.execute("DROP TABLE {}".format(t))
+        cur.close()
+        del _tables[:]
+
+    for c in _conns:
+        c.close()
+    del _conns[:]
+
+
+def test_null():
+    """Inserting NULL into non NULLABLE column"""
+    # https://github.com/PyMySQL/mysqlclient/issues/535
+    table_name = "test_null"
+    conn = connect()
+    cursor = conn.cursor()
+
+    cursor.execute(f"create table {table_name} (c1 int primary key)")
+    _tables.append(table_name)
+
+    with pytest.raises(MySQLdb.IntegrityError):
+        cursor.execute(f"insert into {table_name} values (null)")
+
+
+def test_duplicated_pk():
+    """Inserting row with duplicated PK"""
+    # https://github.com/PyMySQL/mysqlclient/issues/535
+    table_name = "test_duplicated_pk"
+    conn = connect()
+    cursor = conn.cursor()
+
+    cursor.execute(f"create table {table_name} (c1 int primary key)")
+    _tables.append(table_name)
+
+    cursor.execute(f"insert into {table_name} values (1)")
+    with pytest.raises(MySQLdb.IntegrityError):
+        cursor.execute(f"insert into {table_name} values (1)")