Skip to content

Commit ac7d3d5

Browse files
test: Change AI and Events tests to use fields instead of rawData (box/box-codegen#852) (#1475)
1 parent 77461b2 commit ac7d3d5

File tree

3 files changed

+12
-14
lines changed

3 files changed

+12
-14
lines changed

.codegen.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
{ "engineHash": "504b5a7", "specHash": "62fdfd1", "version": "0.1.0" }
1+
{ "engineHash": "10c68c7", "specHash": "62fdfd1", "version": "0.1.0" }

src/intTest/java/com/box/sdkgen/ai/AiITest.java

Lines changed: 10 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import static com.box.sdkgen.internal.utils.UtilsManager.dateTimeFromString;
77
import static com.box.sdkgen.internal.utils.UtilsManager.delayInSeconds;
88
import static com.box.sdkgen.internal.utils.UtilsManager.getUuid;
9-
import static com.box.sdkgen.internal.utils.UtilsManager.getValueFromObjectRawData;
109
import static com.box.sdkgen.internal.utils.UtilsManager.stringToByteStream;
1110

1211
import com.box.sdkgen.client.BoxClient;
@@ -338,13 +337,12 @@ public void testAiExtractStructuredWithFields() {
338337
.build()))
339338
.aiAgent(agentIgnoringOverridingEmbeddingsModel)
340339
.build());
341-
assert convertToString(getValueFromObjectRawData(response, "answer.hobby"))
340+
assert convertToString(response.getAnswer().get("hobby"))
342341
.equals(convertToString(Arrays.asList("guitar")));
343-
assert convertToString(getValueFromObjectRawData(response, "answer.firstName")).equals("John");
344-
assert convertToString(getValueFromObjectRawData(response, "answer.lastName")).equals("Doe");
345-
assert convertToString(getValueFromObjectRawData(response, "answer.dateOfBirth"))
346-
.equals("1990-07-04");
347-
assert convertToString(getValueFromObjectRawData(response, "answer.age")).equals("34");
342+
assert convertToString(response.getAnswer().get("firstName")).equals("John");
343+
assert convertToString(response.getAnswer().get("lastName")).equals("Doe");
344+
assert convertToString(response.getAnswer().get("dateOfBirth")).equals("1990-07-04");
345+
assert convertToString(response.getAnswer().get("age")).equals("34");
348346
assert response.getCompletionReason().equals("done");
349347
client.getFiles().deleteFileById(file.getId());
350348
}
@@ -420,12 +418,11 @@ public void testAiExtractStructuredWithMetadataTemplate() {
420418
.scope("enterprise")
421419
.build())
422420
.build());
423-
assert convertToString(getValueFromObjectRawData(response, "answer.firstName")).equals("John");
424-
assert convertToString(getValueFromObjectRawData(response, "answer.lastName")).equals("Doe");
425-
assert convertToString(getValueFromObjectRawData(response, "answer.dateOfBirth"))
426-
.equals("1990-07-04T00:00:00Z");
427-
assert convertToString(getValueFromObjectRawData(response, "answer.age")).equals("34");
428-
assert convertToString(getValueFromObjectRawData(response, "answer.hobby"))
421+
assert convertToString(response.getAnswer().get("firstName")).equals("John");
422+
assert convertToString(response.getAnswer().get("lastName")).equals("Doe");
423+
assert convertToString(response.getAnswer().get("dateOfBirth")).equals("1990-07-04T00:00:00Z");
424+
assert convertToString(response.getAnswer().get("age")).equals("34");
425+
assert convertToString(response.getAnswer().get("hobby"))
429426
.equals(convertToString(Arrays.asList("guitar")));
430427
assert response.getCompletionReason().equals("done");
431428
client

src/intTest/java/com/box/sdkgen/events/EventsITest.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ public void testEventUpload() {
4646
assert events.getEntries().size() > 0;
4747
Event firstEvent = events.getEntries().get(0);
4848
assert convertToString(firstEvent.getEventType()).equals("UPLOAD");
49+
assert convertToString(firstEvent.getAdditionalDetails().get("hash_type")).equals("sha1");
4950
EventSource source = firstEvent.getSource().getEventSource();
5051
assert convertToString(source.getItemType()).equals("file")
5152
|| convertToString(source.getItemType()).equals("folder");

0 commit comments

Comments
 (0)