JShellStateClosedTest.java revision 3695:40468274ff3b 
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 8164277
26 * @summary Testing IllegalStateException.
27 * @build KullaTesting TestingInputStream JShellStateClosedTest
28 * @run testng JShellStateClosedTest
29 */
30
31import java.util.function.Consumer;
32
33import jdk.jshell.DeclarationSnippet;
34import jdk.jshell.ImportSnippet;
35import jdk.jshell.MethodSnippet;
36import jdk.jshell.Snippet;
37import jdk.jshell.TypeDeclSnippet;
38import jdk.jshell.VarSnippet;
39import org.testng.annotations.Test;
40
41import static org.testng.Assert.fail;
42
43@Test
44public class JShellStateClosedTest extends KullaTesting {
45
46    private void testStateClosedException(Runnable action) {
47        getState().close();
48        try {
49            action.run();
50            fail("Exception expected");
51        } catch (IllegalStateException e) {
52            // Expected
53        }
54    }
55
56    public void testClasses() {
57        TypeDeclSnippet sc = classKey(assertEval("class C { }"));
58        TypeDeclSnippet si = classKey(assertEval("interface I { }"));
59        getState().close();
60        assertStreamMatch(getState().types(), sc, si);
61    }
62
63    public void testVariables() {
64        VarSnippet sx = varKey(assertEval("int x = 5;"));
65        VarSnippet sfoo = varKey(assertEval("String foo;"));
66        getState().close();
67        assertStreamMatch(getState().variables(), sx, sfoo);
68    }
69
70    public void testMethods() {
71        MethodSnippet smm = methodKey(assertEval("int mm() { return 6; }"));
72        MethodSnippet svv = methodKey(assertEval("void vv() { }"));
73        getState().close();
74        assertStreamMatch(getState().methods(), smm, svv);
75    }
76
77    public void testImports() {
78        ImportSnippet simp = importKey(assertEval("import java.lang.reflect.*;"));
79        getState().close();
80        assertStreamMatch(getState().imports(), simp);
81    }
82
83    public void testSnippets() {
84        VarSnippet sx = varKey(assertEval("int x = 5;"));
85        VarSnippet sfoo = varKey(assertEval("String foo;"));
86        MethodSnippet smm = methodKey(assertEval("int mm() { return 6; }"));
87        MethodSnippet svv = methodKey(assertEval("void vv() { }"));
88        TypeDeclSnippet sc = classKey(assertEval("class C { }"));
89        TypeDeclSnippet si = classKey(assertEval("interface I { }"));
90        ImportSnippet simp = importKey(assertEval("import java.lang.reflect.*;"));
91        getState().close();
92        assertStreamMatch(getState().snippets(), sx, sfoo, smm, svv, sc, si, simp);
93    }
94
95    public void testEval() {
96        testStateClosedException(() -> getState().eval("int a;"));
97    }
98
99    private void testStateClosedException(Consumer<Snippet> action) {
100        Snippet k = varKey(assertEval("int a;"));
101        getState().close();
102        try {
103            action.accept(k);
104            fail("IllegalStateException expected since closed");
105        } catch (IllegalStateException e) {
106            // Expected
107        }
108    }
109
110    private void testStateClosedWithoutException(Consumer<Snippet> action) {
111        Snippet k = varKey(assertEval("int a;"));
112        getState().close();
113        try {
114            action.accept(k);
115        } catch (IllegalStateException e) {
116            fail("Expected no IllegalStateException even though closed");
117        }
118    }
119
120    public void testStatus() {
121        testStateClosedWithoutException((key) -> getState().status(key));
122    }
123
124    public void testVarValue() {
125        testStateClosedException((key) -> getState().varValue((VarSnippet) key));
126    }
127
128    public void testDrop() {
129        testStateClosedException((key) -> getState().drop(key));
130    }
131
132    public void testUnresolved() {
133        testStateClosedWithoutException((key) -> getState().unresolvedDependencies((DeclarationSnippet) key));
134    }
135
136    public void testDiagnostics() {
137        testStateClosedWithoutException((key) -> getState().diagnostics(key));
138    }
139}
140