Skip to content

Commit dfb2999

Browse files
committed
test: add jsonassert
1 parent 673ac45 commit dfb2999

2 files changed

Lines changed: 28 additions & 21 deletions

File tree

src/test/java/io/odpf/depot/message/field/proto/MapFieldTest.java

Lines changed: 9 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@
66
import com.google.protobuf.Value;
77
import io.odpf.depot.TestMapMessage;
88
import io.odpf.depot.TestMessage;
9-
import org.json.JSONObject;
109
import org.junit.Assert;
1110
import org.junit.Test;
11+
import org.skyscreamer.jsonassert.JSONAssert;
1212

1313
public class MapFieldTest {
1414

@@ -28,12 +28,10 @@ public void shouldReturnMap() {
2828
Object metadata = message.getField(message.getDescriptorForType().findFieldByName("metadata"));
2929
MapField field1 = new MapField(currentState);
3030
MapField field2 = new MapField(metadata);
31-
Assert.assertEquals(
32-
new JSONObject("{\"country\":\"japan\",\"city\":\"tokyo\",\"currency\":\"jpy\"}").toString(),
33-
new JSONObject(field1.getString()).toString());
34-
Assert.assertEquals(
35-
new JSONObject("{\"23\":{\"order_url\":\"\",\"order_number\":\"order1\",\"order_details\":\"\"},\"10\":{\"order_url\":\"\",\"order_number\":\"order2\",\"order_details\":\"\"}}").toString(),
36-
new JSONObject(field2.getString()).toString());
31+
String expectedJson = "{\"country\":\"japan\",\"city\":\"tokyo\",\"currency\":\"jpy\"}";
32+
JSONAssert.assertEquals(expectedJson, field1.getString(), true);
33+
expectedJson = "{\"23\":{\"order_url\":\"\",\"order_number\":\"order1\",\"order_details\":\"\"},\"10\":{\"order_url\":\"\",\"order_number\":\"order2\",\"order_details\":\"\"}}";
34+
JSONAssert.assertEquals(expectedJson, field2.getString(), true);
3735
}
3836

3937
@Test
@@ -45,9 +43,8 @@ public void shouldReturnDurationMap() {
4543
.build();
4644
Object durations = message.getField(message.getDescriptorForType().findFieldByName("durations"));
4745
MapField field1 = new MapField(durations);
48-
Assert.assertEquals(
49-
new JSONObject("{\"d1\":\"1234s\",\"d2\":\"1200.000000123s\"}").toString(),
50-
new JSONObject(field1.getString()).toString());
46+
String expectedJson = "{\"d1\":\"1234s\",\"d2\":\"1200.000000123s\"}";
47+
JSONAssert.assertEquals(expectedJson, field1.getString(), true);
5148
}
5249

5350
@Test
@@ -59,9 +56,8 @@ public void shouldReturnTimestampMap() {
5956
.build();
6057
Object timeStamps = message.getField(message.getDescriptorForType().findFieldByName("time_stamps"));
6158
MapField field1 = new MapField(timeStamps);
62-
Assert.assertEquals(
63-
new JSONObject("{\"ts2\":\"2022-12-02T06:46:34Z\",\"ts1\":\"2022-12-02T06:29:54Z\"}").toString(),
64-
new JSONObject(field1.getString()).toString());
59+
String expectedJson = "{\"ts2\":\"2022-12-02T06:46:34Z\",\"ts1\":\"2022-12-02T06:29:54Z\"}";
60+
JSONAssert.assertEquals(expectedJson, field1.getString(), true);
6561
}
6662

6763
@Test

src/test/java/io/odpf/depot/message/field/proto/MessageFieldTest.java

Lines changed: 19 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,8 @@
33
import com.google.protobuf.Timestamp;
44
import io.odpf.depot.TestMessage;
55
import io.odpf.depot.TestNestedRepeatedMessage;
6-
import org.json.JSONObject;
7-
import org.junit.Assert;
86
import org.junit.Test;
7+
import org.skyscreamer.jsonassert.JSONAssert;
98

109
public class MessageFieldTest {
1110

@@ -17,9 +16,8 @@ public void shouldReturnJsonStringForMessage() {
1716
.setOrderUrl("url")
1817
.build();
1918
MessageField field = new MessageField(message);
20-
Assert.assertEquals(
21-
new JSONObject("{\"order_number\":\"number\",\"order_url\":\"url\",\"order_details\":\"details\"}").toString(),
22-
new JSONObject(field.getString()).toString());
19+
String expectedJson = "{\"order_number\":\"number\",\"order_url\":\"url\",\"order_details\":\"details\"}";
20+
JSONAssert.assertEquals(expectedJson, field.getString(), true);
2321
}
2422

2523
@Test
@@ -77,8 +75,21 @@ public void shouldReturnMessageForRepeatedMessage() {
7775
+ " },\n"
7876
+ " \"number_field\": 10\n"
7977
+ "}\n";
80-
Assert.assertEquals(
81-
new JSONObject(expectedJson).toString(),
82-
new JSONObject(field.getString()).toString());
78+
JSONAssert.assertEquals(expectedJson, field.getString(), true);
79+
80+
expectedJson = "[\n"
81+
+ " {\n"
82+
+ " \"order_number\": \"number\",\n"
83+
+ " \"order_url\": \"url\",\n"
84+
+ " \"order_details\": \"details\"\n"
85+
+ " },\n"
86+
+ " {\n"
87+
+ " \"order_number\": \"o2\",\n"
88+
+ " \"order_url\": \"url2\",\n"
89+
+ " \"order_details\": \"d2\"\n"
90+
+ " }\n"
91+
+ "]";
92+
field = new MessageField(message.getField(message.getDescriptorForType().findFieldByName("repeated_message")));
93+
JSONAssert.assertEquals(expectedJson, field.getString(), true);
8394
}
8495
}

0 commit comments

Comments
 (0)