fixing typos in UTs

opendistro-for-elasticsearch · Nov 16, 2019 · 7f723d1 · 7f723d1
1 parent 8d4f692
commit 7f723d1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/src/test/java/com/amazon/dlic/auth/http/saml/HTTPSamlAuthenticatorTest.java b/src/test/java/com/amazon/dlic/auth/http/saml/HTTPSamlAuthenticatorTest.java
@@ -485,6 +485,7 @@ public void initialConnectionFailureTest() throws Exception {
 
             Assert.assertEquals("horst", jwt.getClaim("sub"));
         }
+    }
 
     private AuthenticateHeaders getAutenticateHeaders(HTTPSamlAuthenticator samlAuthenticator) {
         RestRequest restRequest = new FakeRestRequest(ImmutableMap.of(), new HashMap<String, String>());
@@ -583,11 +584,6 @@ public XContentBuilder newBuilder(XContentType xContentType, boolean useFilterin
             return null;
         }
 
-        @Override
-        public XContentBuilder newBuilder(XContentType xContentType, XContentType xType, boolean useFiltering) throws IOException {
-            return null;
-        }
-
         @Override
         public BytesStreamOutput bytesOutput() {
             return null;

diff --git a/...test/java/com/amazon/opendistroforelasticsearch/security/auditlog/sink/KafkaSinkTest.java b/...test/java/com/amazon/opendistroforelasticsearch/security/auditlog/sink/KafkaSinkTest.java
@@ -15,6 +15,7 @@
 
 package com.amazon.opendistroforelasticsearch.security.auditlog.sink;
 
+import java.time.Duration;
 import java.util.Arrays;
 import java.util.Properties;
 
@@ -25,6 +26,7 @@
 import org.junit.Assert;
 import org.junit.ClassRule;
 import org.junit.Test;
+import org.springframework.kafka.test.rule.EmbeddedKafkaRule;
 import org.springframework.kafka.test.rule.KafkaEmbedded;
 
 import scala.util.Random;
@@ -45,7 +47,7 @@ public class KafkaSinkTest extends AbstractAuditlogiUnitTest {
 	@Test
 	public void testKafka() throws Exception {
 	    String configYml = FileHelper.loadFile("auditlog/endpoints/sink/configuration_kafka.yml");
-	    configYml = configYml.replace("_RPLC_BOOTSTRAP_SERVERS_",embeddedKafka.getBrokersAsString());
+		configYml = configYml.replace("_RPLC_BOOTSTRAP_SERVERS_",embeddedKafka.getEmbeddedKafka().getBrokersAsString());
 		Settings.Builder settingsBuilder = Settings.builder().loadFromSource(configYml, YamlXContent.yamlXContent.type());
 		try(KafkaConsumer<Long, String> consumer = createConsumer()) {
 		    consumer.subscribe(Arrays.asList("compliance"));

diff --git a/src/test/java/com/amazon/opendistroforelasticsearch/security/dlic/dlsfls/DlsScrollTest.java b/src/test/java/com/amazon/opendistroforelasticsearch/security/dlic/dlsfls/DlsScrollTest.java
@@ -8,8 +8,8 @@
 import org.junit.Assert;
 import org.junit.Test;
 
-import com.floragunn.searchguard.test.helper.file.FileHelper;
-import com.floragunn.searchguard.test.helper.rest.RestHelper.HttpResponse;
+import com.amazon.opendistroforelasticsearch.security.test.helper.file.FileHelper;
+import com.amazon.opendistroforelasticsearch.security.test.helper.rest.RestHelper.HttpResponse;
 
 public class DlsScrollTest extends AbstractDlsFlsTest{