def setUpClass(self):
u = "mysql://*****:*****@localhost:3306/pwmantest"
u = urlparse(u)
# password required, for all hosts
# u = "mysql://<user>:<pass>@localhost/pwmantest"
self.db = MySQLDatabase(u)
self.db._open()
def setUpClass(self):
u = DBURI
u = urlparse(u)
# password required, for all hosts
# u = "mysql://<user>:<pass>@localhost/pwmantest"
self.db = MySQLDatabase(u)
self.db._open()
class TestMySQLDatabase(unittest.TestCase):
@classmethod
def setUpClass(self):
u = DBURI
u = urlparse(u)
# password required, for all hosts
# u = "mysql://<user>:<pass>@localhost/pwmantest"
self.db = MySQLDatabase(u)
self.db._open()
@classmethod
def tearDownClass(self):
for table in ['LOOKUP', 'TAG', 'NODE', 'DBVERSION', 'CRYPTO']:
try:
self.db._cur.execute("DROP TABLE {}".format(table))
except Exception:
pass
self.db._con.commit()
def test_1_con(self):
self.assertIsInstance(self.db._con, pymysql.connections.Connection)
def test_2_create_tables(self):
self.db._create_tables()
# invoking this method a second time should not raise an exception
self.db._create_tables()
def test_3_load_key(self):
self.db.savekey('SECRET$6$KEY')
secretkey = self.db.loadkey()
self.assertEqual(secretkey, 'SECRET$6$KEY')
def test_4_save_crypto(self):
self.db.save_crypto_info("TOP", "SECRET")
secretkey = self.db.loadkey()
self.assertEqual(secretkey, 'TOP$6$SECRET')
row = self.db.fetch_crypto_info()
self.assertEqual(row, ('TOP', 'SECRET'))
def test_5_add_node(self):
innode = ["TBONE", "S3K43T", "example.org", "some note",
["bartag", "footag"]]
self.db.add_node(innode)
outnode = self.db.getnodes([1])[0]
self.assertEqual(innode[:-1] + [t for t in innode[-1]], outnode[1:])
def test_6_list_nodes(self):
ret = self.db.listnodes()
self.assertEqual(ret, [1])
ret = self.db.listnodes("footag")
self.assertEqual(ret, [1])
def test_6a_list_tags(self):
ret = self.db.listtags()
self.assertListEqual(ret, ['bartag', 'footag'])
def test_6b_get_nodes(self):
ret = self.db.getnodes([1])
retb = self.db.getnodes([])
self.assertListEqual(ret, retb)
def test_7_get_or_create_tag(self):
s = self.db._get_or_create_tag("SECRET")
s1 = self.db._get_or_create_tag("SECRET")
self.assertEqual(s, s1)
def test_7a_clean_orphans(self):
self.db._clean_orphans()
rv = self.db._get_tag("SECRET")
self.assertIsNone(rv)
def test_8_remove_node(self):
self.db.removenodes([1])
n = self.db.listnodes()
self.assertEqual(len(n), 0)
def test_9_check_db_version(self):
dburi = DBURI
v = self.db.check_db_version(urlparse(dburi))
self.assertEqual(v, '0.6')
self.db._cur.execute("DROP TABLE DBVERSION")
self.db._con.commit()
v = self.db.check_db_version(urlparse(dburi))
self.assertEqual(v, '0.6')
self.db._cur.execute("CREATE TABLE DBVERSION("
"VERSION TEXT NOT NULL) ")
self.db._con.commit()
class TestMySQLDatabase(unittest.TestCase):
@classmethod
def setUpClass(self):
u = "mysql://*****:*****@localhost:3306/pwmantest"
u = urlparse(u)
# password required, for all hosts
# u = "mysql://<user>:<pass>@localhost/pwmantest"
self.db = MySQLDatabase(u)
self.db._open()
@classmethod
def tearDownClass(self):
for table in ['LOOKUP', 'TAG', 'NODE', 'DBVERSION', 'CRYPTO']:
try:
self.db._cur.execute("DROP TABLE {}".format(table))
except Exception:
pass
self.db._con.commit()
def test_1_con(self):
self.assertIsInstance(self.db._con, pymysql.connections.Connection)
def test_2_create_tables(self):
self.db._create_tables()
# invoking this method a second time should not raise an exception
self.db._create_tables()
def test_3_load_key(self):
self.db.savekey('SECRET$6$KEY')
secretkey = self.db.loadkey()
self.assertEqual(secretkey, 'SECRET$6$KEY')
def test_4_save_crypto(self):
self.db.save_crypto_info("TOP", "SECRET")
secretkey = self.db.loadkey()
self.assertEqual(secretkey, 'TOP$6$SECRET')
row = self.db.fetch_crypto_info()
self.assertEqual(row, ('TOP', 'SECRET'))
def test_5_add_node(self):
innode = [
"TBONE", "S3K43T", "example.org", "some note",
["bartag", "footag"]
]
self.db.add_node(innode)
outnode = self.db.getnodes([1])[0]
self.assertEqual(innode[:-1] + [t for t in innode[-1]], outnode[1:])
def test_6_list_nodes(self):
ret = self.db.listnodes()
self.assertEqual(ret, [1])
ret = self.db.listnodes("footag")
self.assertEqual(ret, [1])
def test_6a_list_tags(self):
ret = self.db.listtags()
self.assertListEqual(ret, ['bartag', 'footag'])
def test_6b_get_nodes(self):
ret = self.db.getnodes([1])
retb = self.db.getnodes([])
self.assertListEqual(ret, retb)
def test_7_get_or_create_tag(self):
s = self.db._get_or_create_tag("SECRET")
s1 = self.db._get_or_create_tag("SECRET")
self.assertEqual(s, s1)
def test_7a_clean_orphans(self):
self.db._clean_orphans()
rv = self.db._get_tag("SECRET")
self.assertIsNone(rv)
def test_8_remove_node(self):
self.db.removenodes([1])
n = self.db.listnodes()
self.assertEqual(len(n), 0)
def test_9_check_db_version(self):
dburi = "mysql://*****:*****@localhost:3306/pwmantest"
v = self.db.check_db_version(urlparse(dburi))
self.assertEqual(v, '0.6')
self.db._cur.execute("DROP TABLE DBVERSION")
self.db._con.commit()
v = self.db.check_db_version(urlparse(dburi))
self.assertEqual(v, '0.6')
self.db._cur.execute("CREATE TABLE DBVERSION("
"VERSION TEXT NOT NULL) ")
self.db._con.commit()
