@@ -43,7 +43,7 @@ internal class StreamingSnapshotTests {
43
43
44
44
@Test
45
45
fun `short reply` () =
46
- goldenStreamingFile(" success-basic-reply-short.txt" ) {
46
+ goldenStreamingFile(" streaming- success-basic-reply-short.txt" ) {
47
47
val responses = model.generateContentStream(" prompt" )
48
48
49
49
withTimeout(testTimeout) {
@@ -57,7 +57,7 @@ internal class StreamingSnapshotTests {
57
57
58
58
@Test
59
59
fun `long reply` () =
60
- goldenStreamingFile(" success-basic-reply-long.txt" ) {
60
+ goldenStreamingFile(" streaming- success-basic-reply-long.txt" ) {
61
61
val responses = model.generateContentStream(" prompt" )
62
62
63
63
withTimeout(testTimeout) {
@@ -73,7 +73,7 @@ internal class StreamingSnapshotTests {
73
73
74
74
@Test
75
75
fun `unknown enum` () =
76
- goldenStreamingFile(" success-unknown-enum.txt" ) {
76
+ goldenStreamingFile(" streaming- success-unknown-enum.txt" ) {
77
77
val responses = model.generateContentStream(" prompt" )
78
78
79
79
withTimeout(testTimeout) {
@@ -88,7 +88,7 @@ internal class StreamingSnapshotTests {
88
88
89
89
@Test
90
90
fun `unknown enum in finish reason` () =
91
- goldenStreamingFile(" failure-unknown-finish-enum.txt" ) {
91
+ goldenStreamingFile(" streaming- failure-unknown-finish-enum.txt" ) {
92
92
val responses = model.generateContentStream(" prompt" )
93
93
94
94
withTimeout(testTimeout) {
@@ -99,7 +99,7 @@ internal class StreamingSnapshotTests {
99
99
100
100
@Test
101
101
fun `quotes escaped` () =
102
- goldenStreamingFile(" success-quotes-escaped.txt" ) {
102
+ goldenStreamingFile(" streaming- success-quotes-escaped.txt" ) {
103
103
val responses = model.generateContentStream(" prompt" )
104
104
105
105
withTimeout(testTimeout) {
@@ -114,7 +114,7 @@ internal class StreamingSnapshotTests {
114
114
115
115
@Test
116
116
fun `prompt blocked for safety` () =
117
- goldenStreamingFile(" failure-prompt-blocked-safety.txt" ) {
117
+ goldenStreamingFile(" streaming- failure-prompt-blocked-safety.txt" ) {
118
118
val responses = model.generateContentStream(" prompt" )
119
119
120
120
withTimeout(testTimeout) {
@@ -125,23 +125,23 @@ internal class StreamingSnapshotTests {
125
125
126
126
@Test
127
127
fun `empty content` () =
128
- goldenStreamingFile(" failure-empty-content.txt" ) {
128
+ goldenStreamingFile(" streaming- failure-empty-content.txt" ) {
129
129
val responses = model.generateContentStream(" prompt" )
130
130
131
131
withTimeout(testTimeout) { shouldThrow<SerializationException > { responses.collect() } }
132
132
}
133
133
134
134
@Test
135
135
fun `http errors` () =
136
- goldenStreamingFile(" failure-http-error.txt" , HttpStatusCode .PreconditionFailed ) {
136
+ goldenStreamingFile(" streaming- failure-http-error.txt" , HttpStatusCode .PreconditionFailed ) {
137
137
val responses = model.generateContentStream(" prompt" )
138
138
139
139
withTimeout(testTimeout) { shouldThrow<ServerException > { responses.collect() } }
140
140
}
141
141
142
142
@Test
143
143
fun `stopped for safety` () =
144
- goldenStreamingFile(" failure-finish-reason-safety.txt" ) {
144
+ goldenStreamingFile(" streaming- failure-finish-reason-safety.txt" ) {
145
145
val responses = model.generateContentStream(" prompt" )
146
146
147
147
withTimeout(testTimeout) {
@@ -152,7 +152,7 @@ internal class StreamingSnapshotTests {
152
152
153
153
@Test
154
154
fun `citation parsed correctly` () =
155
- goldenStreamingFile(" success-citations.txt" ) {
155
+ goldenStreamingFile(" streaming- success-citations.txt" ) {
156
156
val responses = model.generateContentStream(" prompt" )
157
157
158
158
withTimeout(testTimeout) {
@@ -163,7 +163,7 @@ internal class StreamingSnapshotTests {
163
163
164
164
@Test
165
165
fun `stopped for recitation` () =
166
- goldenStreamingFile(" failure-recitation-no-content.txt" ) {
166
+ goldenStreamingFile(" streaming- failure-recitation-no-content.txt" ) {
167
167
val responses = model.generateContentStream(" prompt" )
168
168
169
169
withTimeout(testTimeout) {
@@ -174,39 +174,39 @@ internal class StreamingSnapshotTests {
174
174
175
175
@Test
176
176
fun `image rejected` () =
177
- goldenStreamingFile(" failure-image-rejected.txt" , HttpStatusCode .BadRequest ) {
177
+ goldenStreamingFile(" streaming- failure-image-rejected.txt" , HttpStatusCode .BadRequest ) {
178
178
val responses = model.generateContentStream(" prompt" )
179
179
180
180
withTimeout(testTimeout) { shouldThrow<ServerException > { responses.collect() } }
181
181
}
182
182
183
183
@Test
184
184
fun `unknown model` () =
185
- goldenStreamingFile(" failure-unknown-model.txt" , HttpStatusCode .NotFound ) {
185
+ goldenStreamingFile(" streaming- failure-unknown-model.txt" , HttpStatusCode .NotFound ) {
186
186
val responses = model.generateContentStream(" prompt" )
187
187
188
188
withTimeout(testTimeout) { shouldThrow<ServerException > { responses.collect() } }
189
189
}
190
190
191
191
@Test
192
192
fun `invalid api key` () =
193
- goldenStreamingFile(" failure-api-key.txt" , HttpStatusCode .BadRequest ) {
193
+ goldenStreamingFile(" streaming- failure-api-key.txt" , HttpStatusCode .BadRequest ) {
194
194
val responses = model.generateContentStream(" prompt" )
195
195
196
196
withTimeout(testTimeout) { shouldThrow<InvalidAPIKeyException > { responses.collect() } }
197
197
}
198
198
199
199
@Test
200
200
fun `invalid json` () =
201
- goldenStreamingFile(" failure-invalid-json.txt" ) {
201
+ goldenStreamingFile(" streaming- failure-invalid-json.txt" ) {
202
202
val responses = model.generateContentStream(" prompt" )
203
203
204
204
withTimeout(testTimeout) { shouldThrow<SerializationException > { responses.collect() } }
205
205
}
206
206
207
207
@Test
208
208
fun `malformed content` () =
209
- goldenStreamingFile(" failure-malformed-content.txt" ) {
209
+ goldenStreamingFile(" streaming- failure-malformed-content.txt" ) {
210
210
val responses = model.generateContentStream(" prompt" )
211
211
212
212
withTimeout(testTimeout) { shouldThrow<SerializationException > { responses.collect() } }
0 commit comments