4b8743ff1629ce4f9f9b0931201060e22283e04b
[nonrtric.git] / policy-agent / src / test / java / org / oransc / policyagent / repository / LockTest.java
1 /*-
2  * ========================LICENSE_START=================================
3  * O-RAN-SC
4  * %%
5  * Copyright (C) 2019 Nordix Foundation
6  * %%
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ========================LICENSE_END===================================
19  */
20
21 package org.oransc.policyagent.repository;
22
23 import static org.assertj.core.api.Assertions.assertThat;
24
25 import java.io.IOException;
26
27 import org.junit.jupiter.api.Test;
28 import org.junit.jupiter.api.extension.ExtendWith;
29 import org.mockito.junit.jupiter.MockitoExtension;
30 import org.oransc.policyagent.exceptions.ServiceException;
31 import org.oransc.policyagent.repository.Lock.LockType;
32
33 import reactor.core.publisher.Mono;
34 import reactor.test.StepVerifier;
35
36 @ExtendWith(MockitoExtension.class)
37 class LockTest {
38
39     @SuppressWarnings("squid:S2925") // "Thread.sleep" should not be used in tests.
40     private void sleep() {
41         try {
42             Thread.sleep(100);
43         } catch (InterruptedException e) {
44             // Do nothing.
45         }
46     }
47
48     private void asynchUnlock(Lock lock) {
49         Thread thread = new Thread(() -> {
50             sleep();
51             lock.unlockBlocking();
52         });
53         thread.start();
54     }
55
56     @Test
57     void testLock() throws IOException, ServiceException {
58         Lock lock = new Lock();
59         lock.lockBlocking(LockType.SHARED);
60         lock.unlockBlocking();
61
62         lock.lockBlocking(LockType.EXCLUSIVE);
63         asynchUnlock(lock);
64
65         lock.lockBlocking(LockType.SHARED);
66         lock.unlockBlocking();
67
68         assertThat(lock.getLockCounter()).isZero();
69     }
70
71     @Test
72     void testReactiveLock() {
73         Lock lock = new Lock();
74
75         Mono<Lock> seq = lock.lock(LockType.EXCLUSIVE) //
76             .flatMap(l -> lock.lock(LockType.EXCLUSIVE)) //
77             .flatMap(l -> lock.unlock());
78
79         asynchUnlock(lock);
80         StepVerifier.create(seq) //
81             .expectSubscription() //
82             .expectNext(lock) //
83             .verifyComplete();
84
85         assertThat(lock.getLockCounter()).isZero();
86
87     }
88
89 }