blob: a4dbb9fbea8cd28e52e571ccf25520257b6a58c5 [file] [log] [blame]
Tao Baod770d2e2016-10-03 15:26:06 -07001/*
2 * Copyright (C) 2009 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <string>
18
19#include <gtest/gtest.h>
20
21#include "edify/expr.h"
22
23static void expect(const char* expr_str, const char* expected) {
24 Expr* e;
Tao Bao39119ad2016-10-10 22:52:18 -070025 int error_count = 0;
26 EXPECT_EQ(0, parse_string(expr_str, &e, &error_count));
27 EXPECT_EQ(0, error_count);
Tao Baod770d2e2016-10-03 15:26:06 -070028
Tao Bao59dcb9c2016-10-03 18:06:46 -070029 State state(expr_str, nullptr);
Tao Baod770d2e2016-10-03 15:26:06 -070030
31 char* result = Evaluate(&state, e);
32
33 if (expected == nullptr) {
Tao Bao39119ad2016-10-10 22:52:18 -070034 EXPECT_EQ(nullptr, result);
Tao Baod770d2e2016-10-03 15:26:06 -070035 } else {
Tao Bao39119ad2016-10-10 22:52:18 -070036 EXPECT_STREQ(expected, result);
Tao Baod770d2e2016-10-03 15:26:06 -070037 }
38
Tao Baod770d2e2016-10-03 15:26:06 -070039 free(result);
40}
41
42class EdifyTest : public ::testing::Test {
43 protected:
44 virtual void SetUp() {
45 RegisterBuiltins();
Tao Baod770d2e2016-10-03 15:26:06 -070046 }
47};
48
49TEST_F(EdifyTest, parsing) {
50 expect("a", "a");
51 expect("\"a\"", "a");
52 expect("\"\\x61\"", "a");
53 expect("# this is a comment\n"
54 " a\n"
55 " \n",
56 "a");
57}
58
59TEST_F(EdifyTest, sequence) {
60 // sequence operator
61 expect("a; b; c", "c");
62}
63
64TEST_F(EdifyTest, concat) {
65 // string concat operator
66 expect("a + b", "ab");
67 expect("a + \n \"b\"", "ab");
68 expect("a + b +\nc\n", "abc");
69
70 // string concat function
71 expect("concat(a, b)", "ab");
72 expect("concat(a,\n \"b\")", "ab");
73 expect("concat(a + b,\nc,\"d\")", "abcd");
74 expect("\"concat\"(a + b,\nc,\"d\")", "abcd");
75}
76
77TEST_F(EdifyTest, logical) {
78 // logical and
79 expect("a && b", "b");
80 expect("a && \"\"", "");
81 expect("\"\" && b", "");
82 expect("\"\" && \"\"", "");
83 expect("\"\" && abort()", ""); // test short-circuiting
84 expect("t && abort()", nullptr);
85
86 // logical or
87 expect("a || b", "a");
88 expect("a || \"\"", "a");
89 expect("\"\" || b", "b");
90 expect("\"\" || \"\"", "");
91 expect("a || abort()", "a"); // test short-circuiting
92 expect("\"\" || abort()", NULL);
93
94 // logical not
95 expect("!a", "");
96 expect("! \"\"", "t");
97 expect("!!a", "t");
98}
99
100TEST_F(EdifyTest, precedence) {
101 // precedence
102 expect("\"\" == \"\" && b", "b");
103 expect("a + b == ab", "t");
104 expect("ab == a + b", "t");
105 expect("a + (b == ab)", "a");
106 expect("(ab == a) + b", "b");
107}
108
109TEST_F(EdifyTest, substring) {
110 // substring function
111 expect("is_substring(cad, abracadabra)", "t");
112 expect("is_substring(abrac, abracadabra)", "t");
113 expect("is_substring(dabra, abracadabra)", "t");
114 expect("is_substring(cad, abracxadabra)", "");
115 expect("is_substring(abrac, axbracadabra)", "");
116 expect("is_substring(dabra, abracadabrxa)", "");
117}
118
119TEST_F(EdifyTest, ifelse) {
120 // ifelse function
121 expect("ifelse(t, yes, no)", "yes");
122 expect("ifelse(!t, yes, no)", "no");
123 expect("ifelse(t, yes, abort())", "yes");
124 expect("ifelse(!t, abort(), no)", "no");
125}
126
127TEST_F(EdifyTest, if_statement) {
128 // if "statements"
129 expect("if t then yes else no endif", "yes");
130 expect("if \"\" then yes else no endif", "no");
131 expect("if \"\" then yes endif", "");
132 expect("if \"\"; t then yes endif", "yes");
133}
134
135TEST_F(EdifyTest, comparison) {
136 // numeric comparisons
137 expect("less_than_int(3, 14)", "t");
138 expect("less_than_int(14, 3)", "");
139 expect("less_than_int(x, 3)", "");
140 expect("less_than_int(3, x)", "");
141 expect("greater_than_int(3, 14)", "");
142 expect("greater_than_int(14, 3)", "t");
143 expect("greater_than_int(x, 3)", "");
144 expect("greater_than_int(3, x)", "");
145}
146
147TEST_F(EdifyTest, big_string) {
148 // big string
149 expect(std::string(8192, 's').c_str(), std::string(8192, 's').c_str());
150}
151
Tao Bao39119ad2016-10-10 22:52:18 -0700152TEST_F(EdifyTest, unknown_function) {
153 // unknown function
154 const char* script1 = "unknown_function()";
155 Expr* expr;
156 int error_count = 0;
157 EXPECT_EQ(1, parse_string(script1, &expr, &error_count));
158 EXPECT_EQ(1, error_count);
159
160 const char* script2 = "abc; unknown_function()";
161 error_count = 0;
162 EXPECT_EQ(1, parse_string(script2, &expr, &error_count));
163 EXPECT_EQ(1, error_count);
164
165 const char* script3 = "unknown_function1() || yes";
166 error_count = 0;
167 EXPECT_EQ(1, parse_string(script3, &expr, &error_count));
168 EXPECT_EQ(1, error_count);
169}