Skip to content

Commit 7924813

Browse files
rlazoDavid Motsonashvilidavidmotsondaymxn
authored
[ai] Use .json for unary tests instead of .txt (#6916)
Follow up to #6915 --------- Co-authored-by: David Motsonashvili <[email protected]> Co-authored-by: David Motsonashvili <[email protected]> Co-authored-by: Daymon <[email protected]>
1 parent 7cbb80b commit 7924813

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

firebase-ai/src/test/java/com/google/firebase/ai/DevAPIUnarySnapshotTests.kt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ internal class DevAPIUnarySnapshotTests {
3535

3636
@Test
3737
fun `short reply`() =
38-
goldenDevAPIUnaryFile("unary-success-basic-reply-short.txt") {
38+
goldenDevAPIUnaryFile("unary-success-basic-reply-short.json") {
3939
withTimeout(testTimeout) {
4040
val response = model.generateContent("prompt")
4141

@@ -47,7 +47,7 @@ internal class DevAPIUnarySnapshotTests {
4747

4848
@Test
4949
fun `long reply`() =
50-
goldenDevAPIUnaryFile("unary-success-basic-reply-long.txt") {
50+
goldenDevAPIUnaryFile("unary-success-basic-reply-long.json") {
5151
withTimeout(testTimeout) {
5252
val response = model.generateContent("prompt")
5353

@@ -85,14 +85,14 @@ internal class DevAPIUnarySnapshotTests {
8585

8686
@Test
8787
fun `invalid api key`() =
88-
goldenDevAPIUnaryFile("unary-failure-api-key.txt", HttpStatusCode.BadRequest) {
88+
goldenDevAPIUnaryFile("unary-failure-api-key.json", HttpStatusCode.BadRequest) {
8989
withTimeout(testTimeout) {
9090
shouldThrow<InvalidAPIKeyException> { model.generateContent("prompt") }
9191
}
9292
}
9393
@Test
9494
fun `unknown model`() =
95-
goldenDevAPIUnaryFile("unary-failure-unknown-model.txt", HttpStatusCode.NotFound) {
95+
goldenDevAPIUnaryFile("unary-failure-unknown-model.json", HttpStatusCode.NotFound) {
9696
withTimeout(testTimeout) { shouldThrow<ServerException> { model.generateContent("prompt") } }
9797
}
9898
}

0 commit comments

Comments
 (0)