diff --git a/prisma/schema.prisma b/prisma/schema.prisma
index 4f7ee8a..8571072 100644
--- a/prisma/schema.prisma
+++ b/prisma/schema.prisma
@@ -54,6 +54,7 @@ model Problem {
   userId      String
   user        User       @relation(fields: [userId], references: [id], onDelete: Cascade)
   templates   Template[]
+  testcases   Testcase[]
 
   @@index([userId])
   @@index([difficulty])
@@ -127,6 +128,21 @@ model JudgeResult {
   memoryUsage   Int?
 }
 
+model Testcase {
+  id        String         @id @default(cuid())
+  problemId String
+  problem   Problem        @relation(fields: [problemId], references: [id], onDelete: Cascade)
+  data      TestcaseData[]
+}
+
+model TestcaseData {
+  id         String   @id @default(cuid())
+  label      String
+  value      String
+  testcaseId String
+  testcase   Testcase @relation(fields: [testcaseId], references: [id], onDelete: Cascade)
+}
+
 model Account {
   userId            String
   type              String