7
7
8
8
import static org .opensearch .sql .legacy .TestUtils .getResponseBody ;
9
9
10
+ import java .io .IOException ;
10
11
import lombok .SneakyThrows ;
11
12
import org .json .JSONObject ;
13
+ import org .junit .After ;
12
14
import org .junit .Assert ;
13
15
import org .junit .Test ;
14
16
import org .opensearch .client .Request ;
18
20
19
21
public class DataSourceEnabledIT extends PPLIntegTestCase {
20
22
21
- @ Override
22
- protected boolean preserveClusterUponCompletion () {
23
- return false ;
23
+ @ After
24
+ public void cleanUp () throws IOException {
25
+ wipeAllClusterSettings () ;
24
26
}
25
27
26
28
@ Test
@@ -39,6 +41,7 @@ public void testDataSourceCreationWithDefaultSettings() {
39
41
assertSelectFromDataSourceReturnsSuccess ();
40
42
assertSelectFromDummyIndexInValidDataSourceDataSourceReturnsDoesNotExist ();
41
43
deleteSelfDataSourceCreated ();
44
+ deleteIndex ();
42
45
}
43
46
44
47
@ Test
@@ -55,6 +58,7 @@ public void testAfterPreviousEnable() {
55
58
assertAsyncQueryApiDisabled ();
56
59
setDataSourcesEnabled ("transient" , true );
57
60
deleteSelfDataSourceCreated ();
61
+ deleteIndex ();
58
62
}
59
63
60
64
@ SneakyThrows
@@ -98,6 +102,12 @@ private void createIndex() {
98
102
Assert .assertEquals (200 , response .getStatusLine ().getStatusCode ());
99
103
}
100
104
105
+ private void deleteIndex () {
106
+ Request request = new Request ("DELETE" , "/myindex" );
107
+ Response response = performRequest (request );
108
+ Assert .assertEquals (200 , response .getStatusLine ().getStatusCode ());
109
+ }
110
+
101
111
private void createOpenSearchDataSource () {
102
112
Request request = new Request ("POST" , "/_plugins/_query/_datasources" );
103
113
request .setJsonEntity (
0 commit comments