Fix tests

Author: Sauyon Lee

java/ql/test/library-tests/frameworks/spring/cache/Test.java

@@ -1,15 +1,45 @@
 package generatedtest;
 
+import java.util.Map;
+import java.util.HashMap;
 import java.util.concurrent.Callable;
 import org.springframework.cache.Cache;
 
+
 // Test case generated by GenerateFlowTestCase.ql
 public class Test {
 
-	Object getMapKey(Object container) { return null; }
-	Object getMapValue(Object container) { return null; }
-	Object newWithMapKey(Object element) { return null; }
-	Object newWithMapValue(Object element) { return null; }
+	public class ValueWrapper extends HashMap<Object,Object> implements Cache.ValueWrapper {
+		ValueWrapper(Object element) {
+			super();
+			this.put(null, element);
+		}
+
+		public Object get() { return this.get(null); }
+	}
+
+	public class DummyCache implements Cache {
+		DummyCache(Object key, Object value) {
+			this.put(key, value);
+		}
+
+		public void clear() {}
+		public void evict(Object key) {}
+		public boolean evictIfPresent(Object key) { return false; }
+		public Cache.ValueWrapper get(Object key) { return null; }
+		public <T> T get(Object key, Callable<T> valueLoader) { return null; }
+		public <T> T get(Object key, Class<T> type) { return null; }
+		public String getName() { return null; }
+		public Object getNativeCache() { return null; }
+		//public default boolean invalidate() { return false; }
+		public void put(Object key, Object value) {}
+		//default Cache.ValueWrapper putIfAbsent(Object key, Object value) { return null; }
+	}
+
+	Object getMapKey(Cache.ValueRetrievalException container) { return container.getKey(); }
+	Object getMapKey(Cache container) { return ((Map)container.getNativeCache()).keySet().iterator().next(); }
+	Object getMapValue(Cache container) { return container.get(null, (Class)null); }
+	Object getMapValue(Cache.ValueWrapper container) { return container.get(); }
 	Object source() { return null; }
 	void sink(Object o) { }
 
@@ -19,57 +49,57 @@ public void test() {
 			// "org.springframework.cache;Cache$ValueRetrievalException;false;ValueRetrievalException;;;Argument[0];MapKey of Argument[-1];value"
 			Cache.ValueRetrievalException out = null;
 			Object in = source();
-			out = new ValueRetrievalException(in, null, null);
+			out = new Cache.ValueRetrievalException(in, null, null);
 			sink(getMapKey(out)); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache$ValueRetrievalException;false;getKey;;;MapKey of Argument[-1];ReturnValue;value"
 			Object out = null;
-			Cache.ValueRetrievalException in = (Cache.ValueRetrievalException)newWithMapKey(source());
+			Cache.ValueRetrievalException in = new Cache.ValueRetrievalException(source(), null, null);
 			out = in.getKey();
 			sink(out); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache$ValueWrapper;true;get;;;MapValue of Argument[-1];ReturnValue;value"
 			Object out = null;
-			Cache.ValueWrapper in = (Cache.ValueWrapper)newWithMapValue(source());
+			Cache.ValueWrapper in = new ValueWrapper(source());
 			out = in.get();
 			sink(out); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache;true;get;(Object);;MapValue of Argument[-1];MapValue of ReturnValue;value"
 			Cache.ValueWrapper out = null;
-			Cache in = (Cache)newWithMapValue(source());
+			Cache in = new DummyCache(null, source());
 			out = in.get(null);
 			sink(getMapValue(out)); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache;true;get;(Object,Callable);;MapValue of Argument[-1];ReturnValue;value"
 			Object out = null;
-			Cache in = (Cache)newWithMapValue(source());
+			Cache in = new DummyCache(null, source());
 			out = in.get(null, (Callable)null);
 			sink(out); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache;true;get;(Object,Class);;MapValue of Argument[-1];ReturnValue;value"
 			Object out = null;
-			Cache in = (Cache)newWithMapValue(source());
+			Cache in = new DummyCache(null, source());
 			out = in.get(null, (Class)null);
 			sink(out); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache;true;getNativeCache;;;MapKey of Argument[-1];MapKey of ReturnValue;value"
 			Object out = null;
-			Cache in = (Cache)newWithMapKey(source());
+			Cache in = new DummyCache(source(), null);
 			out = in.getNativeCache();
-			sink(getMapKey(out)); // $hasValueFlow
+			sink(getMapKey((Cache)out)); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache;true;getNativeCache;;;MapValue of Argument[-1];MapValue of ReturnValue;value"
 			Object out = null;
-			Cache in = (Cache)newWithMapValue(source());
+			Cache in = new DummyCache(null, source());
 			out = in.getNativeCache();
-			sink(getMapValue(out)); // $hasValueFlow
+			sink(getMapValue((Cache)out)); // $hasValueFlow
 		}
 		{
 			// "org.springframework.cache;Cache;true;put;;;Argument[0];MapKey of Argument[-1];value"
@@ -102,11 +132,11 @@ public void test() {
 		{
 			// "org.springframework.cache;Cache;true;putIfAbsent;;;MapValue of Argument[-1];MapValue of ReturnValue;value"
 			Cache.ValueWrapper out = null;
-			Cache in = (Cache)newWithMapValue(source());
+			Cache in = new DummyCache(null, source());
 			out = in.putIfAbsent(null, null);
 			sink(getMapValue(out)); // $hasValueFlow
 		}
 
 	}
 
-}
+}

java/ql/test/library-tests/frameworks/spring/cache/test.ql

@@ -1,22 +1,8 @@
 import java
 import semmle.code.java.dataflow.DataFlow
-import semmle.code.java.dataflow.ExternalFlow
 import semmle.code.java.dataflow.TaintTracking
 import TestUtilities.InlineExpectationsTest
 
-class SummaryModelTest extends SummaryModelCsv {
-  override predicate row(string row) {
-    row =
-      [
-        //"package;type;overrides;name;signature;ext;inputspec;outputspec;kind",
-        "generatedtest;Test;false;getMapKey;;;MapKey of Argument[0];ReturnValue;value",
-        "generatedtest;Test;false;newWithMapKey;;;Argument[0];MapKey of ReturnValue;value",
-        "generatedtest;Test;false;getMapValue;;;MapValue of Argument[0];ReturnValue;value",
-        "generatedtest;Test;false;newWithMapValue;;;Argument[0];MapValue of ReturnValue;value"
-      ]
-  }
-}
-
 class ValueFlowConf extends DataFlow::Configuration {
   ValueFlowConf() { this = "qltest:valueFlowConf" }