fix tests

Author: stremovsky

src/test/java/org/databunker/DatabunkerproApiPciTest.java

@@ -16,6 +16,7 @@
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import org.databunker.options.TokenOptions;
 
 import static org.junit.Assert.*;
 
@@ -80,10 +81,11 @@ public void testCreditCardTokenization() throws IOException {
 
         // Step 1: Tokenize a credit card number
         String creditCardNumber = "5467047429390590";
-        Map<String, Object> options = new HashMap<>();
-        options.put("slidingtime", "1d");
-        options.put("finaltime", "12m");
-        options.put("unique", true);
+        TokenOptions options = new TokenOptions.Builder()
+            .slidingtime("1d")
+            .finaltime("12m")
+            .unique(true)
+            .build();
 
         Map<String, Object> tokenResult = api.createToken("creditcard", creditCardNumber, options, null);
         assertNotNull(tokenResult);
@@ -160,10 +162,11 @@ public void testBulkCreditCardTokenization() throws IOException {
             records[i] = record;
         }
 
-        Map<String, Object> options = new HashMap<>();
-        options.put("slidingtime", "30d");
-        options.put("finaltime", "12m");
-        options.put("unique", true);
+        TokenOptions options = new TokenOptions.Builder()
+            .slidingtime("30d")
+            .finaltime("12m")
+            .unique(true)
+            .build();
 
         Map<String, Object> bulkTokens = api.createTokensBulk(records, options, null);
         System.out.println("Bulk tokens: " + bulkTokens);
@@ -244,7 +247,7 @@ public void testBulkCreditCardTokenization() throws IOException {
 
         try {
             // Test with empty records array
-            Map<String, Object> emptyRecordsResponse = api.createTokensBulk(new Map[0], options, null);
+            Map<String, Object> emptyRecordsResponse = api.createTokensBulk(new Map[0], (TokenOptions) null, null);
             if (emptyRecordsResponse != null) {
                 assertEquals("error", emptyRecordsResponse.get("status"));
                 assertNotNull(emptyRecordsResponse.get("message"));

src/test/java/org/databunker/DatabunkerproApiTest.java

@@ -4,8 +4,6 @@
 import org.apache.http.HttpEntity;
 import org.apache.http.client.methods.CloseableHttpResponse;
 import org.apache.http.client.methods.HttpGet;
-import org.apache.http.client.methods.HttpPost;
-import org.apache.http.entity.StringEntity;
 import org.apache.http.impl.client.CloseableHttpClient;
 import org.apache.http.impl.client.HttpClients;
 import org.apache.http.util.EntityUtils;
@@ -18,6 +16,8 @@
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Random;
+import org.databunker.options.BasicOptions;
+import org.databunker.options.SharedRecordOptions;
 
 import static org.junit.Assert.*;
 
@@ -198,10 +198,10 @@ public void testSessionManagement() throws IOException {
         );
 
         // Create/Update session with options
-        Map<String, Object> options = Map.of(
-            "slidingtime", "1h",
-            "finaltime", "24h"
-        );
+        BasicOptions options = new BasicOptions.Builder()
+            .slidingtime("1h")
+            .finaltime("24h")
+            .build();
         Map<String, Object> upsertResult = api.upsertSession(sessionuuid, sessionData, options, null);
         assertNotNull(upsertResult);
         assertEquals("ok", upsertResult.get("status"));
@@ -268,11 +268,11 @@ public void testSharedRecordManagement() throws IOException {
         api.createUser(userData, null, null);
 
         // Create shared record
-        Map<String, Object> options = Map.of(
-            "fields", "name,email",
-            "partner", "test-partner",
-            "finaltime", "1d"
-        );
+        SharedRecordOptions options = new SharedRecordOptions.Builder()
+            .fields("name,email")
+            .partner("test-partner")
+            .finaltime("1d")
+            .build();
         Map<String, Object> createResult = api.createSharedRecord("email", email, options, null);
         assertNotNull(createResult);
         assertEquals("ok", createResult.get("status"));