@@ -57,6 +57,9 @@ void testRepositoriesGetAll() {
5757 void testCreateDeleteCollection () {
5858 final var controller = new GroundingController ();
5959
60+ // (0) DELETE OLD DOCUMENTS
61+ this .deleteOldDocuments ();
62+
6063 // (1) CREATE COLLECTION
6164 final var collectionId = controller .createCollection (JSON_FORMAT );
6265 final var collectionUuid = UUID .fromString (collectionId );
@@ -65,7 +68,7 @@ void testCreateDeleteCollection() {
6568 // (1.1) TEST COLLECTION LOOKUP
6669 this .testCollectionsGetAll ();
6770
68- // (2) SANITY CHECK: NO DOCUMENTS
71+ // (2) SANITY CHECK: NO DOCUMENTS IN COLLECTION
6972 final var documentsEmpty = controller .getDocumentsByCollectionId (collectionUuid , JSON_FORMAT );
7073 assertThat (documentsEmpty ).isInstanceOf (Documents .class );
7174 assertThat (((Documents ) documentsEmpty ).getCount ()).isEqualTo (0 );
@@ -143,4 +146,20 @@ void testGetDocumentById(UUID collectionId, UUID documentId) {
143146 assertThat (chunk .getMetadata ()).isNotNull ().isNotEmpty ();
144147 }
145148 }
149+
150+ void deleteOldDocuments () {
151+ final var controller = new GroundingController ();
152+
153+ final var result = controller .getAllCollections (JSON_FORMAT );
154+ assertThat (result ).isInstanceOf (CollectionsListResponse .class );
155+ final var collectionsList = ((CollectionsListResponse ) result ).getResources ();
156+
157+ for (final var collection : collectionsList ) {
158+ if ("e2e" .equals (collection .getTitle ())) {
159+ final var deletion = controller .deleteDocuments (collection .getId (), JSON_FORMAT );
160+ assertThat (deletion ).isInstanceOf (OpenApiResponse .class );
161+ assertThat (((OpenApiResponse ) deletion ).getStatusCode ()).isEqualTo (202 );
162+ }
163+ }
164+ }
146165}
0 commit comments