test追加

traPtitech · Oct 18, 2023 · e3d6c7c · e3d6c7c
1 parent 6fde0ca
commit e3d6c7c
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/repository/gorm/stamp_test.go b/repository/gorm/stamp_test.go
@@ -364,3 +364,24 @@ func TestGormRepository_GetStampStats(t *testing.T) {
 	})
 
 }
+
+
+func TestRepositoryImpl_StampThumbnailExists(t *testing.T) {
+	t.Parallel()
+	repo, assert, _ := setup(t, ex1)
+
+	n := 10
+	for i := 0; i < 10; i++ {
+		mustMakeStamp(t, repo, rand, uuid.Nil)
+	}
+
+	arr, err := repo.GetAllStamps(repository.StampTypeAll)
+	if assert.NoError(err) {
+		assert.Len(arr, n)
+	}
+
+	arrWithThumb, err := repo.StampThumbnailExists(arr)
+	if assert.NoError(err) {
+		assert.Len(arrWithThumb, n)
+	}
+}
diff --git a/router/v3/stamps_test.go b/router/v3/stamps_test.go
@@ -86,6 +86,7 @@ func TestHandlers_GetStamps(t *testing.T) {
 
 		obj.Length().IsEqual(1)
 		stampEquals(t, stamp, obj.Value(0).Object())
+		obj.Value(0).Object().Value("HasThumbnail").Boolean().IsEqual(false)
 	})
 
 	t.Run("success (type=original)", func(t *testing.T) {
@@ -101,6 +102,7 @@ func TestHandlers_GetStamps(t *testing.T) {
 
 		obj.Length().IsEqual(1)
 		stampEquals(t, stamp, obj.Value(0).Object())
+		obj.Value(0).Object().Value("HasThumbnail").Boolean().IsEqual(false)
 	})
 
 	t.Run("success (type=unicode)", func(t *testing.T) {
@@ -130,6 +132,7 @@ func TestHandlers_GetStamps(t *testing.T) {
 
 		obj.Length().IsEqual(1)
 		stampEquals(t, stamp, obj.Value(0).Object())
+		obj.Value(0).Object().Value("HasThumbnail").Boolean().IsEqual(false)
 	})
 }