Skip to content

Commit 48f1b9b

Browse files
committed
parametrize test environment setup
1 parent dd12f86 commit 48f1b9b

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

src/test/java/com/uber/cadence/workflow/MetricsTest.java

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -108,21 +108,21 @@ public void executeChild() {
108108
}
109109
}
110110

111-
@Before
112-
public void setUp() {
111+
public void setUp(com.uber.m3.util.Duration reportingFrequecy){
113112
reporter = mock(StatsReporter.class);
114113
Scope scope =
115-
new RootScopeBuilder()
116-
.reporter(reporter)
117-
.reportEvery(com.uber.m3.util.Duration.ofMillis(300));
114+
new RootScopeBuilder()
115+
.reporter(reporter)
116+
.reportEvery(reportingFrequecy);
118117

119118
TestEnvironmentOptions testOptions =
120-
new Builder().setDomain(WorkflowTest.DOMAIN).setMetricsScope(scope).build();
119+
new Builder().setDomain(WorkflowTest.DOMAIN).setMetricsScope(scope).build();
121120
testEnvironment = TestWorkflowEnvironment.newInstance(testOptions);
122121
}
123122

124123
@Test
125124
public void testWorkflowMetrics() throws InterruptedException {
125+
setUp(com.uber.m3.util.Duration.ofMillis(10));
126126

127127
Worker worker = testEnvironment.newWorker(taskList);
128128
worker.registerWorkflowImplementationTypes(
@@ -166,6 +166,8 @@ public void testWorkflowMetrics() throws InterruptedException {
166166

167167
@Test
168168
public void testCorruptedSignalMetrics() throws InterruptedException {
169+
setUp(com.uber.m3.util.Duration.ofMillis(300));
170+
169171
Worker worker = testEnvironment.newWorker(taskList, builder ->
170172
builder.setInterceptorFactory(new CorruptedSignalWorkflowInterceptorFactory()));
171173

0 commit comments

Comments
 (0)