From 2e8fcd19cba14f661793e63e22c63e1fc5ff3edf Mon Sep 17 00:00:00 2001
From: Anuraag Agrawal <aanuraag@amazon.co.jp>
Date: Wed, 16 Jun 2021 18:35:07 +0900
Subject: [PATCH] Migrate CallDepthThreadLocalMapTest to Java. (#3331)

---
 .../api/CallDepthThreadLocalMapTest.groovy    | 53 -------------------
 .../api/CallDepthThreadLocalMapTest.java      | 42 +++++++++++++++
 .../PeerServiceAttributesExtractorTest.java   |  0
 3 files changed, 42 insertions(+), 53 deletions(-)
 delete mode 100644 javaagent-api/src/test/groovy/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.groovy
 create mode 100644 javaagent-api/src/test/java/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.java
 rename javaagent-api/src/test/{groovy => java}/io/opentelemetry/javaagent/instrumentation/api/instrumenter/PeerServiceAttributesExtractorTest.java (100%)

diff --git a/javaagent-api/src/test/groovy/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.groovy b/javaagent-api/src/test/groovy/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.groovy
deleted file mode 100644
index 20ad9be2b0..0000000000
--- a/javaagent-api/src/test/groovy/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.groovy
+++ /dev/null
@@ -1,53 +0,0 @@
-/*
- * Copyright The OpenTelemetry Authors
- * SPDX-License-Identifier: Apache-2.0
- */
-
-package io.opentelemetry.javaagent.instrumentation.api
-
-import spock.lang.Specification
-
-class CallDepthThreadLocalMapTest extends Specification {
-
-  def "test CallDepthThreadLocalMap"() {
-    setup:
-    Class<?> k1 = String
-    Class<?> k2 = Integer
-    Class<?> k3 = Double
-
-    expect:
-    CallDepthThreadLocalMap.incrementCallDepth(k1) == 0
-    CallDepthThreadLocalMap.incrementCallDepth(k2) == 0
-
-    CallDepthThreadLocalMap.incrementCallDepth(k1) == 1
-    CallDepthThreadLocalMap.incrementCallDepth(k2) == 1
-
-    when:
-    CallDepthThreadLocalMap.reset(k1)
-
-    then:
-    CallDepthThreadLocalMap.incrementCallDepth(k2) == 2
-
-    when:
-    CallDepthThreadLocalMap.reset(k2)
-
-    then:
-    CallDepthThreadLocalMap.incrementCallDepth(k1) == 0
-    CallDepthThreadLocalMap.incrementCallDepth(k2) == 0
-
-    CallDepthThreadLocalMap.incrementCallDepth(k1) == 1
-    CallDepthThreadLocalMap.incrementCallDepth(k2) == 1
-
-    expect:
-    CallDepthThreadLocalMap.decrementCallDepth(k1) == 1
-    CallDepthThreadLocalMap.decrementCallDepth(k2) == 1
-
-    CallDepthThreadLocalMap.decrementCallDepth(k1) == 0
-    CallDepthThreadLocalMap.decrementCallDepth(k2) == 0
-
-    and:
-    CallDepthThreadLocalMap.incrementCallDepth(k3) == 0
-    CallDepthThreadLocalMap.decrementCallDepth(k3) == 0
-
-  }
-}
diff --git a/javaagent-api/src/test/java/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.java b/javaagent-api/src/test/java/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.java
new file mode 100644
index 0000000000..9b255d714a
--- /dev/null
+++ b/javaagent-api/src/test/java/io/opentelemetry/javaagent/instrumentation/api/CallDepthThreadLocalMapTest.java
@@ -0,0 +1,42 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.api;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import org.junit.jupiter.api.Test;
+
+class CallDepthThreadLocalMapTest {
+
+  @Test
+  void incrementDecrement() {
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(String.class)).isZero();
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(Integer.class)).isZero();
+
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(String.class)).isOne();
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(Integer.class)).isOne();
+
+    CallDepthThreadLocalMap.reset(String.class);
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(Integer.class)).isEqualTo(2);
+
+    CallDepthThreadLocalMap.reset(Integer.class);
+
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(String.class)).isZero();
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(Integer.class)).isZero();
+
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(String.class)).isOne();
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(Integer.class)).isOne();
+
+    assertThat(CallDepthThreadLocalMap.decrementCallDepth(String.class)).isOne();
+    assertThat(CallDepthThreadLocalMap.decrementCallDepth(Integer.class)).isOne();
+
+    assertThat(CallDepthThreadLocalMap.decrementCallDepth(String.class)).isZero();
+    assertThat(CallDepthThreadLocalMap.decrementCallDepth(Integer.class)).isZero();
+
+    assertThat(CallDepthThreadLocalMap.incrementCallDepth(Double.class)).isZero();
+    assertThat(CallDepthThreadLocalMap.decrementCallDepth(Double.class)).isZero();
+  }
+}
diff --git a/javaagent-api/src/test/groovy/io/opentelemetry/javaagent/instrumentation/api/instrumenter/PeerServiceAttributesExtractorTest.java b/javaagent-api/src/test/java/io/opentelemetry/javaagent/instrumentation/api/instrumenter/PeerServiceAttributesExtractorTest.java
similarity index 100%
rename from javaagent-api/src/test/groovy/io/opentelemetry/javaagent/instrumentation/api/instrumenter/PeerServiceAttributesExtractorTest.java
rename to javaagent-api/src/test/java/io/opentelemetry/javaagent/instrumentation/api/instrumenter/PeerServiceAttributesExtractorTest.java