fix(coderd/database): fix down migration and add fixture

johnstcn · johnstcn · commit fb58caa1d119 · 2025-05-01T14:08:20.000+01:00
diff --git a/coderd/database/migrations/000318_chat.down.sql b/coderd/database/migrations/000318_chat.down.sql
@@ -1,34 +1,3 @@
--- Remove the created_at column
-ALTER TABLE chat_messages
-DROP COLUMN IF EXISTS created_at;
+DROP TABLE IF EXISTS chat_messages;
 
--- Add back message_type and message_index (assuming previous state was TEXT and SERIAL)
-ALTER TABLE chat_messages
-ADD COLUMN IF NOT EXISTS message_type TEXT NOT NULL DEFAULT '', -- Provide a default or handle NULLs
-ADD COLUMN IF NOT EXISTS message_index SERIAL;
-
--- Change content back to TEXT (data loss may occur if JSONB data is not representable as TEXT)
-ALTER TABLE chat_messages
-ALTER COLUMN content TYPE TEXT USING content::TEXT;
-
--- Attempt to revert id back to UUID with default
--- WARNING: This is complex and potentially destructive. It might fail if data exists.
--- It drops the existing primary key, sequence, and default, then attempts to set a new one.
-ALTER TABLE chat_messages DROP CONSTRAINT IF EXISTS chat_messages_pkey;
-DROP SEQUENCE IF EXISTS chat_messages_id_seq;
-ALTER TABLE chat_messages ALTER COLUMN id DROP DEFAULT;
-ALTER TABLE chat_messages ALTER COLUMN id TYPE UUID USING (gen_random_uuid()); -- Attempt conversion, may fail
-ALTER TABLE chat_messages ALTER COLUMN id SET DEFAULT gen_random_uuid();
-ALTER TABLE chat_messages ADD PRIMARY KEY (id);
-
--- Revert changes to chat_conversations (removing deleted_at)
-ALTER TABLE chat_conversations
-DROP COLUMN IF EXISTS deleted_at;
-ALTER TABLE chat_conversations DROP CONSTRAINT IF EXISTS chat_conversations_user_id_fkey;
-ALTER TABLE chat_conversations ADD CONSTRAINT chat_conversations_user_id_fkey FOREIGN KEY (user_id) REFERENCES users(id);
-ALTER TABLE chat_conversations ALTER COLUMN created_at SET DEFAULT CURRENT_TIMESTAMP;
-ALTER TABLE chat_conversations ALTER COLUMN updated_at SET DEFAULT CURRENT_TIMESTAMP;
-
--- Revert changes to chat_messages (removing ON DELETE CASCADE)
-ALTER TABLE chat_messages DROP CONSTRAINT IF EXISTS chat_messages_conversation_id_fkey;
-ALTER TABLE chat_messages ADD CONSTRAINT chat_messages_conversation_id_fkey FOREIGN KEY (conversation_id) REFERENCES chat_conversations(id);
+DROP TABLE IF EXISTS chats;
diff --git a/coderd/database/migrations/testdata/fixtures/000318_chat.up.sql b/coderd/database/migrations/testdata/fixtures/000318_chat.up.sql
@@ -0,0 +1,6 @@
+INSERT INTO chats (id, owner_id, created_at, updated_at, title) VALUES
+('00000000-0000-0000-0000-000000000001', '0ed9befc-4911-4ccf-a8e2-559bf72daa94', '2023-10-01 12:00:00+00', '2023-10-01 12:00:00+00', 'Test Chat 1');
+
+INSERT INTO chat_messages (id, chat_id, created_at, model, provider, content) VALUES
+(1, '00000000-0000-0000-0000-000000000001', '2023-10-01 12:00:00+00', 'annie-oakley', 'cowboy-coder', '{"role":"user","content":"Hello"}'),
+(2, '00000000-0000-0000-0000-000000000001', '2023-10-01 12:01:00+00', 'annie-oakley', 'cowboy-coder', '{"role":"assistant","content":"Howdy pardner! What can I do ya for?"}');
