Skip to content

Commit 4f7bd7e

Browse files
saileshwar-skyflowsaileshwar-skyflow
committed
SK-2291: add unit test cases
1 parent 7b1b3bc commit 4f7bd7e

File tree

6 files changed

+182
-10
lines changed

6 files changed

+182
-10
lines changed

v3/test/java/com/skyflow/SkyflowTests.java renamed to v3/src/test/java/com/skyflow/SkyflowTests.java

File renamed without changes.

v3/test/java/com/skyflow/VaultClientTests.java renamed to v3/src/test/java/com/skyflow/VaultClientTests.java

File renamed without changes.

v3/test/java/com/skyflow/utils/UtilsTests.java renamed to v3/src/test/java/com/skyflow/utils/UtilsTests.java

File renamed without changes.

v3/test/java/com/skyflow/vault/controller/VaultControllerTests.java renamed to v3/src/test/java/com/skyflow/vault/controller/VaultControllerTests.java

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -61,15 +61,6 @@ private void writeEnv(String content) {
6161
} catch (IOException e) {
6262
throw new RuntimeException(e);
6363
}
64-
// Print the contents of the .env file
65-
try (Scanner scanner = new Scanner(envFile)) {
66-
System.out.println("Current .env contents:");
67-
while (scanner.hasNextLine()) {
68-
System.out.println(scanner.nextLine());
69-
}
70-
} catch (IOException e) {
71-
System.out.println("Could not read .env file: " + e.getMessage());
72-
}
7364
}
7465

