From 49617f4e98effa813f98325e48751f12f3e1ca16 Mon Sep 17 00:00:00 2001
From: Irine Sistiana <49315432+IrineSistiana@users.noreply.github.com>
Date: Wed, 23 Aug 2023 19:27:13 +0800
Subject: [PATCH] Update main_test.go

---
 main_test.go | 36 ++++++++++++++++++++++++------------
 1 file changed, 24 insertions(+), 12 deletions(-)

diff --git a/main_test.go b/main_test.go
index 60a388f7..04710161 100644
--- a/main_test.go
+++ b/main_test.go
@@ -1,20 +1,32 @@
 package main
 
 import (
+	"log"
+	"os"
 	"testing"
-)
-
-// GORM_REPO: https://github.com/go-gorm/gorm.git
-// GORM_BRANCH: master
-// TEST_DRIVERS: sqlite, mysql, postgres, sqlserver
 
-func TestGORM(t *testing.T) {
-	user := User{Name: "jinzhu"}
+	"github.com/stretchr/testify/require"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+)
 
-	DB.Create(&user)
+type M struct {
+	ID uint64 `gorm:"primarykey"`
+	I  string `gorm:"uniqueIndex:idx_name"`
+	I2 string `gorm:"uniqueIndex:idx_name"`
+}
 
-	var result User
-	if err := DB.First(&result, user.ID).Error; err != nil {
-		t.Errorf("Failed, got error: %v", err)
-	}
+func Test(t *testing.T) {
+	r := require.New(t)
+	db, err := gorm.Open(sqlite.Open("test.sqlite3?_journal=WAL"), &gorm.Config{
+		Logger: logger.New(
+			log.New(os.Stdout, "\r\n", log.LstdFlags),
+			logger.Config{
+				LogLevel: logger.Info,
+			},
+		),
+	})
+	r.NoError(err)
+	db.AutoMigrate(&M{})
 }