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 
31 import java.util.function.Consumer;
32 
33 import jdk.jshell.DeclarationSnippet;
34 import jdk.jshell.ImportSnippet;
35 import jdk.jshell.MethodSnippet;
36 import jdk.jshell.Snippet;
37 import jdk.jshell.TypeDeclSnippet;
38 import jdk.jshell.VarSnippet;
39 import org.testng.annotations.Test;
40 
41 import static org.testng.Assert.fail;
42 
43 @Test
44 public class JShellStateClosedTest extends KullaTesting {
45 
testStateClosedException(Runnable action)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 
testClasses()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 
testVariables()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 
testMethods()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 
testImports()77     public void testImports() {
78         ImportSnippet simp = importKey(assertEval("import java.lang.reflect.*;"));
79         getState().close();
80         assertStreamMatch(getState().imports(), simp);
81     }
82 
testSnippets()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 
testEval()95     public void testEval() {
96         testStateClosedException(() -> getState().eval("int a;"));
97     }
98 
testStateClosedException(Consumer<Snippet> action)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 
testStateClosedWithoutException(Consumer<Snippet> action)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 
testStatus()120     public void testStatus() {
121         testStateClosedWithoutException((key) -> getState().status(key));
122     }
123 
testVarValue()124     public void testVarValue() {
125         testStateClosedException((key) -> getState().varValue((VarSnippet) key));
126     }
127 
testDrop()128     public void testDrop() {
129         testStateClosedException((key) -> getState().drop(key));
130     }
131 
testUnresolved()132     public void testUnresolved() {
133         testStateClosedWithoutException((key) -> getState().unresolvedDependencies((DeclarationSnippet) key));
134     }
135 
testDiagnostics()136     public void testDiagnostics() {
137         testStateClosedWithoutException((key) -> getState().diagnostics(key));
138     }
139 }
140