7566
private VaultController createController() throws SkyflowException {
@@ -395,7 +386,7 @@ public void testFractionalLastBatch() throws Exception {
395386

396387
// Last batch should have 50 records, concurrency should be 101
397388
assertEquals(100, getPrivateInt(controller, "insertBatchSize"));
398-
assertEquals(Constants.INSERT_CONCURRENCY_LIMIT.intValue(), getPrivateInt(controller, "insertConcurrencyLimit"));
389+
assertEquals(10, getPrivateInt(controller, "insertConcurrencyLimit"));
399390
}
400391

401392
private int getPrivateInt(Object obj, String field) throws Exception {

‎v3/src/test/java/com/skyflow/vault/data/DetokenizeRequestTests.java‎

Lines changed: 181 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,181 @@
1+
package com.skyflow.vault.data;
2+
3+
import com.skyflow.errors.ErrorCode;
4+
import com.skyflow.errors.ErrorMessage;
5+
import com.skyflow.errors.SkyflowException;
6+
import com.skyflow.utils.Constants;
7+
import com.skyflow.utils.Utils;
8+
import com.skyflow.utils.validations.Validations;
9+
import org.junit.Assert;
10+
import org.junit.Test;
11+
12+
import java.util.ArrayList;
13+
import java.util.Arrays;
14+
import java.util.Collections;
15+
import java.util.List;
16+
17+
public class DetokenizeRequestTests {
18+
@Test
19+
public void testDetokenizeRequestBuilderAndGetters() {
20+
List<String> tokens = Arrays.asList("token1", "token2");
21+
TokenGroupRedactions group = TokenGroupRedactions.builder()
22+
.tokenGroupName("group1")
23+
.redaction("PLAIN_TEXT")
24+
.build();
25+
List<TokenGroupRedactions> groupRedactions = Collections.singletonList(group);
26+
27+
DetokenizeRequest request = DetokenizeRequest.builder()
28+
.tokens(tokens)
29+
.tokenGroupRedactions(groupRedactions)
30+
.build();
31+
32+
Assert.assertEquals(tokens, request.getTokens());
33+
Assert.assertEquals(groupRedactions, request.getTokenGroupRedactions());
34+
}
35+
36+
@Test
37+
public void testTokenGroupRedactionsBuilderAndGetters() {
38+
TokenGroupRedactions group = TokenGroupRedactions.builder()
39+
.tokenGroupName("groupA")
40+
.redaction("MASKED")
41+
.build();
42+
Assert.assertEquals("groupA", group.getTokenGroupName());
43+
Assert.assertEquals("MASKED", group.getRedaction());
44+
}
45+
46+
@Test
47+
public void testValidateDetokenizeRequestValid() {
48+
try {
49+
List<String> tokens = Arrays.asList("token1", "token2");
50+
TokenGroupRedactions group = TokenGroupRedactions.builder()
51+
.tokenGroupName("group1")
52+
.redaction("PLAIN_TEXT")
53+
.build();
54+
List<TokenGroupRedactions> groupRedactions = Collections.singletonList(group);
55+
DetokenizeRequest request = DetokenizeRequest.builder()
56+
.tokens(tokens)
57+
.tokenGroupRedactions(groupRedactions)
58+
.build();
59+
Validations.validateDetokenizeRequest(request);
60+
} catch (SkyflowException e) {
61+
Assert.fail("Should not have thrown exception for valid request");
62+
}
63+
}
64+
65+
@Test
66+
public void testValidateDetokenizeRequestNull() {
67+
try {
68+
Validations.validateDetokenizeRequest(null);
69+
Assert.fail("Should have thrown exception for null request");
70+
} catch (SkyflowException e) {
71+
Assert.assertEquals(ErrorCode.INVALID_INPUT.getCode(), e.getHttpCode());
72+
Assert.assertEquals(ErrorMessage.DetokenizeRequestNull.getMessage(), e.getMessage());
73+
}
74+
}
75+
76+
@Test
77+
public void testValidateDetokenizeRequestEmptyTokens() {
78+
try {
79+
DetokenizeRequest request = DetokenizeRequest.builder()
80+
.tokens(new ArrayList<>())
81+
.build();
82+
Validations.validateDetokenizeRequest(request);
83+
Assert.fail("Should have thrown exception for empty tokens");
84+
} catch (SkyflowException e) {
85+
Assert.assertEquals(ErrorCode.INVALID_INPUT.getCode(), e.getHttpCode());
86+
Assert.assertEquals(ErrorMessage.EmptyDetokenizeData.getMessage(), e.getMessage());
87+
}
88+
}
89+
90+
@Test
91+
public void testValidateDetokenizeRequestNullOrEmptyToken() {
92+
try {
93+
List<String> tokens = Arrays.asList("token1", null, "");
94+
DetokenizeRequest request = DetokenizeRequest.builder()
95+
.tokens(tokens)
96+
.build();
97+
Validations.validateDetokenizeRequest(request);
98+
Assert.fail("Should have thrown exception for null/empty token");
99+
} catch (SkyflowException e) {
100+
Assert.assertEquals(ErrorCode.INVALID_INPUT.getCode(), e.getHttpCode());
101+
Assert.assertEquals(ErrorMessage.EmptyTokenInDetokenizeData.getMessage(), e.getMessage());
102+
}
103+
}
104+
105+
@Test
106+
public void testValidateDetokenizeRequestTokensSizeExceed() {
107+
try {
108+
List<String> tokens = new ArrayList<>();
109+
for (int i = 0; i < 10001; i++) {
110+
tokens.add("token" + i);
111+
}
112+
DetokenizeRequest request = DetokenizeRequest.builder()
113+
.tokens(tokens)
114+
.build();
115+
Validations.validateDetokenizeRequest(request);
116+
Assert.fail("Should have thrown exception for tokens size exceed");
117+
} catch (SkyflowException e) {
118+
Assert.assertEquals(ErrorCode.INVALID_INPUT.getCode(), e.getHttpCode());
119+
Assert.assertEquals(ErrorMessage.TokensSizeExceedError.getMessage(), e.getMessage());
120+
}
121+
}
122+
123+
@Test
124+
public void testValidateDetokenizeRequestNullGroupRedaction() {
125+
try {
126+
List<String> tokens = Arrays.asList("token1");
127+
List<TokenGroupRedactions> groupRedactions = Arrays.asList((TokenGroupRedactions) null);
128+
DetokenizeRequest request = DetokenizeRequest.builder()
129+
.tokens(tokens)
130+
.tokenGroupRedactions(groupRedactions)
131+
.build();
132+
Validations.validateDetokenizeRequest(request);
133+
Assert.fail("Should have thrown exception for null group redaction");
134+
} catch (SkyflowException e) {
135+
Assert.assertEquals(ErrorCode.INVALID_INPUT.getCode(), e.getHttpCode());
136+
Assert.assertEquals(ErrorMessage.NullTokenGroupRedactions.getMessage(), e.getMessage());
137+
}
138+
}
139+
140+
@Test
141+
public void testValidateDetokenizeRequestEmptyGroupName() {
142+
try {
143+
List<String> tokens = Arrays.asList("token1");
144+
TokenGroupRedactions group = TokenGroupRedactions.builder()
145+
.tokenGroupName("")
146+
.redaction("PLAIN_TEXT")
147+
.build();
148+
List<TokenGroupRedactions> groupRedactions = Collections.singletonList(group);
149+
DetokenizeRequest request = DetokenizeRequest.builder()
150+
.tokens(tokens)
151+
.tokenGroupRedactions(groupRedactions)
152+
.build();
153+
Validations.validateDetokenizeRequest(request);
154+
Assert.fail("Should have thrown exception for empty group name");
155+
} catch (SkyflowException e) {
156+
Assert.assertEquals(ErrorCode.INVALID_INPUT.getCode(), e.getHttpCode());
157+
Assert.assertEquals(ErrorMessage.NullTokenGroupNameInTokenGroup.getMessage(), e.getMessage());
158+
}
159+
}
160+
161+
@Test
162+
public void testValidateDetokenizeRequestEmptyRedaction() {
163+
try {
164+
List<String> tokens = Arrays.asList("token1");
165+
TokenGroupRedactions group = TokenGroupRedactions.builder()
166+
.tokenGroupName("group1")
167+
.redaction("")
168+
.build();
169+
List<TokenGroupRedactions> groupRedactions = Collections.singletonList(group);
170+
DetokenizeRequest request = DetokenizeRequest.builder()
171+
.tokens(tokens)
172+
.tokenGroupRedactions(groupRedactions)
173+
.build();
174+
Validations.validateDetokenizeRequest(request);
175+
Assert.fail("Should have thrown exception for empty redaction");
176+
} catch (SkyflowException e) {
177+
Assert.assertEquals(ErrorCode.INVALID_INPUT.getCode(), e.getHttpCode());
178+
Assert.assertEquals(ErrorMessage.NullRedactionInTokenGroup.getMessage(), e.getMessage());
179+
}
180+
}
181+
}

v3/test/java/com/skyflow/vault/data/InsertTests.java renamed to v3/src/test/java/com/skyflow/vault/data/InsertTests.java

File renamed without changes.

0 commit comments

Comments
 (0)