def test_get_kdb_class(self):
# v3
self.assertIsNotNone(libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB65]))
self.assertEquals(libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB65]),
libkeepass.kdb3.KDB3Reader)
# v4
self.assertIsNotNone(libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB67]))
self.assertEquals(libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB67]),
libkeepass.kdb4.KDB4Reader)
# mythical pre2.x signature
with self.assertRaisesRegexp(IOError, "Unknown sub signature."):
libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB66, 3, 0])
# unknown sub signature
with self.assertRaisesRegexp(IOError, "Unknown sub signature."):
libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB60, 3, 0])
# valid sub signature, unknown base signature
with self.assertRaisesRegexp(IOError, "Unknown base signature."):
libkeepass.get_kdb_reader([0x9AA2D900, 0xB54BFB65, 3, 0])
# unknown sub signature, unknown base signature
with self.assertRaisesRegexp(IOError, "Unknown base signature."):
libkeepass.get_kdb_reader([0x9AA2D900, 0xB54BFB60, 3, 0])
def test_get_kdb_class(self):
# v3
self.assertIsNotNone(
libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB65]))
self.assertEquals(libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB65]),
libkeepass.kdb3.KDB3Reader)
# v4
self.assertIsNotNone(
libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB67]))
self.assertEquals(libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB67]),
libkeepass.kdb4.KDB4Reader)
# mythical pre2.x signature
with self.assertRaisesRegexp(IOError, "Unknown sub signature."):
libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB66, 3, 0])
# unknown sub signature
with self.assertRaisesRegexp(IOError, "Unknown sub signature."):
libkeepass.get_kdb_reader([0x9AA2D903, 0xB54BFB60, 3, 0])
# valid sub signature, unknown base signature
with self.assertRaisesRegexp(IOError, "Unknown base signature."):
libkeepass.get_kdb_reader([0x9AA2D900, 0xB54BFB65, 3, 0])
# unknown sub signature, unknown base signature
with self.assertRaisesRegexp(IOError, "Unknown base signature."):
libkeepass.get_kdb_reader([0x9AA2D900, 0xB54BFB60, 3, 0])