Cleanup, testing

errors.go

@@ -14,10 +14,6 @@ func ErrIsForeignKey(err error) bool {
 	return ErrHasCode(err, "23503")
 }
 
-func ErrIsCheck(err error) bool {
-	return ErrHasCode(err, "23514")
-}
-
 func ErrIsSerializationFaiilure(err error) bool {
 	return ErrHasCode(err, "40001")
 }

errors_test.go

@@ -0,0 +1,36 @@
+package pgutil
+
+import (
+	"database/sql"
+	"testing"
+)
+
+func TestErrors(t *testing.T) {
+	db, err := sql.Open(
+		"postgres",
+		"host=127.0.0.1 dbname=test sslmode=disable user=test password=test")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if err := DropAllTables(db); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := Migrate(db, testMigrationFS); err != nil {
+		t.Fatal(err)
+	}
+
+	_, err = db.Exec(`INSERT INTO users(UserID, Email) VALUES (2, 'q@r.com')`)
+	if !ErrIsDuplicateKey(err) {
+		t.Fatal(err)
+	}
+	_, err = db.Exec(`INSERT INTO users(UserID, Email) VALUES (3, 'c@d.com')`)
+	if !ErrIsDuplicateKey(err) {
+		t.Fatal(err)
+	}
+	_, err = db.Exec(`INSERT INTO user_notes(UserID, NoteID, Note) VALUES (4, 1, 'hello')`)
+	if !ErrIsForeignKey(err) {
+		t.Fatal(err)
+	}
+}

test-migrations/000.sql

@@ -1,3 +1,9 @@
 CREATE TABLE users(
        UserID BIGINT NOT NULL PRIMARY KEY,
        Email  TEXT   NOT NULL UNIQUE);
+
+CREATE TABLE user_notes(
+       UserID BIGINT NOT NULL REFERENCES users(UserID),
+       NoteID BIGINT NOT NULL,
+       Note   Text   NOT NULL,
+       PRIMARY KEY(UserID,NoteID));