Updated agent tests

kornilova203 · Feb 8, 2018 · 4bb42b6 · 4bb42b6
1 parent 40c5f3a
commit 4bb42b6
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 1 deletion.
diff --git a/agent-test/src/test/java/com/github/kornilova_l/flamegraph/proxy/TestWithProxy.java b/agent-test/src/test/java/com/github/kornilova_l/flamegraph/proxy/TestWithProxy.java
@@ -67,7 +67,8 @@ static void addProxy() {
         try {
             Method method = urlClass.getDeclaredMethod("addURL", URL.class);
             method.setAccessible(true);
-            method.invoke(urlClassLoader, Paths.get(pathToAgentDir.toString(), "proxy.jar").toUri().toURL());
+            method.invoke(urlClassLoader,
+                    new URL(Paths.get(pathToAgentDir.toString(), "proxy.jar").toUri().toString().replaceAll("%20", " ")));
         } catch (NoSuchMethodException | IllegalAccessException | MalformedURLException | InvocationTargetException e) {
             e.printStackTrace();
         }

diff --git a/agent-test/src/test/java/com/github/kornilova_l/proxy_test_classes/ReturnsValue.java b/agent-test/src/test/java/com/github/kornilova_l/proxy_test_classes/ReturnsValue.java
@@ -5,6 +5,11 @@ public class ReturnsValue implements TestModule {
     @Override
     public int run(long l, String s, double d) {
         System.out.println(l + s + d);
+        try {
+            Thread.sleep(10);
+        } catch (InterruptedException e) {
+            e.printStackTrace();
+        }
         return 42;
     }
 }
diff --git a/agent-test/src/test/java/com/github/kornilova_l/proxy_test_classes/ThrowsException.java b/agent-test/src/test/java/com/github/kornilova_l/proxy_test_classes/ThrowsException.java
@@ -4,6 +4,11 @@ public class ThrowsException implements TestModule {
     @Override
     public int run(long l, String s, double d) {
         System.out.println(l + s + d);
+        try {
+            Thread.sleep(10);
+        } catch (InterruptedException e) {
+            e.printStackTrace();
+        }
         throw new RuntimeException("I am an exception");
     }
 }