contract inference initial

java/java-tests/testSrc/com/intellij/codeInspection/ContractInferenceFromSourceTest.groovy

@@ -0,0 +1,182 @@
+/*
+ * Copyright 2000-2014 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.intellij.codeInspection
+
+import com.intellij.codeInspection.dataFlow.ContractInference
+import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase
+
+/**
+ * @author peter
+ */
+class ContractInferenceFromSourceTest extends LightCodeInsightFixtureTestCase {
+
+  void "test if null return null"() {
+    def c = inferContract("""
+  String smth(String s) {
+    if (s == null) return null;
+    return s.substring(1);
+  }
+""")
+    assert c == 'null -> null'
+  }
+
+  void "test if not null return true"() {
+    def c = inferContract("""
+  boolean smth(int a, String s) {
+    if (s != null) { return true; }
+    return a == 2;
+  }
+""")
+    assert c == '_, !null -> true'
+  }
+
+  void "test if null fail"() {
+    def c = inferContract("""
+  boolean smth(int a, String s) {
+    if (null == s) { throw new RuntimeException(); }
+    return a == 2;
+  }
+""")
+    assert c == '_, null -> fail'
+  }
+
+  void "test if true return the same"() {
+    def c = inferContract("""
+  boolean smth(boolean b, int a) {
+    if (b) return b;
+    return a == 2;
+  }
+""")
+    assert c == 'true, _ -> true'
+  }
+
+  void "test if false return negation"() {
+    def c = inferContract("""
+  boolean smth(boolean b, int a) {
+    if (!b) return !(b);
+    return a == 2;
+  }
+""")
+    assert c == 'false, _ -> true'
+  }
+
+  void "test nested if"() {
+    def c = inferContract("""
+  boolean smth(boolean b, Object o) {
+    if (!b) if (o != null) return true;
+    return a == 2;
+  }
+""")
+    assert c == 'false, !null -> true'
+  }
+
+  void "test conjunction"() {
+    def c = inferContract("""
+  boolean smth(boolean b, Object o) {
+    if (!b && o != null) return true;
+    return a == 2;
+  }
+""")
+    assert c == 'false, !null -> true'
+  }
+
+  void "test disjunction"() {
+    def c = inferContracts("""
+  boolean smth(boolean b, Object o) {
+    if (!b || o != null) return true;
+    return a == 2;
+  }
+""")
+    assert c == ['false, _ -> true', 'true, !null -> true']
+  }
+
+  void "test ternary"() {
+    def c = inferContracts("""
+  boolean smth(boolean b, Object o, Object o1) {
+    return (!b || o != null) ? true : (o1 != null && o1.hashCode() == 3);
+  }
+""")
+    assert c == ['false, _, _ -> true', 'true, !null, _ -> true', 'true, null, null -> false']
+  }
+
+  void "test instanceof"() {
+    def c = inferContracts("""
+  boolean smth(Object o) {
+    return o instanceof String;
+  }
+""")
+    assert c == ['null -> false']
+  }
+
+  void "test if-else"() {
+    def c = inferContracts("""
+  boolean smth(Object o) {
+    if (o instanceof String) return false;
+    else return true;
+  }
+""")
+    assert c == ['null -> true']
+  }
+
+  void "test if return without else"() {
+    def c = inferContracts("""
+  boolean smth(Object o) {
+    if (o instanceof String) return false;
+    return true;
+  }
+""")
+    assert c == ['null -> true']
+  }
+
+  void "test if no-return without else"() {
+    def c = inferContracts("""
+  boolean smth(Object o) {
+    if (o instanceof String) callSomething();
+    return true;
+  }
+""")
+    assert c == []
+  }
+
+  void "test assertion"() {
+    def c = inferContracts("""
+  boolean smth(Object o) {
+    assert o instanceof String;
+    return true;
+  }
+""")
+    assert c == ['null -> fail']
+  }
+
+  void "_test no NotNull duplication"() {
+    def c = inferContracts("""
+  boolean smth(@org.jetbrains.annotations.NotNull Object o) {
+    if (o == null) throw new RuntimeException();
+    return o.hashCode() == 1;
+  }
+""")
+    assert c == []
+  }
+
+  private String inferContract(String method) {
+    return assertOneElement(inferContracts(method))
+  }
+
+  private List<String> inferContracts(String method) {
+    def clazz = myFixture.addClass("class Foo { $method }")
+    return ContractInference.inferContracts(clazz.methods[0]).collect { it as String }
+  }
+}

java/java-tests/testSrc/com/intellij/codeInspection/DataFlowInspectionTestSuite.java

@@ -29,6 +29,7 @@ public class DataFlowInspectionTestSuite {
     suite.addTestSuite(DataFlowInspection8Test.class);
     suite.addTestSuite(DataFlowInspectionAncientTest.class);
     suite.addTestSuite(ContractCheckTest.class);
+    suite.addTestSuite(ContractInferenceFromSourceTest.class);
     suite.addTestSuite(SliceTreeTest.class);
     suite.addTestSuite(SliceBackwardTest.class);
     suite.addTestSuite(SmartTypeCompletionDfaTest.class);