Skip to content

Commit

Permalink
Fix tests.
Browse files Browse the repository at this point in the history
Signed-off-by: Yury-Fridlyand <[email protected]>
  • Loading branch information
Yury-Fridlyand committed Aug 18, 2023
1 parent 9ba8341 commit 662ad07
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 2 deletions.
6 changes: 5 additions & 1 deletion ppl/src/test/java/org/opensearch/sql/ppl/PPLServiceTest.java
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import org.opensearch.sql.common.response.ResponseListener;
import org.opensearch.sql.common.setting.Settings;
import org.opensearch.sql.executor.DefaultQueryManager;
import org.opensearch.sql.executor.ExecutionEngine;
import org.opensearch.sql.executor.ExecutionEngine.ExplainResponse;
Expand All @@ -44,13 +45,16 @@ public class PPLServiceTest {

@Mock private ExecutionEngine.Schema schema;

@Mock private Settings settings;

/** Setup the test context. */
@Before
public void setUp() {
queryManager = DefaultQueryManager.defaultQueryManager();

pplService =
new PPLService(new PPLSyntaxParser(), queryManager, new QueryPlanFactory(queryService));
new PPLService(
new PPLSyntaxParser(), queryManager, new QueryPlanFactory(queryService, settings));
}

@After
Expand Down
6 changes: 5 additions & 1 deletion sql/src/test/java/org/opensearch/sql/sql/SQLServiceTest.java
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.opensearch.sql.common.response.ResponseListener;
import org.opensearch.sql.common.setting.Settings;
import org.opensearch.sql.executor.DefaultQueryManager;
import org.opensearch.sql.executor.ExecutionEngine.ExplainResponse;
import org.opensearch.sql.executor.ExecutionEngine.ExplainResponseNode;
Expand All @@ -46,11 +47,14 @@ class SQLServiceTest {

@Mock private QueryService queryService;

@Mock private Settings settings;

@BeforeEach
public void setUp() {
queryManager = DefaultQueryManager.defaultQueryManager();
sqlService =
new SQLService(new SQLSyntaxParser(), queryManager, new QueryPlanFactory(queryService));
new SQLService(
new SQLSyntaxParser(), queryManager, new QueryPlanFactory(queryService, settings));
}

@AfterEach
Expand Down

0 comments on commit 662ad07

Please sign in to comment.