aboutsummaryrefslogtreecommitdiffstats
path: root/messagebus/src/test/java/com/yahoo/messagebus/MessengerTestCase.java
blob: cacbd08ab1c005f67a10b79b892895ad99993bbb (plain) (blame)
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
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.messagebus;

import org.junit.jupiter.api.Test;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import static org.junit.jupiter.api.Assertions.assertTrue;

/**
 * @author Simon Thoresen Hult
 */
public class MessengerTestCase {

    @Test
    void requireThatSyncWithSelfDoesNotCauseDeadLock() throws InterruptedException {
        final Messenger msn = new Messenger();
        msn.start();

        final CountDownLatch latch = new CountDownLatch(1);
        msn.enqueue(new Messenger.Task() {

            @Override
            public void run() {
                msn.sync();
            }

            @Override
            public void destroy() {
                latch.countDown();
            }
        });
        assertTrue(latch.await(60, TimeUnit.SECONDS));
    }

    @Test
    void requireThatTaskIsExecuted() throws InterruptedException {
        Messenger msn = new Messenger();
        msn.start();
        assertTrue(tryMessenger(msn));
    }

    @Test
    void requireThatRunExceptionIsCaught() throws InterruptedException {
        Messenger msn = new Messenger();
        msn.start();
        msn.enqueue(new Messenger.Task() {
            @Override
            public void run() {
                throw new RuntimeException();
            }

            @Override
            public void destroy() {

            }
        });
        assertTrue(tryMessenger(msn));
    }

    @Test
    void requireThatDestroyExceptionIsCaught() throws InterruptedException {
        Messenger msn = new Messenger();
        msn.start();
        msn.enqueue(new Messenger.Task() {
            @Override
            public void run() {

            }

            @Override
            public void destroy() {
                throw new RuntimeException();
            }
        });
        assertTrue(tryMessenger(msn));
    }

    @Test
    void requireThatRunAndDestroyExceptionsAreCaught() throws InterruptedException {
        Messenger msn = new Messenger();
        msn.start();
        msn.enqueue(new Messenger.Task() {
            @Override
            public void run() {
                throw new RuntimeException();
            }

            @Override
            public void destroy() {
                throw new RuntimeException();
            }
        });
        assertTrue(tryMessenger(msn));
    }

    private static boolean tryMessenger(Messenger msn) {
        MyTask task = new MyTask();
        msn.enqueue(task);
        try {
            return task.runLatch.await(60, TimeUnit.SECONDS) &&
                   task.destroyLatch.await(60, TimeUnit.SECONDS);
        } catch (InterruptedException e) {
            return false;
        }
    }

    private static class MyTask implements Messenger.Task {

        final CountDownLatch runLatch = new CountDownLatch(1);
        final CountDownLatch destroyLatch = new CountDownLatch(1);

        @Override
        public void run() {
            runLatch.countDown();
        }

        @Override
        public void destroy() {
            destroyLatch.countDown();
        }
    }
}