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()