Skip to content

Commit

Permalink
context logs
Browse files Browse the repository at this point in the history
  • Loading branch information
@stas-panasiuk
stas-panasiuk committed Feb 12, 2024
1 parent 8d1c7b1 commit d8003db
Showing 1 changed file with 23 additions and 24 deletions.
47 changes: 23 additions & 24 deletions ...r/src/test/java/org/apache/metron/stellar/dsl/functions/RestFunctionsIntegrationTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,6 @@
import org.junit.jupiter.migrationsupport.rules.EnableRuleMigrationSupport;
import org.junit.rules.TemporaryFolder;
import org.mockserver.integration.ClientAndServer;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.nio.charset.StandardCharsets;
Expand All @@ -49,10 +48,9 @@

@EnableRuleMigrationSupport
public class RestFunctionsIntegrationTest {
public static final Context CONTEXT = new Context.Builder().build();
@Rule
public TemporaryFolder tempDir = new TemporaryFolder();

private static final org.slf4j.Logger LOG = LoggerFactory.getLogger(RestFunctionsIntegrationTest.class);

private static final int MOCK_PROXY_PORT = 1080;

Expand Down Expand Up @@ -124,6 +122,7 @@ public void setup() throws Exception {
.withPath("/post/empty"))
.respond(response()
.withStatusCode(404));
assertNotNull(CONTEXT);
}

