SnippetStatusListenerTest.java revision 3062:15bdc18525ff 
1/*
2 * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * @test
26 * @summary Subscribe tests
27 * @build KullaTesting TestingInputStream
28 * @run testng SnippetStatusListenerTest
29 */
30
31import java.util.ArrayList;
32import java.util.Collections;
33import java.util.List;
34import java.util.function.Consumer;
35
36import jdk.jshell.DeclarationSnippet;
37import jdk.jshell.JShell.Subscription;
38import jdk.jshell.SnippetEvent;
39import jdk.jshell.TypeDeclSnippet;
40import org.testng.annotations.Test;
41
42import static jdk.jshell.Snippet.Status.*;
43import static org.testng.Assert.assertEquals;
44
45@Test
46public class SnippetStatusListenerTest extends KullaTesting {
47
48    public void testTwoSnippetEventListeners() {
49        SnippetListener listener1 = new SnippetListener();
50        SnippetListener listener2 = new SnippetListener();
51
52        Subscription subscription1 = getState().onSnippetEvent(listener1);
53        getState().onSnippetEvent(listener2);
54
55        TypeDeclSnippet a = classKey(assertEval("class A {}"));
56        assertEval("interface A {}",
57                ste(MAIN_SNIPPET, VALID, VALID, true, null),
58                ste(a, VALID, OVERWRITTEN, false, null));
59        DeclarationSnippet f = (DeclarationSnippet) assertDeclareFail("double f() { }", "compiler.err.missing.ret.stmt");
60        assertEvalException("throw new RuntimeException();");
61        assertEval("int a = 0;");
62
63        List<SnippetEvent> events1 = Collections.unmodifiableList(listener1.getEvents());
64        assertEquals(events1, listener2.getEvents(), "Checking got events");
65        getState().unsubscribe(subscription1);
66
67        assertDrop(f, DiagCheck.DIAG_IGNORE, DiagCheck.DIAG_IGNORE, ste(f, REJECTED, DROPPED, false, null));
68        assertEval("void f() { }", ste(MAIN_SNIPPET, DROPPED, VALID, true, null));
69        assertEvalException("throw new RuntimeException();");
70        assertEquals(listener1.getEvents(), events1, "Checking that unsubscribed listener does not get events");
71
72        List<SnippetEvent> events2 = new ArrayList<>(listener2.getEvents());
73        events2.removeAll(events1);
74
75        assertEquals(events2.size(), 3, "The second listener got events");
76    }
77
78    @Test(expectedExceptions = NullPointerException.class)
79    public void testNullCallback() {
80        getState().onSnippetEvent(null);
81    }
82
83    @Test(expectedExceptions = IllegalStateException.class)
84    public void testSubscriptionAfterClose() {
85        getState().close();
86        getState().onSnippetEvent(e -> {});
87    }
88
89    @Test(expectedExceptions = IllegalStateException.class,
90          enabled = false) //TODO 8139873
91    public void testSubscriptionAfterShutdown() {
92        assertEval("System.exit(0);");
93        getState().onSnippetEvent(e -> {});
94    }
95
96    public void testSubscriptionToAnotherState() {
97        SnippetListener listener = new SnippetListener();
98        Subscription subscription = getState().onSnippetEvent(listener);
99        tearDown();
100        setUp();
101        assertEval("int x;");
102        assertEquals(Collections.emptyList(), listener.getEvents(), "No events");
103        getState().unsubscribe(subscription);
104    }
105
106    private static class SnippetListener implements Consumer<SnippetEvent> {
107        private final List<SnippetEvent> events = new ArrayList<>();
108
109        @Override
110        public void accept(SnippetEvent event) {
111            events.add(event);
112        }
113
114        public List<SnippetEvent> getEvents() {
115            return events;
116        }
117    }
118}
119