db2 test case

aml2732 · aml2732 · commit fa2bbf0ffd37 · 2017-10-31T09:49:43.000-05:00
diff --git a/tests/dat/db2action.py b/tests/dat/db2action.py
@@ -1,4 +1,35 @@
 import ibm_db
 
 def main(args):
+  #Connect to the DB2 database
+  conn = ibm_db.connect("DATABASE=TEST;HOSTNAME=172.17.0.1;PORT=50000;PROTOCOL=TCPIP;UID=DB2INST1;PWD=db2inst1-pwd;", "", "")
+  cmd = "DELETE FROM TESTTABLE WHERE name='Angela'"
+  result = ibm_db.exec_immediate(conn, cmd)
+  #output of the above can succeed or fail. Either is fine.
+
+  # Insert into TestTable
+  cmd = "INSERT INTO TESTTABLE (NAME, AGE, LOCATION) VALUES ('Angela', 27, 'Texas')"
+  result = ibm_db.exec_immediate(conn, cmd)
+  if not result:
+      return {"err":"error :"+cmd}
+
+  # Select from TestTable (the row just inserted)
+  cmd = "SELECT * FROM TESTTABLE WHERE name='Angela'"
+  result = ibm_db.exec_immediate(conn, cmd)
+  if not result:
+      return {"err":"error :"+cmd}
+  else:
+      ibm_db.fetch_both(result,0)
+      value = ibm_db.result(result,"NAME")
+      # Make sure the row was correctly inserted
+      if value != 'Angela' :
+          return {"err":"Expected name 'Angela', but instead found: "+ value}
+
+  # Delete the row from TestTable
+  cmd = "DELETE FROM TESTTABLE WHERE name='Angela'"
+  result = ibm_db.exec_immediate(conn, cmd)
+  if not result:
+      return {"err":"error :"+cmd}
+
+  # If no detected errors occurred so far; return Success status
   return {"message":"Successfully called db2 action"}
diff --git a/tests/dat/setup.sql b/tests/dat/setup.sql
@@ -0,0 +1,8 @@
+CREATE DATABASE TEST AUTOMATIC STORAGE YES USING CODESET UTF-8 TERRITORY US PAGESIZE 4096;
+CONNECT TO TEST;
+
+CREATE TABLE TESTTABLE (
+  NAME VARCHAR(20) NOT NULL,
+  AGE INTEGER NOT NULL,
+  LOCATION VARCHAR(20)
+);
