-
Notifications
You must be signed in to change notification settings - Fork 774
/
JcToolsTest.java
128 lines (104 loc) · 3.01 KB
/
JcToolsTest.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*
* Copyright The OpenTelemetry Authors
* SPDX-License-Identifier: Apache-2.0
*/
package io.opentelemetry.sdk.trace.internal;
import static org.assertj.core.api.Assertions.assertThat;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.ArrayList;
import java.util.Queue;
import java.util.concurrent.ArrayBlockingQueue;
import org.jctools.queues.MpscArrayQueue;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.junit.jupiter.MockitoExtension;
import org.mockito.junit.jupiter.MockitoSettings;
import org.mockito.quality.Strictness;
@ExtendWith(MockitoExtension.class)
@MockitoSettings(strictness = Strictness.LENIENT)
class JcToolsTest {
ArrayList<String> batch = new ArrayList<>(10);
@Test
void drain_ArrayBlockingQueue() {
// Arrange
batch.add("Test3");
Queue<String> queue = new ArrayBlockingQueue<>(10);
queue.add("Test1");
queue.add("Test2");
// Act
JcTools.drain(queue, 5, batch::add);
// Assert
assertThat(batch).hasSize(3);
assertThat(queue).hasSize(0);
}
@Test
void drain_MessagePassingQueue() {
// Arrange
batch.add("Test3");
Queue<String> queue = new MpscArrayQueue<>(10);
queue.add("Test1");
queue.add("Test2");
// Act
JcTools.drain(queue, 5, batch::add);
// Assert
assertThat(batch).hasSize(3);
assertThat(queue).hasSize(0);
}
@Test
void drain_MaxBatch() {
// Arrange
Queue<String> queue = new MpscArrayQueue<>(10);
queue.add("Test1");
queue.add("Test2");
// Act
JcTools.drain(queue, 1, batch::add);
// Assert
assertThat(batch).hasSize(1);
assertThat(queue).hasSize(1);
}
@Test
void newFixedSize_MpscQueue() {
// Arrange
int capacity = 10;
// Act
Queue<Object> objects = JcTools.newFixedSizeQueue(capacity);
// Assert
assertThat(objects).isInstanceOf(MpscArrayQueue.class);
}
@Test
void capacity_MpscQueue() {
// Arrange
int capacity = 10;
Queue<Object> queue = JcTools.newFixedSizeQueue(capacity);
// Act
long queueSize = JcTools.capacity(queue);
// Assert
assertThat(queueSize).isGreaterThan(capacity);
}
@Test
void capacity_ArrayBlockingQueue() {
// Arrange
Queue<String> queue = new ArrayBlockingQueue<>(10);
// Act
long queueSize = JcTools.capacity(queue);
// Assert
assertThat(queueSize).isEqualTo(10);
}
@Test
void newFixedSizeQueue_works_with_sunMisc_prohibited() {
Assertions.assertNull(System.getSecurityManager());
SunMiscProhibitedSecurityManager testingSecurityManager =
new SunMiscProhibitedSecurityManager();
try {
System.setSecurityManager(testingSecurityManager);
Queue<Object> queue =
AccessController.doPrivileged(
(PrivilegedAction<Queue<Object>>) () -> JcTools.newFixedSizeQueue(10));
Assertions.assertNotNull(queue);
} finally {
System.setSecurityManager(null);
}
}
}