tests: add GetByXXX and GetByXXXWithTx method #41

go-generalize · Apr 4, 2022 · 9779365 · 9779365
1 parent ab45270
commit 9779365
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/generator/testfiles/not_auto/tests/model_test.go b/generator/testfiles/not_auto/tests/model_test.go
@@ -315,6 +315,12 @@ func TestFirestore(t *testing.T) {
 				ttrr.Fatalf("unexpected err == ErrUniqueConstraint")
 			}
 		})
+
+		tr.Run("GetByXXX", func(ttr *testing.T) {
+			if _, err := taskRepo.GetByDesc(ctx, fmt.Sprintf("%s%d", desc, 1001)); err != nil {
+				ttr.Fatalf("%+v", err)
+			}
+		})
 	})
 }
 
@@ -495,6 +501,18 @@ func TestFirestoreTransaction_Single(t *testing.T) {
 			tr.Fatalf("unexpected err == ErrUniqueConstraint")
 		}
 	})
+
+	t.Run("GetByXXXWithTx", func(tr *testing.T) {
+		if err := client.RunTransaction(ctx, func(cx context.Context, tx *firestore.Transaction) error {
+			if _, err := taskRepo.GetByDescWithTx(tx, fmt.Sprintf("%s01", desc)); err != nil {
+				return err
+			}
+
+			return nil
+		}); err != nil {
+			tr.Fatalf("error: %+v", err)
+		}
+	})
 }
 
 func TestFirestoreTransaction_Multi(t *testing.T) {