def test_lookup_of_address(self):
with TemporaryPocketBookRoot() as ctx:
address_book = AddressBook(root=ctx.root)
address_book.add('sample', SAMPLE_ADDRESS)
self.assertEqual(address_book.lookup_address('sample'),
SAMPLE_ADDRESS)
def test_rename(self):
with TemporaryPocketBookRoot() as ctx:
address_book1 = AddressBook(root=ctx.root)
address_book1.add('sample', SAMPLE_ADDRESS)
self.assertTrue(address_book1.rename('sample', 'sample2'))
self.assertIn('sample2', address_book1.keys())
self.assertNotIn('sample', address_book1.keys())
def test_throw_on_duplicate_add(self):
with TemporaryPocketBookRoot() as ctx:
address_book = AddressBook(root=ctx.root)
address_book.add('sample', SAMPLE_ADDRESS)
with self.assertRaises(RuntimeError):
address_book.add('sample', SAMPLE_ADDRESS)
def test_check_keys_exists(self):
with TemporaryPocketBookRoot() as ctx:
address_book = AddressBook(root=ctx.root)
address_book.add('sample', SAMPLE_ADDRESS)
self.assertIn('sample', address_book.keys())
self.assertNotIn('foo', address_book.keys())
def test_simple_write(self):
with TemporaryPocketBookRoot() as ctx:
# add the sample address
address_book = AddressBook(root=ctx.root)
address_book.add('sample', SAMPLE_ADDRESS)
self.assertAddressIsPresentOnDisk('sample', SAMPLE_ADDRESS, ctx)
def test_remove_is_stored_on_disk(self):
with TemporaryPocketBookRoot() as ctx:
address_book = AddressBook(root=ctx.root)
address_book.add('sample', SAMPLE_ADDRESS)
self.assertTrue(address_book.remove('sample'))
self.assertAddressIsNotPresentOnDisk('sample', ctx)
def test_load_of_addresses(self):
with TemporaryPocketBookRoot() as ctx:
address_book1 = AddressBook(root=ctx.root)
address_book1.add('sample', SAMPLE_ADDRESS)
address_book2 = AddressBook(root=ctx.root)
self.assertEqual(set(address_book1.items()),
set(address_book2.items()))
def test_get_items(self):
with TemporaryPocketBookRoot() as ctx:
address_book = AddressBook(root=ctx.root)
address_book.add('sample', SAMPLE_ADDRESS)
address_book_items = list(address_book.items())
self.assertEqual(len(address_book_items), 1)
self.assertEqual(address_book_items[0], ('sample', SAMPLE_ADDRESS))
def test_rename_occurs_on_disk(self):
with TemporaryPocketBookRoot() as ctx:
address_book1 = AddressBook(root=ctx.root)
address_book1.add('sample', SAMPLE_ADDRESS)
self.assertTrue(address_book1.rename('sample', 'sample2'))
self.assertAddressIsNotPresentOnDisk('sample', ctx)
self.assertAddressIsPresentOnDisk('sample2', SAMPLE_ADDRESS, ctx)
def run_add(args):
from pocketbook.address_book import AddressBook
address_book = AddressBook()
address_book.add(args.name, args.address)