#============================================================================ # This file is part of Pwman3. # # Pwman3 is free software; you can redistribute iut and/or modify # it under the terms of the GNU General Public License, version 2 # as published by the Free Software Foundation; # # Pwman3 is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with Pwman3; if not, write to the Free Software # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA #============================================================================ # Copyright (C) 2012 Oz Nahum Tiram #============================================================================ import os import unittest from pwman.data.drivers.sqlite import SQLite class TestSQLite(unittest.TestCase): def setUp(self): self.db = SQLite('test.db') self.db._open() def test_1_create_tables(self): self.db._create_tables() self.db._con.commit() # the method _open calls _create_tables self.db.save_crypto_info("foo", "bar") self.db._create_tables() def test_1a_create_tables(self): self.db._create_tables() def test_2_crypto_info(self): self.db._create_tables() self.db.save_crypto_info("foo", "bar") f = self.db.fetch_crypto_info() self.assertListEqual([u'foo', u'bar'], list(f)) def test_3_add_node(self): node = ("alice", "secret", "wonderland.com", "a really great place") self.db.add_node(node) rv = self.db._cur.execute("select * from node") self.assertIn('alice', rv.fetchone()) def tearDown(self): self.db.close() if __name__ == '__main__': try: unittest.main(verbosity=2) except SystemExit: os.remove('test.db')