Add error case test (#28)

craigpastro · Oct 19, 2023 · 2a8b009 · 2a8b009
1 parent 292d669
commit 2a8b009
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/mocks/row.go b/mocks/row.go
diff --git a/pgmq_test.go b/pgmq_test.go
@@ -350,6 +350,12 @@ func TestErrorCases(t *testing.T) {
 		require.ErrorContains(t, err, "postgres error")
 	})
 
+	t.Run("createUnloggedQueueError", func(t *testing.T) {
+		mockDB.EXPECT().Exec(ctx, "SELECT pgmq.create_unlogged($1)", queue).Return(cmdTag, testErr)
+		err := q.CreateUnloggedQueue(ctx, queue)
+		require.ErrorContains(t, err, "postgres error")
+	})
+
 	t.Run("dropQueueError", func(t *testing.T) {
 		mockDB.EXPECT().Exec(ctx, "SELECT pgmq.drop_queue($1)", queue).Return(cmdTag, testErr)
 		err := q.DropQueue(ctx, queue)