def test_2402_rollback_on_del(self):
     "2402 - connection rolls back before being destroyed"
     pool = base.get_pool()
     connection = pool.acquire()
     cursor = connection.cursor()
     cursor.execute("truncate table TestTempTable")
     cursor.execute("insert into TestTempTable (IntCol) values (1)")
     pool = base.get_pool()
     connection = pool.acquire()
     cursor = connection.cursor()
     cursor.execute("select count(*) from TestTempTable")
     count, = cursor.fetchone()
     self.assertEqual(count, 0)
 def test_2403_rollback_on_release(self):
     "2403 - connection rolls back before released back to the pool"
     pool = base.get_pool()
     connection = pool.acquire()
     cursor = connection.cursor()
     cursor.execute("truncate table TestTempTable")
     cursor.execute("insert into TestTempTable (IntCol) values (1)")
     cursor.close()
     pool.release(connection)
     pool = base.get_pool()
     connection = pool.acquire()
     cursor = connection.cursor()
     cursor.execute("select count(*) from TestTempTable")
     count, = cursor.fetchone()
     self.assertEqual(count, 0)
 def test_2409_heterogeneous_wrong_password(self):
     "2409 - test heterogeneous pool with wrong password specified"
     pool = base.get_pool(min=2, max=8, increment=3,
                          getmode=oracledb.SPOOL_ATTRVAL_WAIT,
                          homogeneous=False)
     self.assertRaises(oracledb.DatabaseError, pool.acquire,
                       base.get_proxy_user(), "this is the wrong password")
    def test_2406_purity(self):
        "2406 - test session pool with various types of purity"
        pool = base.get_pool(min=1, max=8, increment=1,
                             getmode=oracledb.SPOOL_ATTRVAL_WAIT)

        # get connection and set the action
        action = "TEST_ACTION"
        connection = pool.acquire()
        connection.action = action
        cursor = connection.cursor()
        cursor.execute("select 1 from dual")
        cursor.close()
        pool.release(connection)
        self.assertEqual(pool.opened, 1, "opened (1)")

        # verify that the connection still has the action set on it
        connection = pool.acquire()
        cursor = connection.cursor()
        cursor.execute("select sys_context('userenv', 'action') from dual")
        result, = cursor.fetchone()
        self.assertEqual(result, action)
        cursor.close()
        pool.release(connection)
        self.assertEqual(pool.opened, 1, "opened (2)")

        # get a new connection with new purity (should not have state)
        connection = pool.acquire(purity=oracledb.ATTR_PURITY_NEW)
        cursor = connection.cursor()
        cursor.execute("select sys_context('userenv', 'action') from dual")
        result, = cursor.fetchone()
        self.assertEqual(result, None)
        cursor.close()
        self.assertEqual(pool.opened, 2, "opened (3)")
        pool.drop(connection)
        self.assertEqual(pool.opened, 1, "opened (4)")
 def test_2412_tagging_invalid_key(self):
     "2412 - testTagging with Invalid key"
     pool = base.get_pool(getmode=oracledb.SPOOL_ATTRVAL_NOWAIT)
     conn = pool.acquire()
     self.assertRaises(TypeError, pool.release, conn, tag=12345)
     if base.get_client_version() >= (12, 2):
         self.assertRaises(oracledb.DatabaseError, pool.release, conn,
                           tag="INVALID_TAG")
 def test_2401_proxy_auth(self):
     "2401 - test that proxy authentication is possible"
     pool = base.get_pool(min=2, max=8, increment=3,
                          getmode=oracledb.SPOOL_ATTRVAL_WAIT)
     self.assertEqual(pool.homogeneous, True,
                      "homogeneous should be True by default")
     self.assertRaises(oracledb.DatabaseError, pool.acquire,
                       user="******")
     pool = base.get_pool(min=2, max=8, increment=3,
                          getmode=oracledb.SPOOL_ATTRVAL_WAIT,
                          homogeneous=False)
     msg = "homogeneous should be False after setting it in the constructor"
     self.assertEqual(pool.homogeneous, False, msg)
     connection = pool.acquire(user=base.get_proxy_user())
     cursor = connection.cursor()
     cursor.execute('select user from dual')
     result, = cursor.fetchone()
     self.assertEqual(result, base.get_proxy_user().upper())
 def test_2405_threading_with_errors(self):
     "2405 - test session pool with multiple threads (with errors)"
     self.pool = base.get_pool(min=5, max=20, increment=2, threaded=True,
                               getmode=oracledb.SPOOL_ATTRVAL_WAIT)
     threads = []
     for i in range(20):
         thread = threading.Thread(None, self.__connect_and_generate_error)
         threads.append(thread)
         thread.start()
     for thread in threads:
         thread.join()
 def test_2408_heterogenous_without_user(self):
     "2408 - test heterogeneous pool without user and password specified"
     pool = base.get_pool(user="", password="", min=2, max=8, increment=3,
                          getmode=oracledb.SPOOL_ATTRVAL_WAIT,
                          homogeneous=False)
     self.__verify_connection(pool.acquire(base.get_main_user(),
                              base.get_main_password()),
                              base.get_main_user())
     self.__verify_connection(pool.acquire(base.get_proxy_user(),
                              base.get_proxy_password()),
                              base.get_proxy_user())
     user_str = "%s[%s]" % (base.get_main_user(), base.get_proxy_user())
     self.__verify_connection(pool.acquire(user_str,
                              base.get_main_password()),
                              base.get_proxy_user(), base.get_main_user())
 def test_2407_heterogeneous(self):
     "2407 - test heterogeneous pool with user and password specified"
     pool = base.get_pool(min=2, max=8, increment=3, homogeneous=False,
                          getmode=oracledb.SPOOL_ATTRVAL_WAIT)
     self.assertEqual(pool.homogeneous, 0)
     self.__verify_connection(pool.acquire(), base.get_main_user())
     self.__verify_connection(pool.acquire(base.get_main_user(),
                              base.get_main_password()),
                              base.get_main_user())
     self.__verify_connection(pool.acquire(base.get_proxy_user(),
                              base.get_proxy_password()),
                              base.get_proxy_user())
     user_str = "%s[%s]" % (base.get_main_user(), base.get_proxy_user())
     self.__verify_connection(pool.acquire(user_str,
                              base.get_main_password()),
                              base.get_proxy_user(), base.get_main_user())
 def test_2400_pool(self):
     "2400 - test that the pool is created and has the right attributes"
     pool = base.get_pool(min=2, max=8, increment=3,
                          getmode=oracledb.SPOOL_ATTRVAL_WAIT)
     self.assertEqual(pool.username, base.get_main_user(),
                      "user name differs")
     self.assertEqual(pool.tnsentry, base.get_connect_string(),
                      "tnsentry differs")
     self.assertEqual(pool.dsn, base.get_connect_string(), "dsn differs")
     self.assertEqual(pool.max, 8, "max differs")
     self.assertEqual(pool.min, 2, "min differs")
     self.assertEqual(pool.increment, 3, "increment differs")
     self.assertEqual(pool.opened, 2, "opened differs")
     self.assertEqual(pool.busy, 0, "busy not 0 at start")
     connection_1 = pool.acquire()
     self.assertEqual(pool.busy, 1, "busy not 1 after acquire")
     self.assertEqual(pool.opened, 2, "opened not unchanged (1)")
     connection_2 = pool.acquire()
     self.assertEqual(pool.busy, 2, "busy not 2 after acquire")
     self.assertEqual(pool.opened, 2, "opened not unchanged (2)")
     connection_3 = pool.acquire()
     self.assertEqual(pool.busy, 3, "busy not 3 after acquire")
     self.assertEqual(pool.opened, 5, "opened not changed (3)")
     pool.release(connection_3)
     self.assertEqual(pool.busy, 2, "busy not 2 after release")
     del connection_2
     self.assertEqual(pool.busy, 1, "busy not 1 after del")
     pool.getmode = oracledb.SPOOL_ATTRVAL_NOWAIT
     self.assertEqual(pool.getmode, oracledb.SPOOL_ATTRVAL_NOWAIT)
     if base.get_client_version() >= (12, 2):
         pool.getmode = oracledb.SPOOL_ATTRVAL_TIMEDWAIT
         self.assertEqual(pool.getmode, oracledb.SPOOL_ATTRVAL_TIMEDWAIT)
     pool.stmtcachesize = 50
     self.assertEqual(pool.stmtcachesize, 50)
     pool.timeout = 10
     self.assertEqual(pool.timeout, 10)
     if base.get_client_version() >= (12, 1):
         pool.max_lifetime_session = 10
         self.assertEqual(pool.max_lifetime_session, 10)
    def test_2411_plsql_session_callbacks(self):
        "2411 - test PL/SQL session callbacks"
        if base.get_client_version() < (12, 2):
            self.skipTest("PL/SQL session callbacks not supported before 12.2")
        pool = base.get_pool(min=2, max=8, increment=3,
                             getmode=oracledb.SPOOL_ATTRVAL_NOWAIT,
                             sessionCallback="pkg_SessionCallback.TheCallback")
        tags = [
            "NLS_DATE_FORMAT=SIMPLE",
            "NLS_DATE_FORMAT=FULL;TIME_ZONE=UTC",
            "NLS_DATE_FORMAT=FULL;TIME_ZONE=MST"
        ]
        actual_tags = [None, None, "NLS_DATE_FORMAT=FULL;TIME_ZONE=UTC"]

        # truncate PL/SQL session callback log
        conn = pool.acquire()
        cursor = conn.cursor()
        cursor.execute("truncate table PLSQLSessionCallbacks")
        conn.close()

        # request sessions with each of the first two tags
        for tag in tags[:2]:
            conn = pool.acquire(tag=tag)
            conn.close()

        # for the last tag, use the matchanytag flag
        conn = pool.acquire(tag=tags[2], matchanytag=True)
        conn.close()

        # verify the PL/SQL session callback log is accurate
        conn = pool.acquire()
        cursor = conn.cursor()
        cursor.execute("""
                select RequestedTag, ActualTag
                from PLSQLSessionCallbacks
                order by FixupTimestamp""")
        results = cursor.fetchall()
        expected_results = list(zip(tags, actual_tags))
        self.assertEqual(results, expected_results)
    def test_2410_tagging_session(self):
        "2410 - test tagging a session"
        pool = base.get_pool(min=2, max=8, increment=3,
                             getmode=oracledb.SPOOL_ATTRVAL_NOWAIT)

        tag_mst = "TIME_ZONE=MST"
        tag_utc = "TIME_ZONE=UTC"

        conn = pool.acquire()
        self.assertEqual(conn.tag, None)
        pool.release(conn, tag=tag_mst)

        conn = pool.acquire()
        self.assertEqual(conn.tag, None)
        conn.tag = tag_utc
        conn.close()

        conn = pool.acquire(tag=tag_mst)
        self.assertEqual(conn.tag, tag_mst)
        conn.close()

        conn = pool.acquire(tag=tag_utc)
        self.assertEqual(conn.tag, tag_utc)
        conn.close()