add db2 tests (ibm-functions#12)

csantanapr · web-flow · commit 581aef8ae98f · 2017-12-12T23:19:16.000-05:00
diff --git a/tests/dat/db2/testDB2Service.py b/tests/dat/db2/testDB2Service.py
@@ -0,0 +1,30 @@
+"""Python test fixture to check that action can connect to db2 service"""
+
+import ibm_db
+
+def main(args):
+    ssldsn = args["__bx_creds"]["dashDB"]["ssldsn"]
+    conn = ibm_db.connect(ssldsn, "", "")
+    if conn:
+        print ("Connection succeeded.")
+    else:
+        print ("Connection failed.")
+        return {"error":"Error connecting to db2"}
+
+        # Select from TestTable (the row just inserted)
+    cmd = "SELECT HISP_DESC FROM SAMPLES.HISPANIC_ORIGIN WHERE HISP_CODE='03'"
+    result = ibm_db.exec_immediate(conn, cmd)
+
+    if not result:
+        return {"error":"error :"+cmd}
+    else:
+        ibm_db.fetch_both(result,0)
+        value = ibm_db.result(result,"HISP_DESC")
+        ibm_db.close(conn)
+        return {"HISP_DESC":value}
+
+
+if __name__ == "__main__":
+    # execute only if run as a script
+    input = {"__bx_creds":{"dashDB":{"ssldsn":"<ssldsn from credentials>"}}}
+    print(main(input))
diff --git a/tests/src/test/scala/integration/CredentialsIBMPythonDb2CloudTests.scala b/tests/src/test/scala/integration/CredentialsIBMPythonDb2CloudTests.scala
@@ -0,0 +1,59 @@
+/*
+ * Copyright 2017 IBM Corporation
+ *
+ * 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 integration
+
+import common._
+import org.junit.runner.RunWith
+import org.scalatest.junit.JUnitRunner
+import java.io.File
+import common.rest.WskRest
+import spray.json._
+
+@RunWith(classOf[JUnitRunner])
+class CredentialsIBMPythonDb2CloudTests extends TestHelpers with WskTestHelpers {
+
+  implicit val wskprops: WskProps = WskProps()
+  val defaultKind = Some("python-jessie:3")
+  val wsk = new WskRest
+  val datdir = System.getProperty("user.dir") + "/dat/db2"
+  val actionName = "testDB2Service"
+  val actionFileName = "testDB2Service.py"
+
+  val creds = TestUtils.getVCAPcredentials("dashDB")
+  val ssldsn = creds.get("ssldsn")
+  val __bx_creds = JsObject("dashDB" -> JsObject("ssldsn" -> JsString(ssldsn)))
+
+  it should "Test connection to DB2 on IBM Cloud" in withAssetCleaner(wskprops) { (wp, assetHelper) =>
+    val file = Some(new File(datdir, actionFileName).toString())
+
+    assetHelper.withCleaner(wsk.action, actionName) { (action, _) =>
+      action.create(
+        actionName,
+        file,
+        main = Some("main"),
+        kind = defaultKind,
+        parameters = Map("__bx_creds" -> __bx_creds))
+    }
+
+    withActivation(wsk.activation, wsk.action.invoke(actionName)) { activation =>
+      val response = activation.response
+      response.result.get.fields.get("error") shouldBe empty
+      response.result.get.fields.get("HISP_DESC") should be(Some(JsString("Puerto Rican")))
+    }
+
+  }
+
+}
diff --git a/tests/src/test/scala/integration/CredentialsIBMPythonWatsonTests.scala b/tests/src/test/scala/integration/CredentialsIBMPythonWatsonTests.scala
@@ -15,15 +15,12 @@
  */
 package integration
 
+import common._
 import org.junit.runner.RunWith
 import org.scalatest.junit.JUnitRunner
-import common.WskProps
 import java.io.File
 import spray.json._
-import common.TestUtils
 import common.rest.WskRest
-import common.TestHelpers
-import common.WskTestHelpers
 
 @RunWith(classOf[JUnitRunner])
 class CredentialsIBMPythonWatsonTests extends TestHelpers with WskTestHelpers {
