blob: 35299f55f44890ba3125e707a92fda3d1c489b39 (
plain)
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
// Copyright 2011 Google Inc. All Rights Reserved.
package com.google.common.testing;
import com.google.common.testing.GcFinalization.FinalizationPredicate;
import com.google.common.util.concurrent.SettableFuture;
import junit.framework.TestCase;
import java.lang.ref.WeakReference;
import java.util.WeakHashMap;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicBoolean;
/**
* Tests for {@link GcFinalization}.
*
* @author martinrb@google.com (Martin Buchholz)
* @author schmoe@google.com (mike nonemacher)
*/
public class GcFinalizationTest extends TestCase {
//----------------------------------------------------------------
// Ordinary tests of successful method execution
//----------------------------------------------------------------
public void testAwait_CountDownLatch() {
final CountDownLatch latch = new CountDownLatch(1);
Object x = new Object() {
protected void finalize() { latch.countDown(); }
};
x = null; // Hint to the JIT that x is unreachable
GcFinalization.await(latch);
assertEquals(0, latch.getCount());
}
public void testAwaitDone_Future() {
final SettableFuture<Void> future = SettableFuture.create();
Object x = new Object() {
protected void finalize() { future.set(null); }
};
x = null; // Hint to the JIT that x is unreachable
GcFinalization.awaitDone(future);
assertTrue(future.isDone());
assertFalse(future.isCancelled());
}
public void testAwaitDone_Future_Cancel() {
final SettableFuture<Void> future = SettableFuture.create();
Object x = new Object() {
protected void finalize() { future.cancel(false); }
};
x = null; // Hint to the JIT that x is unreachable
GcFinalization.awaitDone(future);
assertTrue(future.isDone());
assertTrue(future.isCancelled());
}
public void testAwaitClear() {
final WeakReference<Object> ref = new WeakReference<Object>(new Object());
GcFinalization.awaitClear(ref);
assertNull(ref.get());
}
public void testAwaitDone_FinalizationPredicate() {
final WeakHashMap<Object, Object> map = new WeakHashMap<Object, Object>();
map.put(new Object(), Boolean.TRUE);
GcFinalization.awaitDone(new FinalizationPredicate() {
public boolean isDone() {
return map.isEmpty();
}
});
assertTrue(map.isEmpty());
}
//----------------------------------------------------------------
// Test that interrupts result in RuntimeException, not InterruptedException.
// Trickier than it looks, because runFinalization swallows interrupts.
//----------------------------------------------------------------
class Interruptenator extends Thread {
final AtomicBoolean shutdown;
Interruptenator(final Thread interruptee) {
this(interruptee, new AtomicBoolean(false));
}
Interruptenator(final Thread interruptee,
final AtomicBoolean shutdown) {
super(new Runnable() {
public void run() {
while (!shutdown.get()) {
interruptee.interrupt();
Thread.yield();
}}});
this.shutdown = shutdown;
start();
}
void shutdown() {
shutdown.set(true);
while (this.isAlive()) {
Thread.yield();
}
}
}
void assertWrapsInterruptedException(RuntimeException e) {
assertTrue(e.getMessage().contains("Unexpected interrupt"));
assertTrue(e.getCause() instanceof InterruptedException);
}
public void testAwait_CountDownLatch_Interrupted() {
Interruptenator interruptenator = new Interruptenator(Thread.currentThread());
try {
final CountDownLatch latch = new CountDownLatch(1);
try {
GcFinalization.await(latch);
fail("should throw");
} catch (RuntimeException expected) {
assertWrapsInterruptedException(expected);
}
} finally {
interruptenator.shutdown();
Thread.interrupted();
}
}
public void testAwaitDone_Future_Interrupted_Interrupted() {
Interruptenator interruptenator = new Interruptenator(Thread.currentThread());
try {
final SettableFuture<Void> future = SettableFuture.create();
try {
GcFinalization.awaitDone(future);
fail("should throw");
} catch (RuntimeException expected) {
assertWrapsInterruptedException(expected);
}
} finally {
interruptenator.shutdown();
Thread.interrupted();
}
}
public void testAwaitClear_Interrupted() {
Interruptenator interruptenator = new Interruptenator(Thread.currentThread());
try {
final WeakReference<Object> ref = new WeakReference<Object>(Boolean.TRUE);
try {
GcFinalization.awaitClear(ref);
fail("should throw");
} catch (RuntimeException expected) {
assertWrapsInterruptedException(expected);
}
} finally {
interruptenator.shutdown();
Thread.interrupted();
}
}
public void testAwaitDone_FinalizationPredicate_Interrupted() {
Interruptenator interruptenator = new Interruptenator(Thread.currentThread());
try {
try {
GcFinalization.awaitDone(new FinalizationPredicate() {
public boolean isDone() {
return false;
}
});
fail("should throw");
} catch (RuntimeException expected) {
assertWrapsInterruptedException(expected);
}
} finally {
interruptenator.shutdown();
Thread.interrupted();
}
}
}
|
