polish

ybkuroki · Oct 3, 2020 · 322d920 · 322d920
1 parent df0aed6
commit 322d920
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/controller/book_test.go b/controller/book_test.go
@@ -113,17 +113,9 @@ func TestPostBookDelete(t *testing.T) {
 }
 
 func setUpTestData() {
-	model := &model.Book{
-		ID:         1,
-		Title:      "Test1",
-		Isbn:       "123-123-123-1",
-		CategoryID: 1,
-		Category:   &model.Category{ID: 1, Name: "技術書"},
-		FormatID:   1,
-		Format:     &model.Format{ID: 1, Name: "書籍"},
-	}
+	model := model.NewBook("Test1", "123-123-123-1", 1, 1)
 	repo := repository.GetRepository()
-	_, _ = model.Save(repo)
+	_, _ = model.Create(repo)
 }
 
 func createRegDto() *dto.RegBookDto {