Vertx gradle test

Author: jandro996

build.gradle

@@ -22,7 +22,7 @@ plugins {
   id 'pl.allegro.tech.build.axion-release' version '1.14.4'
   id 'io.github.gradle-nexus.publish-plugin' version '1.3.0'
 
-  id "com.github.johnrengelman.shadow" version "7.1.2" apply false
+  id "com.github.johnrengelman.shadow" version "8.1.1" apply false
   id "me.champeau.jmh" version "0.7.0" apply false
   id 'org.gradle.playframework' version '0.13' apply false
   id 'info.solidsoft.pitest' version '1.9.11'  apply false

dd-java-agent/agent-iast/src/main/java/com/datadog/iast/overhead/Operations.java

@@ -1,5 +1,7 @@
 package com.datadog.iast.overhead;
 
+import datadog.trace.api.Config;
+
 import javax.annotation.Nullable;
 
 public class Operations {
@@ -18,6 +20,9 @@ public boolean hasQuota(@Nullable final OverheadContext context) {
 
         @Override
         public boolean consumeQuota(@Nullable final OverheadContext context) {
+//          if(Config.get().isCiVisibilityEnabled()){
+//            return true;
+//          }
           if (context == null) {
             return false;
           }

dd-smoke-tests/vertx3-iast-ci/build.gradle

@@ -0,0 +1,20 @@
+plugins {
+  id 'java'
+  id 'application'
+}
+
+repositories {
+  mavenCentral()
+}
+
+dependencies {
+  implementation 'io.vertx:vertx-core:3.9.13'
+  implementation 'io.vertx:vertx-web:3.9.13'
+
+  testImplementation 'io.vertx:vertx-unit:3.9.13'
+  testImplementation 'junit:junit:4.13.2'
+}
+
+application {
+  mainClass = 'com.example.MainVerticle'
+}

dd-smoke-tests/vertx3-iast-ci/settings.gradle

@@ -0,0 +1,2 @@
+rootProject.name = 'Vertx3-IW'
+

dd-smoke-tests/vertx3-iast-ci/src/main/java/com.example/MainVerticle.java

@@ -0,0 +1,53 @@
+package com.example;
+
+import io.vertx.core.AbstractVerticle;
+import io.vertx.core.Future;
+import io.vertx.ext.web.Router;
+
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+import java.util.Arrays;
+
+public class MainVerticle extends AbstractVerticle {
+
+    @Override
+    public void start(Future<Void> startFuture) {
+        Router router = Router.router(vertx);
+
+        router.get("/greetings").handler(ctx -> {
+            String param = ctx.request().getParam("param");
+            if (param == null) {
+                ctx.response()
+                        .setStatusCode(400)
+                        .end("Missing 'param' query parameter");
+            } else {
+                ctx.response()
+                        .putHeader("content-type", "text/plain")
+                        .end("Hello " + param);
+            }
+        });
+
+        router.get("/insecure_hash").handler(ctx -> {
+            try {
+                MessageDigest md = MessageDigest.getInstance("MD5");
+                ctx.response()
+                        .putHeader("content-type", "text/plain")
+                        .end("insecure hash");
+            } catch (NoSuchAlgorithmException e) {
+                throw new RuntimeException(e);
+            }
+        });
+
+        // Iniciar servidor HTTP
+        vertx.createHttpServer()
+                .requestHandler(router)
+                .listen(8080, http -> {
+                    if (http.succeeded()) {
+                        startFuture.complete();
+                        System.out.println("HTTP server started on port 8080");
+                    } else {
+                        startFuture.fail(http.cause());
+                    }
+                });
+    }
+}

dd-smoke-tests/vertx3-iast-ci/src/test/java/com/example/MainVerticleTest.java

@@ -0,0 +1,69 @@
+package com.example;
+
+import io.vertx.core.Vertx;
+import io.vertx.ext.unit.Async;
+import io.vertx.ext.unit.TestContext;
+import io.vertx.ext.unit.junit.VertxUnitRunner;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+import java.security.MessageDigest;
+import java.security.NoSuchAlgorithmException;
+
+@RunWith(VertxUnitRunner.class)
+public class MainVerticleTest {
+
+  private Vertx vertx;
+  private TestContext context;
+
+  @Before
+  public void setUp(TestContext testContext) {
+    context = testContext;
+    vertx = Vertx.vertx();
+    vertx.deployVerticle(MainVerticle.class.getName(), testContext.asyncAssertSuccess());
+  }
+
+  @After
+  public void tearDown() {
+    vertx.close(context.asyncAssertSuccess());
+  }
+
+    @Test
+    public void testGreetingsEndpoint() {
+        Async async = context.async();
+
+        vertx.createHttpClient().getNow(8080, "localhost", "/greetings?param=World", response -> {
+            context.assertEquals(200, response.statusCode());
+
+            response.bodyHandler(body -> {
+                context.assertEquals("Hello World", body.toString());
+                async.complete();
+            });
+        });
+    }
+
+    @Test
+    public void testInsecureHashEndpoint() {
+        Async async = context.async();
+
+        vertx.createHttpClient().getNow(8080, "localhost", "/insecure_hash", response -> {
+            context.assertEquals(200, response.statusCode());
+
+            response.bodyHandler(body -> {
+                context.assertEquals("insecure hash", body.toString());
+                async.complete();
+            });
+        });
+    }
+
+  @Test
+  public void testInsecureHash() throws NoSuchAlgorithmException {
+    MessageDigest.getInstance("MD5");
+    assert true;
+  }
+
+
+
+}

dd-smoke-tests/vertx3-iast-ci/src/test/resources/readme.md

@@ -0,0 +1,9 @@
+./gradlew dd-smoke-tests:vertx3-iast-ci:test --rerun-tasks -Dorg.gradle.jvmargs=\
+-javaagent:/Users/alejandro.gonzalez/IdeaProjects/dd-trace-java/dd-java-agent/build/libs/dd-java-agent-1.44.0-SNAPSHOT.jar=\
+dd.civisibility.enabled=true,\
+dd.env=local,\
+dd.service=dd—java-trace-vertx-iw-q42024,\
+dd.iast.enabled=true,\
+dd.iast.detection.mode=FULL,\
+dd.iast.context.mode=GLOBAL,\
+dd.civisibility.debug.port=5005

settings.gradle

@@ -507,3 +507,6 @@ include ':dd-java-agent:benchmark'
 include ':dd-java-agent:benchmark-integration'
 include ':dd-java-agent:benchmark-integration:jetty-perftest'
 include ':dd-java-agent:benchmark-integration:play-perftest'
+include 'dd-smoke-tests:vertx3-iast-ci'
+findProject(':dd-smoke-tests:vertx3-iast-ci')?.name = 'vertx3-iast-ci'
+