1 //===-- JSONTest.cpp ------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8
9 #include "gmock/gmock.h"
10 #include "gtest/gtest.h"
11
12 #include "JSON.h"
13
14 template <typename T>
TestJSON(JSONValue * json_val,const std::function<void (T &)> & test_func)15 void TestJSON(JSONValue *json_val, const std::function<void(T &)> &test_func) {
16 ASSERT_THAT(json_val, testing::NotNull());
17 ASSERT_TRUE(T::classof(json_val));
18 test_func(static_cast<T &>(*json_val));
19 }
20
ParseJSON(const char * json_string)21 JSONValue::SP ParseJSON(const char *json_string) {
22 return JSONParser(json_string).ParseJSONValue();
23 }
24
25 template <typename T>
ParseAndTestJSON(const char * json_string,const std::function<void (T &)> & test_func=[](T &){})26 void ParseAndTestJSON(
27 const char *json_string,
28 const std::function<void(T &)> &test_func = [](T &) {}) {
29 auto json_val = ParseJSON(json_string);
30 TestJSON<T>(json_val.get(), test_func);
31 }
32
TEST(JSON,Parse)33 TEST(JSON, Parse) {
34 ParseAndTestJSON<JSONString>("\"foo\"", [](JSONString &string_val) {
35 EXPECT_EQ(string_val.GetData(), "foo");
36 });
37 EXPECT_THAT(ParseJSON("\"foo"), testing::IsNull());
38 ParseAndTestJSON<JSONNumber>("3", [](JSONNumber &number_val) {
39 EXPECT_EQ(number_val.GetAsSigned(), 3);
40 EXPECT_EQ(number_val.GetAsUnsigned(), 3u);
41 EXPECT_EQ(number_val.GetAsDouble(), 3.0);
42 });
43 ParseAndTestJSON<JSONNumber>("-5", [](JSONNumber &number_val) {
44 EXPECT_EQ(number_val.GetAsSigned(), -5);
45 EXPECT_EQ(number_val.GetAsDouble(), -5.0);
46 });
47 ParseAndTestJSON<JSONNumber>("-6.4", [](JSONNumber &number_val) {
48 EXPECT_EQ(number_val.GetAsSigned(), -6);
49 EXPECT_EQ(number_val.GetAsDouble(), -6.4);
50 });
51 EXPECT_THAT(ParseJSON("-1.2.3"), testing::IsNull());
52 ParseAndTestJSON<JSONTrue>("true");
53 ParseAndTestJSON<JSONFalse>("false");
54 ParseAndTestJSON<JSONNull>("null");
55 ParseAndTestJSON<JSONObject>(
56 "{ \"key1\": 4, \"key2\": \"foobar\" }", [](JSONObject &obj_val) {
57 TestJSON<JSONNumber>(obj_val.GetObject("key1").get(),
58 [](JSONNumber &number_val) {
59 EXPECT_EQ(number_val.GetAsSigned(), 4);
60 EXPECT_EQ(number_val.GetAsUnsigned(), 4u);
61 EXPECT_EQ(number_val.GetAsDouble(), 4.0);
62 });
63 TestJSON<JSONString>(obj_val.GetObject("key2").get(),
64 [](JSONString &string_val) {
65 EXPECT_EQ(string_val.GetData(), "foobar");
66 });
67 });
68 ParseAndTestJSON<JSONArray>("[1, \"bar\", 3.14]", [](JSONArray &array_val) {
69 EXPECT_EQ(array_val.GetNumElements(), 3u);
70 TestJSON<JSONNumber>(array_val.GetObject(0).get(),
71 [](JSONNumber &number_val) {
72 EXPECT_EQ(number_val.GetAsSigned(), 1);
73 EXPECT_EQ(number_val.GetAsUnsigned(), 1u);
74 EXPECT_EQ(number_val.GetAsDouble(), 1.0);
75 });
76 TestJSON<JSONString>(
77 array_val.GetObject(1).get(),
78 [](JSONString &string_val) { EXPECT_EQ(string_val.GetData(), "bar"); });
79 TestJSON<JSONNumber>(array_val.GetObject(2).get(),
80 [](JSONNumber &number_val) {
81 EXPECT_EQ(number_val.GetAsSigned(), 3);
82 EXPECT_EQ(number_val.GetAsUnsigned(), 3u);
83 EXPECT_EQ(number_val.GetAsDouble(), 3.14);
84 });
85 });
86 ParseAndTestJSON<JSONArray>("[]", [](JSONArray &array_val) {
87 EXPECT_EQ(array_val.GetNumElements(), 0u);
88 });
89 }
90