def test_conn_params(self):
conn_params = {
'database': 'json-rpc.php'
}
conn = Database.connect(**conn_params)
self.assertEqual('http://localhost/json-rpc.php', conn.get_url())
conn_params = {
'database': '/json-rpc.php'
}
conn = Database.connect(**conn_params)
self.assertEqual('http://localhost/json-rpc.php', conn.get_url())
conn_params = {
'host': 'test.com',
'port': 4000,
'database': '/json-rpc.php'
}
conn = Database.connect(**conn_params)
self.assertEqual('http://test.com:4000/json-rpc.php', conn.get_url())
conn_params = {
'schema': 'https',
'host': 'test.com',
'port': 4000,
'database': '/json-rpc.php'
}
conn = Database.connect(**conn_params)
self.assertEqual('https://test.com:4000/json-rpc.php', conn.get_url())
def test_is_auth(self):
conn_params = {
'port': 4000,
'user': '******',
'password': '******',
'auth_type': 'token'
}
conn = Database.connect(**conn_params)
self.assertEqual(True, conn.is_auth())
def setUp(self):
conn_param = {
'database': '',
'user': '',
'password': '',
'schema': 'http',
'host': 'localhost',
'port': 4000
}
self.conn = Database.connect(**conn_param)
def test_protected_res(self):
conn_params = {
'port': 4000,
'user': '******',
'password': '******',
'auth_type': 'token'
}
conn = Database.connect(**conn_params)
cur = conn.cursor()
data = {
'params': [1]
}
cur.execute('protected_res', data)
result = cur.fetchone()
self.assertEqual((1,), result)
cur.auth.set_token('fail_token')
with self.assertRaises(ProgrammingError):
cur.execute('protected_res', data)
def test_empty_conn_params(self):
conn_params = {}
conn = Database.connect(**conn_params)
self.assertEqual('http://localhost', conn.get_url())
def get_new_connection(self, conn_params):
connection = Database.connect(**conn_params)
return connection