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
|
package org.jgroups.blocks;
import org.jgroups.Global;
import org.jgroups.JChannel;
import org.jgroups.tests.ChannelTestBase;
import org.jgroups.util.Util;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.util.Map;
/**
* Testcase for the DistributedLockManager
*
* @author Robert Schaffar-Taurok (robert@fusion.at)
*/
@Test(groups=Global.STACK_DEPENDENT,sequential=true)
public class DistributedLockManagerTest extends ChannelTestBase {
private JChannel channel1;
private JChannel channel2;
protected VotingAdapter adapter1;
protected VotingAdapter adapter2;
protected LockManager lockManager1;
protected LockManager lockManager2;
@BeforeClass
void setUp() throws Exception {
channel1=createChannel(true);
adapter1=new VotingAdapter(channel1);
channel1.connect("DistributedLockManagerTest");
lockManager1=new DistributedLockManager(adapter1, "1");
Util.sleep(1000); // give some time for the channel to become a coordinator
channel2=createChannel(channel1);
adapter2=new VotingAdapter(channel2);
lockManager2=new DistributedLockManager(adapter2, "2");
channel2.connect("DistributedLockManagerTest");
Util.sleep(1000);
}
@AfterClass
void tearDown() throws Exception {
channel2.close();
channel1.close();
}
public void test() throws Exception {
lockManager1.lock("obj1", "owner1", 10000);
try {
lockManager1.lock("obj1", "owner2", 10000);
throw new IllegalStateException("obj1 should not be locked");
}
catch (LockNotGrantedException ex) {
System.out.println("got a lock not granted exception - expected");
}
lockManager2.lock("obj2", "owner2", 1000);
lockManager1.unlock("obj1", "owner1");
try {
lockManager1.unlock("obj2", "owner1");
throw new IllegalStateException("obj2 should not be released");
}
catch (LockNotReleasedException ex) {
System.out.println("got a lock not released exception, as expected");
}
lockManager1.unlock("obj2", "owner2");
}
public void testMultiLock() throws Exception {
lockManager1.lock("obj1", "owner1", 10000);
// Override private members and simulate the errorcase which is, when two lockManagers have locked the same object
// This can happen after a merge
Class<?> acquireLockDecreeClass = Class.forName("org.jgroups.blocks.DistributedLockManager$AcquireLockDecree");
Constructor<?> acquireLockDecreeConstructor = acquireLockDecreeClass.getDeclaredConstructor(new Class[] {Object.class, Object.class, Object.class});
acquireLockDecreeConstructor.setAccessible(true);
Object acquireLockDecree = acquireLockDecreeConstructor.newInstance("obj1", "owner2", "2");
Field heldLocksField = lockManager2.getClass().getDeclaredField("heldLocks");
heldLocksField.setAccessible(true);
Map<String,Object> heldLocks = (Map<String,Object>)heldLocksField.get(lockManager2);
heldLocks.put("obj1", acquireLockDecree);
// Both lockManagers hold a lock on obj1 now
try {
lockManager1.unlock("obj1", "owner1", true);
throw new IllegalStateException("obj1 should throw a lockMultiLockedException upon release");
} catch (LockMultiLockedException e) {
// everything is ok
}
try {
lockManager1.lock("obj1", "owner1", 10000);
throw new IllegalStateException("obj1 should throw a LockNotGrantedException because it is still locked by lockManager2");
} catch (LockNotGrantedException e) {
// everything is ok
}
try {
lockManager2.unlock("obj1", "owner2", true);
throw new IllegalStateException("obj1 should throw a lockMultiLockedException upon release");
} catch (LockMultiLockedException e) {
// everything is ok
}
// Everything should be unlocked now
try {
lockManager1.lock("obj1", "owner1", 10000);
}
catch (LockNotGrantedException e) {
throw new IllegalStateException("obj1 should be unlocked");
}
lockManager1.unlock("obj1", "owner1", true);
}
}
|
