chore: tests for list type

Author: tandemdude

tests/src/main/resources/postgres/queries.sql

@@ -10,6 +10,7 @@ SELECT * FROM users;
 -- name: GetMessage :one
 SELECT * FROM messages WHERE message_id = $1;
 
--- name: CreateMessage :exec
+-- name: CreateMessage :one
 INSERT INTO messages(chat_id, user_id, content, attachments)
-VALUES ($1, $2, $3, $4);
+VALUES ($1, $2, $3, $4)
+RETURNING message_id;

tests/src/main/resources/postgres/schema.sql

@@ -9,7 +9,7 @@ CREATE TABLE users (
 -- Tokens table
 CREATE TABLE tokens (
     token_id SERIAL PRIMARY KEY,
-    user_id  UUID NOT NULL REFERENCES users(user_id),
+    user_id  UUID NOT NULL,
     token    VARCHAR(255) UNIQUE NOT NULL,
     expiry   TIMESTAMP NOT NULL
 );
@@ -24,8 +24,8 @@ CREATE TABLE chats (
 -- Messages table with array column for attachments
 CREATE TABLE messages (
     message_id SERIAL PRIMARY KEY,
-    chat_id    INTEGER NOT NULL REFERENCES chats(chat_id),
-    user_id    UUID NOT NULL REFERENCES users(user_id),
+    chat_id    INTEGER NOT NULL,
+    user_id    UUID NOT NULL,
     content    TEXT NOT NULL,
     attachments TEXT[], -- Array column for storing attachment URLs
     sent_at_time    TIME DEFAULT NOW()::time without time zone,

tests/src/test/java/io/github/tandemdude/sgj/postgres/TestQueries.java

@@ -9,6 +9,7 @@
 import java.sql.Connection;
 import java.sql.DriverManager;
 import java.sql.SQLException;
+import java.util.List;
 import java.util.UUID;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -76,4 +77,19 @@ public void listUsersReturnsPopulatedListRecordsFound() throws Exception {
             assertThat(found.size()).isEqualTo(2);
         }
     }
+
+    @Test
+    @DisplayName("CreateMessage processes input list correctly")
+    public void createMessageProcessesInputListCorrectly() throws Exception {
+        try (var conn = getConn()) {
+            var q = new Queries(conn);
+
+            var created = q.createMessage(1, UUID.randomUUID(), "foo", List.of("bar", "baz", "bork"));
+            assertThat(created).isPresent();
+
+            var found = q.getMessage(created.get().message_id());
+            assertThat(found).isPresent();
+            assertThat(found.get().attachments()).containsExactly("bar", "baz", "bork");
+        }
+    }
 }