27 lines
663 B
Python
27 lines
663 B
Python
"""Define test environment."""
|
|
|
|
import cx_Oracle
|
|
import os
|
|
import sys
|
|
import unittest
|
|
|
|
lang = os.environ.get("NLS_LANG")
|
|
if lang is None:
|
|
os.environ["NLS_LANG"] = ".UTF8"
|
|
if sys.version_info[0] < 3:
|
|
input = raw_input
|
|
|
|
def GetValue(name, label):
|
|
value = os.environ.get("CX_ORACLE_" + name)
|
|
if value is None:
|
|
value = input(label + ": ")
|
|
if hasattr(cx_Oracle, "UNICODE") or sys.version_info[0] >= 3:
|
|
return value
|
|
return unicode(value)
|
|
|
|
USERNAME = GetValue("USERNAME", "user name")
|
|
PASSWORD = GetValue("PASSWORD", "password")
|
|
TNSENTRY = GetValue("TNSENTRY", "TNS entry")
|
|
ARRAY_SIZE = int(GetValue("ARRAY_SIZE", "array size"))
|
|
|