@AfterEach
Expand All @@ -137,7 +136,7 @@ public void teardown() {
@Test
@SuppressWarnings("unchecked")
public void restGetShouldSucceed() {
assertEquals(String.format("TESTING[restGetShouldSucceed] getUri: [%s]", getUri), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(String.format("REST_GET('%s')", getUri), context);

assertEquals(1, actual.size());
Expand All @@ -159,7 +158,7 @@ public void restGetShouldSucceedWithQueryParameters() {
.withBody("{\"get.with.query.parameters\":\"success\"}"));

Map<String, Object> variables = ImmutableMap.of("queryParameters", ImmutableMap.of("key", "value"));
assertEquals(String.format("TESTING[restGetShouldSucceedWithQueryParameters] baseUri: [%s], variables: [%s]", baseUri, variables), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(String.format("REST_GET('%s', {}, queryParameters)",
baseUri + "/get/with/query/parameters"), variables, context);

Expand All @@ -185,7 +184,7 @@ public void restGetShouldSucceedWithProxy() {
put(PROXY_PORT, MOCK_PROXY_PORT);
}});

assertEquals(String.format("TESTING[restGetShouldSucceedWithProxy] getUri: [%s]", getUri), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(String.format("REST_GET('%s')", getUri), context);

assertEquals(1, actual.size());
Expand All @@ -204,7 +203,7 @@ public void restGetShouldHandleErrorStatusCode() {
.respond(response()
.withStatusCode(403));

assertEquals(String.format("TESTING[restGetShouldHandleErrorStatusCode] getUri: [%s]", getUri), "");
assertEquals(context, CONTEXT);
assertNull(run(String.format("REST_GET('%s')", getUri), context));
}

Expand All @@ -222,7 +221,7 @@ public void restGetShouldHandleErrorStatusCode() {
*/
@Test
public void restGetShouldReturnEmptyContentOverride() {
assertEquals(String.format("TESTING[restGetShouldReturnEmptyContentOverride] emptyGetUri: [%s], emptyContentOverride: [%s]", emptyGetUri, emptyContentOverride), "");
assertEquals(context, CONTEXT);
assertEquals("function config override", run(String.format("REST_GET('%s', %s)", emptyGetUri, emptyContentOverride), context));
}

Expand All @@ -246,7 +245,7 @@ public void restGetShouldReturnErrorValueOverride() {
.respond(response()
.withStatusCode(500));

assertEquals(String.format("TESTING[restGetShouldReturnErrorValueOverride] getUri: [%s], errorValueOverride: [%s]", getUri, errorValueOverride), "");
assertEquals(context, CONTEXT);
Object result = run(String.format("REST_GET('%s', %s)", getUri, errorValueOverride), context);
assertEquals("error message" , result);
}
Expand Down Expand Up @@ -275,7 +274,7 @@ public void restGetShouldTimeout() {

context.addCapability(Context.Capabilities.GLOBAL_CONFIG, () -> globalConfig);

assertEquals(String.format("TESTING[restGetShouldTimeout] uri: [%s]", uri), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(String.format("REST_GET('%s')", uri), context);
assertNull(actual);
}
Expand Down Expand Up @@ -304,7 +303,7 @@ public void restGetShouldTimeoutWithSuppliedTimeout() {
.withBody("{\"get\":\"success\"}"));

String expression = String.format("REST_GET('%s', %s)", uri, timeoutConfig);
assertEquals(String.format("TESTING[restGetShouldTimeoutWithSuppliedTimeout] expression: [%s]", expression), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(expression, context);
assertNull(actual);
}
Expand All @@ -314,7 +313,7 @@ public void restGetShouldTimeoutWithSuppliedTimeout() {
*/
@Test
public void restGetShouldHandleURISyntaxException() {
assertEquals(String.format("TESTING[restGetShouldHandleURISyntaxException]"), "");
assertEquals(context, CONTEXT);
ParseException e = assertThrows(ParseException.class, () -> run("REST_GET('some invalid uri')", context));
assertEquals("Unable to parse REST_GET('some invalid uri'): Unable to parse: REST_GET('some invalid uri') due to: Illegal character in path at index 4: some invalid uri", e.getMessage());
}
Expand All @@ -326,7 +325,7 @@ public void restGetShouldHandleURISyntaxException() {
*/
@Test
public void restGetShouldThrownExceptionOnMissingParameter() {
assertEquals(String.format("TESTING[restGetShouldThrownExceptionOnMissingParameter]"), "");
assertEquals(context, CONTEXT);
ParseException e = assertThrows(ParseException.class, () -> run("REST_GET()", context));
assertEquals("Unable to parse REST_GET(): Unable to parse: REST_GET() due to: Expected at least 1 argument(s), found 0", e.getMessage());
}
Expand All @@ -344,7 +343,7 @@ public void restGetShouldUseGlobalConfig() {
}};
context.addCapability(Context.Capabilities.GLOBAL_CONFIG, () -> globalConfig);

assertEquals(String.format("TESTING[restGetShouldUseGlobalConfig] emptyGetUri: [%s]", emptyGetUri), "");
assertEquals(context, CONTEXT);
assertEquals("global config override", run(String.format("REST_GET('%s')", emptyGetUri), context));
}

Expand All @@ -364,7 +363,7 @@ public void restGetShouldUseGetConfig() {
}};
context.addCapability(Context.Capabilities.GLOBAL_CONFIG, () -> globalConfig);

assertEquals(String.format("TESTING[restGetShouldUseGetConfig] emptyGetUri: [%s]", emptyGetUri), "");
assertEquals(context, CONTEXT);
assertEquals("get config override", run(String.format("REST_GET('%s')", emptyGetUri), context));
}

Expand All @@ -384,7 +383,7 @@ public void restGetShouldUseFunctionConfig() {
}};
context.addCapability(Context.Capabilities.GLOBAL_CONFIG, () -> globalConfig);

assertEquals(String.format("TESTING[restGetShouldUseFunctionConfig] emptyGetUri: [%s], emptyContentOverride: [%s]", emptyGetUri, emptyContentOverride), "");
assertEquals(context, CONTEXT);
assertEquals("function config override", run(String.format("REST_GET('%s', %s)", emptyGetUri, emptyContentOverride), context));
}

Expand All @@ -394,7 +393,7 @@ public void restGetShouldUseFunctionConfig() {
@Test
@SuppressWarnings("unchecked")
public void restPostShouldSucceed() {
assertEquals(String.format("TESTING[restPostShouldSucceed] postUri: [%s]", postUri), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(String.format("REST_POST('%s', '{\"key\":\"value\"}')", postUri), context);

assertEquals(1, actual.size());
Expand All @@ -416,7 +415,7 @@ public void restPostShouldSucceedWithQueryParameters() {
.withBody("{\"post.with.query.parameters\":\"success\"}"));

Map<String, Object> variables = ImmutableMap.of("queryParameters", ImmutableMap.of("key", "value"));
assertEquals(String.format("TESTING[restPostShouldSucceedWithQueryParameters] baseUri: [%s], variables: [%s]", baseUri,variables), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(String.format("REST_POST('%s', {}, {}, queryParameters)",
baseUri + "/post/with/query/parameters"), variables, context);

Expand All @@ -431,7 +430,7 @@ public void restPostShouldSucceedWithQueryParameters() {
@SuppressWarnings("unchecked")
public void restPostShouldSucceedWithStellarMap() {
Map<String, Object> variables = ImmutableMap.of("body", ImmutableMap.of("key", "value"));
assertEquals(String.format("TESTING[restPostShouldSucceedWithStellarMap] postUri: [%s], variables: [%s]", postUri,variables), "");
assertEquals(context, CONTEXT);
Map<String, Object> actual = (Map<String, Object>) run(String.format("REST_POST('%s', body)", postUri), variables, context);

assertEquals(1, actual.size());
Expand All @@ -443,7 +442,7 @@ public void restPostShouldSucceedWithStellarMap() {
*/
@Test
public void restPostShouldHandleURISyntaxException() {
assertEquals(String.format("TESTING[restPostShouldHandleURISyntaxException]"), "");
assertEquals(context, CONTEXT);
ParseException e = assertThrows(ParseException.class, () -> run("REST_POST('some invalid uri', {})", context));
assertEquals("Unable to parse REST_POST('some invalid uri', {}): Unable to parse: REST_POST('some invalid uri', {}) due to: Illegal character in path at index 4: some invalid uri", e.getMessage());
}
Expand All @@ -453,7 +452,7 @@ public void restPostShouldHandleURISyntaxException() {
*/
@Test
public void restPostShouldThrowExceptionOnMalformedJson() {
assertEquals(String.format("TESTING[restPostShouldThrowExceptionOnMalformedJson] postUri: [%s]", postUri), "");
assertEquals(context, CONTEXT);
ParseException e = assertThrows(ParseException.class, () -> run(String.format("REST_POST('%s', 'malformed json')", postUri), context));
assertEquals(
String.format(
Expand All @@ -477,7 +476,7 @@ public void restPostShouldUseGlobalConfig() {
}};
context.addCapability(Context.Capabilities.GLOBAL_CONFIG, () -> globalConfig);

assertEquals(String.format("TESTING[restPostShouldUseGlobalConfig] emptyGetUri: [%s]", emptyGetUri), "");
assertEquals(context, CONTEXT);
assertEquals("global config override", run(String.format("REST_POST('%s', {})", emptyGetUri), context));
}

Expand All @@ -497,7 +496,7 @@ public void restPostShouldUseGetConfig() {
}};
context.addCapability(Context.Capabilities.GLOBAL_CONFIG, () -> globalConfig);

assertEquals(String.format("TESTING[restPostShouldUseGetConfig] emptyGetUri: [%s]", emptyGetUri), "");
assertEquals(context, CONTEXT);
assertEquals("post config override", run(String.format("REST_POST('%s', {})", emptyGetUri), context));
}

Expand All @@ -517,7 +516,7 @@ public void restPostShouldUseFunctionConfig() {
}};
context.addCapability(Context.Capabilities.GLOBAL_CONFIG, () -> globalConfig);

assertEquals(String.format("TESTING[restPostShouldUseFunctionConfig] emptyGetUri: [%s], emptyContentOverride: [%s]", emptyGetUri, emptyContentOverride), "");
assertEquals(context, CONTEXT);
assertEquals("function config override", run(String.format("REST_POST('%s', {}, %s)", emptyGetUri, emptyContentOverride), context));
}

Expand Down

0 comments on commit d8003db

Please sign in to comment.