Skip to content

Commit 6a41ad0

Browse files
xinyualxinyual
committed
fix UT
Signed-off-by: xinyual <xinyual@amazon.com>
1 parent 0e601c6 commit 6a41ad0

3 files changed

Lines changed: 19 additions & 19 deletions

File tree

‎core/src/test/java/org/opensearch/sql/analysis/AnalyzerSearchTest.java‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ public void testVisitSearchWithSimpleQuery() {
6565
// Arrange
6666
String queryString = "field1:value1";
6767
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
68-
Search searchNode = new Search(relation, queryString);
68+
Search searchNode = new Search(relation, queryString, null);
6969

7070
LogicalRelation logicalRelation = new LogicalRelation("test_index", mockTable);
7171
FunctionExpression queryStringExpr = mock(FunctionExpression.class);
@@ -103,7 +103,7 @@ public void testVisitSearchWithComplexQuery() {
103103
// Arrange
104104
String queryString = "(field1:value1 OR field2:value2) AND NOT field3:value3";
105105
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
106-
Search searchNode = new Search(relation, queryString);
106+
Search searchNode = new Search(relation, queryString, null);
107107

108108
FunctionExpression queryStringExpr = mock(FunctionExpression.class);
109109
when(queryStringExpr.type()).thenReturn(ExprCoreType.BOOLEAN);
@@ -134,7 +134,7 @@ public void testVisitSearchPreservesChildPlan() {
134134
// Arrange
135135
String queryString = "test:query";
136136
UnresolvedPlan mockChild = mock(UnresolvedPlan.class);
137-
Search searchNode = new Search(mockChild, queryString);
137+
Search searchNode = new Search(mockChild, queryString, null);
138138

139139
LogicalPlan mockLogicalPlan = mock(LogicalPlan.class);
140140
FunctionExpression queryStringExpr = mock(FunctionExpression.class);
@@ -158,7 +158,7 @@ public void testVisitSearchWithEmptyQuery() {
158158
// Arrange
159159
String queryString = "";
160160
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
161-
Search searchNode = new Search(relation, queryString);
161+
Search searchNode = new Search(relation, queryString, null);
162162

163163
FunctionExpression queryStringExpr = mock(FunctionExpression.class);
164164

@@ -193,7 +193,7 @@ public void testVisitSearchCreatesCorrectQueryStringFunction() {
193193
// Arrange
194194
String queryString = "field:\"exact phrase\" AND field2:wildcard*";
195195
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
196-
Search searchNode = new Search(relation, queryString);
196+
Search searchNode = new Search(relation, queryString, null);
197197

198198
FunctionExpression queryStringExpr = mock(FunctionExpression.class);
199199

‎core/src/test/java/org/opensearch/sql/ast/tree/SearchTest.java‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ public class SearchTest {
2929
public void setUp() {
3030
mockChild = mock(UnresolvedPlan.class);
3131
testQueryString = "field1:value1 AND field2:value2";
32-
search = new Search(mockChild, testQueryString);
32+
search = new Search(mockChild, testQueryString, null);
3333
}
3434

3535
@Test
@@ -69,7 +69,7 @@ public void testAccept() {
6969
public void testEquals() {
7070
UnresolvedPlan sameChild = mockChild;
7171
String sameQueryString = testQueryString;
72-
Search sameSearch = new Search(sameChild, sameQueryString);
72+
Search sameSearch = new Search(sameChild, sameQueryString, null);
7373

7474
assertEquals(search, sameSearch);
7575
assertEquals(search, search);
@@ -78,14 +78,14 @@ public void testEquals() {
7878
@Test
7979
public void testNotEqualsWithDifferentChild() {
8080
UnresolvedPlan differentChild = mock(UnresolvedPlan.class);
81-
Search differentSearch = new Search(differentChild, testQueryString);
81+
Search differentSearch = new Search(differentChild, testQueryString, null);
8282

8383
assertNotEquals(search, differentSearch);
8484
}
8585

8686
@Test
8787
public void testNotEqualsWithDifferentQueryString() {
88-
Search differentSearch = new Search(mockChild, "different:query");
88+
Search differentSearch = new Search(mockChild, "different:query", null);
8989

9090
assertNotEquals(search, differentSearch);
9191
}
@@ -102,14 +102,14 @@ public void testNotEqualsWithDifferentClass() {
102102

103103
@Test
104104
public void testHashCode() {
105-
Search sameSearch = new Search(mockChild, testQueryString);
105+
Search sameSearch = new Search(mockChild, testQueryString, null);
106106
assertEquals(search.hashCode(), sameSearch.hashCode());
107107
}
108108

109109
@Test
110110
public void testHashCodeWithDifferentValues() {
111111
UnresolvedPlan differentChild = mock(UnresolvedPlan.class);
112-
Search differentSearch = new Search(differentChild, "different:query");
112+
Search differentSearch = new Search(differentChild, "different:query", null);
113113
assertNotEquals(search.hashCode(), differentSearch.hashCode());
114114
}
115115

@@ -123,22 +123,22 @@ public void testToString() {
123123

124124
@Test
125125
public void testWithEmptyQueryString() {
126-
Search emptySearch = new Search(mockChild, "");
126+
Search emptySearch = new Search(mockChild, "", null);
127127
assertEquals("", emptySearch.getQueryString());
128128
assertEquals(mockChild, emptySearch.getChild().get(0));
129129
}
130130

131131
@Test
132132
public void testWithComplexQueryString() {
133133
String complexQuery = "(field1:value1 OR field2:value2) AND NOT field3:value3";
134-
Search complexSearch = new Search(mockChild, complexQuery);
134+
Search complexSearch = new Search(mockChild, complexQuery, null);
135135
assertEquals(complexQuery, complexSearch.getQueryString());
136136
}
137137

138138
@Test
139139
public void testWithSpecialCharactersInQueryString() {
140140
String specialCharsQuery = "field:\"value with spaces\" AND field2:value*";
141-
Search specialSearch = new Search(mockChild, specialCharsQuery);
141+
Search specialSearch = new Search(mockChild, specialCharsQuery, null);
142142
assertEquals(specialCharsQuery, specialSearch.getQueryString());
143143
}
144144
}

‎core/src/test/java/org/opensearch/sql/calcite/CalciteRelNodeVisitorSearchSimpleTest.java‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ public void testVisitSearchRequiresContext() {
3636
// Arrange
3737
String queryString = "field:value";
3838
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
39-
Search searchNode = new Search(relation, queryString);
39+
Search searchNode = new Search(relation, queryString, null);
4040

4141
// Act & Assert - should throw NPE without proper context
4242
assertThrows(
@@ -53,7 +53,7 @@ public void testSearchNodeStructure() {
5353
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
5454

5555
// Act
56-
Search searchNode = new Search(relation, queryString);
56+
Search searchNode = new Search(relation, queryString, null);
5757

5858
// Assert
5959
assertNotNull(searchNode);
@@ -70,7 +70,7 @@ public void testSearchWithEmptyQuery() {
7070
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
7171

7272
// Act
73-
Search searchNode = new Search(relation, queryString);
73+
Search searchNode = new Search(relation, queryString, null);
7474

7575
// Assert
7676
assertEquals("", searchNode.getQueryString());
@@ -84,7 +84,7 @@ public void testSearchWithComplexQuery() {
8484
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
8585

8686
// Act
87-
Search searchNode = new Search(relation, queryString);
87+
Search searchNode = new Search(relation, queryString, null);
8888

8989
// Assert
9090
assertEquals(queryString, searchNode.getQueryString());
@@ -97,7 +97,7 @@ public void testSearchWithSpecialCharacters() {
9797
Relation relation = new Relation(AstDSL.qualifiedName("test_index"));
9898

9999
// Act
100-
Search searchNode = new Search(relation, queryString);
100+
Search searchNode = new Search(relation, queryString, null);
101101

102102
// Assert
103103
assertEquals(queryString, searchNode.getQueryString());

0 commit comments

Comments
 (0)