Swift: Add some tests.

Author: geoffw0

swift/ql/test/query-tests/Security/CWE-089/other.swift

@@ -0,0 +1,64 @@
+
+// --- stubs ---
+
+struct URL
+{
+	init?(string: String) {}
+}
+
+extension String {
+	init(contentsOf: URL) throws {
+		self.init("")
+	}
+}
+
+class NSObject {
+}
+
+class NSString : NSObject {
+    init(string: String) { }
+}
+
+class Sql {
+}
+
+class MyDatabase {
+	init(sql code: String? = nil) { }
+
+	func execute1(_ sql: String) { }
+	func execute2(_ sql: String?) { }
+	func execute3(_ sql: NSString) { }
+	func execute4(_ sql: Sql) { }
+
+	func query(sql: String) { }
+	func query(sqlLiteral: String) { }
+	func query(sqlStatement: String) { }
+	func query(sqliteStatement: String) { }
+
+    // non-examples
+	func doSomething(sqlIndex: Int) { }
+	func doSomething(sqliteContext: Sql) { }
+}
+
+// --- tests ---
+
+func test_heuristic(db: MyDatabase) throws {
+	let remoteString = try String(contentsOf: URL(string: "http://example.com/")!)
+
+	_ = MyDatabase()
+	_ = MyDatabase(sql: "some_fixed_sql")
+	_ = MyDatabase(sql: remoteString) // BAD [NOT DETECTED]
+
+	db.execute1(remoteString) // BAD [NOT DETECTED]
+	db.execute2(remoteString) // BAD [NOT DETECTED]
+	db.execute3(NSString(string: remoteString)) // BAD [NOT DETECTED]
+	db.execute4(remoteString as! Sql) // BAD [NOT DETECTED]
+
+	db.query(sql: remoteString) // BAD [NOT DETECTED]
+	db.query(sqlLiteral: remoteString) // BAD [NOT DETECTED]
+	db.query(sqlStatement: remoteString) // BAD [NOT DETECTED]
+	db.query(sqliteStatement: remoteString) // BAD [NOT DETECTED]
+
+	db.doSomething(sqlIndex: Int(remoteString) ?? 0)
+	db.doSomething(sqliteContext: remoteString as! Sql)
+}