Skip to content

Commit

Permalink
Convert rocketmq unit test from groovy to java (#10520)
Browse files Browse the repository at this point in the history
  • Loading branch information
steverao authored Feb 20, 2024
1 parent 3b15791 commit b14f1fe
Show file tree
Hide file tree
Showing 8 changed files with 563 additions and 456 deletions.

This file was deleted.

Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/

package io.opentelemetry.instrumentation.rocketmqclient.v4_8;

import io.opentelemetry.instrumentation.testing.junit.AgentInstrumentationExtension;
import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.junit.jupiter.api.extension.RegisterExtension;

class RocketMqClientTest extends AbstractRocketMqClientTest {

@RegisterExtension
private static final InstrumentationExtension testing = AgentInstrumentationExtension.create();

@Override
InstrumentationExtension testing() {
return testing;
}

@Override
void configureMqProducer(DefaultMQProducer producer) {}

@Override
void configureMqPushConsumer(DefaultMQPushConsumer consumer) {}
}

This file was deleted.

Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/

package io.opentelemetry.instrumentation.rocketmqclient.v4_8;

import static java.util.Collections.singletonList;

import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
import io.opentelemetry.instrumentation.testing.junit.LibraryInstrumentationExtension;
import org.apache.rocketmq.client.consumer.DefaultMQPushConsumer;
import org.apache.rocketmq.client.producer.DefaultMQProducer;
import org.junit.jupiter.api.extension.RegisterExtension;

class RocketMqClientTest extends AbstractRocketMqClientTest {

@RegisterExtension
private static final InstrumentationExtension testing = LibraryInstrumentationExtension.create();

@Override
InstrumentationExtension testing() {
return testing;
}

@Override
@SuppressWarnings("deprecation")
// testing instrumentation of deprecated class
void configureMqProducer(DefaultMQProducer producer) {
producer
.getDefaultMQProducerImpl()
.registerSendMessageHook(
RocketMqTelemetry.builder(testing.getOpenTelemetry())
.setCapturedHeaders(singletonList("test-message-header"))
.setCaptureExperimentalSpanAttributes(true)
.build()
.newTracingSendMessageHook());
}

@Override
@SuppressWarnings("deprecation")
// testing instrumentation of deprecated class
void configureMqPushConsumer(DefaultMQPushConsumer consumer) {
consumer
.getDefaultMQPushConsumerImpl()
.registerConsumeMessageHook(
RocketMqTelemetry.builder(testing.getOpenTelemetry())
.setCapturedHeaders(singletonList("test-message-header"))
.setCaptureExperimentalSpanAttributes(true)
.build()
.newTracingConsumeMessageHook());
}
}
Loading

0 comments on commit b14f1fe

Please sign in to comment.