Skip to content

Commit 0b026b2

Browse files
authored
VertexAI: update test cases to use full name of mock response files (#6127)
1 parent 1b49171 commit 0b026b2

File tree

3 files changed

+55
-52
lines changed

3 files changed

+55
-52
lines changed

firebase-vertexai/src/test/java/com/google/firebase/vertexai/StreamingSnapshotTests.kt

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ internal class StreamingSnapshotTests {
4343

4444
@Test
4545
fun `short reply`() =
46-
goldenStreamingFile("success-basic-reply-short.txt") {
46+
goldenStreamingFile("streaming-success-basic-reply-short.txt") {
4747
val responses = model.generateContentStream("prompt")
4848

4949
withTimeout(testTimeout) {
@@ -57,7 +57,7 @@ internal class StreamingSnapshotTests {
5757

5858
@Test
5959
fun `long reply`() =
60-
goldenStreamingFile("success-basic-reply-long.txt") {
60+
goldenStreamingFile("streaming-success-basic-reply-long.txt") {
6161
val responses = model.generateContentStream("prompt")
6262

6363
withTimeout(testTimeout) {
@@ -73,7 +73,7 @@ internal class StreamingSnapshotTests {
7373

7474
@Test
7575
fun `unknown enum`() =
76-
goldenStreamingFile("success-unknown-enum.txt") {
76+
goldenStreamingFile("streaming-success-unknown-enum.txt") {
7777
val responses = model.generateContentStream("prompt")
7878

7979
withTimeout(testTimeout) {
@@ -88,7 +88,7 @@ internal class StreamingSnapshotTests {
8888

8989
@Test
9090
fun `unknown enum in finish reason`() =
91-
goldenStreamingFile("failure-unknown-finish-enum.txt") {
91+
goldenStreamingFile("streaming-failure-unknown-finish-enum.txt") {
9292
val responses = model.generateContentStream("prompt")
9393

9494
withTimeout(testTimeout) {
@@ -99,7 +99,7 @@ internal class StreamingSnapshotTests {
9999

100100
@Test
101101
fun `quotes escaped`() =
102-
goldenStreamingFile("success-quotes-escaped.txt") {
102+
goldenStreamingFile("streaming-success-quotes-escaped.txt") {
103103
val responses = model.generateContentStream("prompt")
104104

105105
withTimeout(testTimeout) {
@@ -114,7 +114,7 @@ internal class StreamingSnapshotTests {
114114

115115
@Test
116116
fun `prompt blocked for safety`() =
117-
goldenStreamingFile("failure-prompt-blocked-safety.txt") {
117+
goldenStreamingFile("streaming-failure-prompt-blocked-safety.txt") {
118118
val responses = model.generateContentStream("prompt")
119119

120120
withTimeout(testTimeout) {
@@ -125,23 +125,23 @@ internal class StreamingSnapshotTests {
125125

126126
@Test
127127
fun `empty content`() =
128-
goldenStreamingFile("failure-empty-content.txt") {
128+
goldenStreamingFile("streaming-failure-empty-content.txt") {
129129
val responses = model.generateContentStream("prompt")
130130

131131
withTimeout(testTimeout) { shouldThrow<SerializationException> { responses.collect() } }
132132
}
133133

134134
@Test
135135
fun `http errors`() =
136-
goldenStreamingFile("failure-http-error.txt", HttpStatusCode.PreconditionFailed) {
136+
goldenStreamingFile("streaming-failure-http-error.txt", HttpStatusCode.PreconditionFailed) {
137137
val responses = model.generateContentStream("prompt")
138138

139139
withTimeout(testTimeout) { shouldThrow<ServerException> { responses.collect() } }
140140
}
141141

142142
@Test
143143
fun `stopped for safety`() =
144-
goldenStreamingFile("failure-finish-reason-safety.txt") {
144+
goldenStreamingFile("streaming-failure-finish-reason-safety.txt") {
145145
val responses = model.generateContentStream("prompt")
146146

147147
withTimeout(testTimeout) {
@@ -152,7 +152,7 @@ internal class StreamingSnapshotTests {
152152

153153
@Test
154154
fun `citation parsed correctly`() =
155-
goldenStreamingFile("success-citations.txt") {
155+
goldenStreamingFile("streaming-success-citations.txt") {
156156
val responses = model.generateContentStream("prompt")
157157

158158
withTimeout(testTimeout) {
@@ -163,7 +163,7 @@ internal class StreamingSnapshotTests {
163163

164164
@Test
165165
fun `stopped for recitation`() =
166-
goldenStreamingFile("failure-recitation-no-content.txt") {
166+
goldenStreamingFile("streaming-failure-recitation-no-content.txt") {
167167
val responses = model.generateContentStream("prompt")
168168

169169
withTimeout(testTimeout) {
@@ -174,39 +174,39 @@ internal class StreamingSnapshotTests {
174174

175175
@Test
176176
fun `image rejected`() =
177-
goldenStreamingFile("failure-image-rejected.txt", HttpStatusCode.BadRequest) {
177+
goldenStreamingFile("streaming-failure-image-rejected.txt", HttpStatusCode.BadRequest) {
178178
val responses = model.generateContentStream("prompt")
179179

180180
withTimeout(testTimeout) { shouldThrow<ServerException> { responses.collect() } }
181181
}
182182

183183
@Test
184184
fun `unknown model`() =
185-
goldenStreamingFile("failure-unknown-model.txt", HttpStatusCode.NotFound) {
185+
goldenStreamingFile("streaming-failure-unknown-model.txt", HttpStatusCode.NotFound) {
186186
val responses = model.generateContentStream("prompt")
187187

188188
withTimeout(testTimeout) { shouldThrow<ServerException> { responses.collect() } }
189189
}
190190

191191
@Test
192192
fun `invalid api key`() =
193-
goldenStreamingFile("failure-api-key.txt", HttpStatusCode.BadRequest) {
193+
goldenStreamingFile("streaming-failure-api-key.txt", HttpStatusCode.BadRequest) {
194194
val responses = model.generateContentStream("prompt")
195195

196196
withTimeout(testTimeout) { shouldThrow<InvalidAPIKeyException> { responses.collect() } }
197197
}
198198

199199
@Test
200200
fun `invalid json`() =
201-
goldenStreamingFile("failure-invalid-json.txt") {
201+
goldenStreamingFile("streaming-failure-invalid-json.txt") {
202202
val responses = model.generateContentStream("prompt")
203203

204204
withTimeout(testTimeout) { shouldThrow<SerializationException> { responses.collect() } }
205205
}
206206

207207
@Test
208208
fun `malformed content`() =
209-
goldenStreamingFile("failure-malformed-content.txt") {
209+
goldenStreamingFile("streaming-failure-malformed-content.txt") {
210210
val responses = model.generateContentStream("prompt")
211211

212212
withTimeout(testTimeout) { shouldThrow<SerializationException> { responses.collect() } }

0 commit comments

Comments
 (0)