wip

IceWhaleTech · Aug 5, 2024 · c2fb315 · c2fb315
1 parent 5d104e3
commit c2fb315
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/service/ysk_test.go b/service/ysk_test.go
@@ -11,16 +11,19 @@ import (
 	"gotest.tools/assert"
 )
 
-func setup(t *testing.T) *service.YSKService {
+func setup(t *testing.T) (*service.YSKService, func()) {
 	repository, err := repository.NewDatabaseRepositoryInMemory()
 	assert.NilError(t, err)
 
 	yskService := service.NewYSKService(&repository)
-	return yskService
+	return yskService, func() {
+		repository.Close()
+	}
 }
 
 func TestInsertAndGetCardList(t *testing.T) {
-	yskService := setup(t)
+	yskService, cleanup := setup(t)
+	defer cleanup()
 
 	cardList, err := yskService.YskCardList(context.Background())
 	assert.NilError(t, err)
@@ -51,7 +54,8 @@ func TestInsertAndGetCardList(t *testing.T) {
 }
 
 func TestInsertAllTypeCardList(t *testing.T) {
-	yskService := setup(t)
+	yskService, cleanup := setup(t)
+	defer cleanup()
 
 	cardList, err := yskService.YskCardList(context.Background())
 	assert.NilError(t, err)