Skip to content

Commit 5655aeb

Browse files
committed
add: Collections tests
1 parent e2eed6d commit 5655aeb

File tree

2 files changed

+68
-1
lines changed

2 files changed

+68
-1
lines changed
Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
package org.typesense.api;
2+
3+
import junit.framework.TestCase;
4+
import org.junit.jupiter.api.Test;
5+
import org.typesense.model.CollectionResponse;
6+
import org.typesense.model.CollectionSchema;
7+
import org.typesense.model.Field;
8+
9+
import java.util.ArrayList;
10+
11+
12+
public class CollectionsTest extends TestCase {
13+
14+
public Client client;
15+
private Helper helper;
16+
17+
public void setUp() throws Exception {
18+
super.setUp();
19+
helper = new Helper();
20+
this.client = helper.getClient();
21+
}
22+
23+
public void tearDown() throws Exception {
24+
super.tearDown();
25+
helper.teardown();
26+
}
27+
28+
29+
@Test
30+
public void testRetrieveAllCollections() {
31+
helper.createTestCollection();
32+
CollectionResponse[] collectionResponses = client.collections().retrieve();
33+
for(CollectionResponse c:collectionResponses)
34+
System.out.println(c);
35+
}
36+
37+
@Test
38+
public void testRetrieveSingleCollection(){
39+
helper.createTestCollection();
40+
System.out.println(client.collections("books").retrieve());
41+
}
42+
43+
@Test
44+
public void testDeleteCollection(){
45+
helper.createTestCollection();
46+
System.out.println(client.collections("books").delete());
47+
}
48+
49+
@Test
50+
public void testCreateCollection(){
51+
52+
ArrayList<Field> fields = new ArrayList<>();
53+
fields.add(new Field().name("countryName").type(Field.TypeEnum.STRING));
54+
fields.add(new Field().name("capital").type(Field.TypeEnum.STRING));
55+
fields.add(new Field().name("gdp").type(Field.TypeEnum.INT32).facet(true));
56+
57+
CollectionSchema collectionSchema = new CollectionSchema();
58+
collectionSchema.name("Countries").fields(fields);
59+
60+
CollectionResponse cr = client.collections().create(collectionSchema);
61+
System.out.println(cr);
62+
}
63+
}

src/test/java/org/typesense/api/Helper.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
package org.typesense.api;
22

3+
import org.typesense.model.CollectionResponse;
34
import org.typesense.model.CollectionSchema;
45
import org.typesense.model.Field;
56
import org.typesense.resources.Node;
@@ -51,6 +52,9 @@ public Client getClient() {
5152
}
5253

5354
public void teardown() {
54-
client.collections("books").delete();
55+
CollectionResponse[] collectionResponses = client.collections().retrieve();
56+
for(CollectionResponse c:collectionResponses) {
57+
client.collections(c.getName()).delete();
58+
}
5559
}
5660
}

0 commit comments

Comments
 (0)