Add tests, and make the messages more similar

Author: encukou

Lib/test/test_dbm_gnu.py

@@ -217,6 +217,25 @@ def test_localized_error(self):
             create_empty_file(os.path.join(d, 'test'))
             self.assertRaises(gdbm.error, gdbm.open, filename, 'r')
 
+    def test_type_errors(self):
+        self.g = gdbm.open(filename, 'c')
+        with self.assertRaisesRegex(
+            TypeError, "^a bytes-like object is required, not 'int'$",
+        ):
+            self.g[123]
+        with self.assertRaisesRegex(
+            TypeError, "^gdbm key must be bytes or str, not 'int'$",
+        ):
+            123 in self.g
+        with self.assertRaisesRegex(
+            TypeError, "^gdbm key must be bytes or str, not 'NoneType'$",
+        ):
+            self.g[None] = 123
+        with self.assertRaisesRegex(
+            TypeError, "^gdbm value must be bytes or str, not 'int'$",
+        ):
+            self.g['foo'] = 123
+
 
 if __name__ == '__main__':
     unittest.main()

Lib/test/test_dbm_ndbm.py

@@ -160,6 +160,25 @@ def test_clear(self):
                 self.assertNotIn(k, db)
             self.assertEqual(len(db), 0)
 
+    def test_type_errors(self):
+        with dbm.ndbm.open(self.filename, 'c') as db:
+            with self.assertRaisesRegex(
+                TypeError, "^a bytes-like object is required, not 'int'$",
+            ):
+                db[123]
+            with self.assertRaisesRegex(
+                TypeError, "^dbm key must be bytes or str, not 'int'$",
+            ):
+                123 in db
+            with self.assertRaisesRegex(
+                TypeError, "^dbm key must be bytes or str, not 'NoneType'$",
+            ):
+                db[None] = 123
+            with self.assertRaisesRegex(
+                TypeError, "^dbm value must be bytes or str, not 'int'$",
+            ):
+                db['foo'] = 123
+
 
 if __name__ == '__main__':
     unittest.main()

Modules/_dbmmodule.c

@@ -232,7 +232,7 @@ dbm_ass_sub_lock_held(PyObject *self, PyObject *v, PyObject *w)
 
     if ( !PyArg_Parse(v, "s#", &krec.dptr, &tmp_size) ) {
         PyErr_Format(PyExc_TypeError,
-                     "database keys must be bytes or str, not %T",
+                     "dbm key must be bytes or str, not '%T'",
                      v);
         return -1;
     }
@@ -260,7 +260,7 @@ dbm_ass_sub_lock_held(PyObject *self, PyObject *v, PyObject *w)
     } else {
         if ( !PyArg_Parse(w, "s#", &drec.dptr, &tmp_size) ) {
             PyErr_Format(PyExc_TypeError,
-                         "database values must be bytes or str, not %T",
+                         "dbm value must be bytes or str, not '%T'",
                          w);
             return -1;
         }
@@ -371,8 +371,7 @@ dbm_contains_lock_held(PyObject *self, PyObject *arg)
     }
     else if (!PyBytes_Check(arg)) {
         PyErr_Format(PyExc_TypeError,
-                     "dbm key must be bytes or string, not %.100s",
-                     Py_TYPE(arg)->tp_name);
+                     "dbm key must be bytes or str, not '%T'", arg);
         return -1;
     }
     else {

Modules/_gdbmmodule.c

@@ -237,13 +237,15 @@ gdbm_bool(PyObject *op)
 // This function is needed to support PY_SSIZE_T_CLEAN.
 // Return 1 on success, same to PyArg_Parse().
 static int
-parse_datum(PyObject *o, datum *d, const char *items_name)
+parse_datum(PyObject *o, datum *d, const char *item_name)
 {
     Py_ssize_t size;
     if (!PyArg_Parse(o, "s#", &d->dptr, &size)) {
-        PyErr_Format(PyExc_TypeError,
-                     "database %s must be bytes or str, not %T",
-                     items_name, o);
+        if (item_name) {
+            PyErr_Format(PyExc_TypeError,
+                         "gdbm %s must be bytes or str, not '%T'",
+                         item_name, o);
+        }
         return 0;
     }
     if (INT_MAX < size) {
@@ -262,7 +264,7 @@ gdbm_subscript_lock_held(PyObject *op, PyObject *key)
     gdbmobject *dp = _gdbmobject_CAST(op);
     _gdbm_state *state = PyType_GetModuleState(Py_TYPE(dp));
 
-    if (!parse_datum(key, &krec, "keys")) {
+    if (!parse_datum(key, &krec, NULL)) {
         return NULL;
     }
     if (dp->di_dbm == NULL) {
@@ -321,7 +323,7 @@ gdbm_ass_sub_lock_held(PyObject *op, PyObject *v, PyObject *w)
     gdbmobject *dp = _gdbmobject_CAST(op);
     _gdbm_state *state = PyType_GetModuleState(Py_TYPE(dp));
 
-    if (!parse_datum(v, &krec, "keys")) {
+    if (!parse_datum(v, &krec, "key")) {
         return -1;
     }
     if (dp->di_dbm == NULL) {
@@ -342,7 +344,7 @@ gdbm_ass_sub_lock_held(PyObject *op, PyObject *v, PyObject *w)
         }
     }
     else {
-        if (!parse_datum(w, &drec, "values")) {
+        if (!parse_datum(w, &drec, "value")) {
             return -1;
         }
         errno = 0;
@@ -490,8 +492,7 @@ gdbm_contains_lock_held(PyObject *self, PyObject *arg)
     }
     else if (!PyBytes_Check(arg)) {
         PyErr_Format(PyExc_TypeError,
-                     "gdbm key must be bytes or string, not %.100s",
-                     Py_TYPE(arg)->tp_name);
+                     "gdbm key must be bytes or str, not '%T'", arg);
         return -1;
     }
     else {