class TestBBContact:
def __init__(self, config_fn, state_fn, bbfn):
logging.debug('Getting started... Reading Config File...')
self.config = Config(config_fn, state_fn)
self.bb = BBPIMDB(self.config, bbfn)
ms = self.bb.get_def_msgstore()
self.deff = ms.get_folder(ms.get_def_folder_name())
def print_contacts(self, cnt=0, name=None):
self.deff.print_contacts(cnt=cnt, name=name)
def write_to_file(self):
self.deff.save()
class TestBBContact:
def __init__ (self, config_fn, state_fn, bbfn):
logging.debug('Getting started... Reading Config File...')
self.config = Config(config_fn, state_fn)
self.bb = BBPIMDB(self.config, bbfn)
ms = self.bb.get_def_msgstore()
self.deff = ms.get_folder(ms.get_def_folder_name())
def print_contacts (self, cnt=0, name=None):
self.deff.print_contacts(cnt=cnt, name=name)
def write_to_file (self):
self.deff.save()
class TestBBDB(unittest.TestCase):
def setUp (self):
self.config = config
self.bbdbfn = bbfn
self.bb = BBPIMDB(self.config, bbfn)
def test_parse (self):
self.bb = BBPIMDB(self.config, bbfn)
def test_ver (self):
ver_check = self.get_ver_from_filename()
# print "Ver_check: ", ver_check
if ver_check:
assert(ver_check == self.bb.get_def_msgstore().get_file_format())
def get_ver_from_filename (self):
v = re.search('\.v(\d+)\.', self.bbdbfn)
return v.group(1) if v else None
class TestBBDB(unittest.TestCase):
def setUp(self):
self.config = config
self.bbdbfn = bbfn
self.bb = BBPIMDB(self.config, bbfn)
def test_parse(self):
self.bb = BBPIMDB(self.config, bbfn)
def test_ver(self):
ver_check = self.get_ver_from_filename()
# print "Ver_check: ", ver_check
if ver_check:
assert (ver_check == self.bb.get_def_msgstore().get_file_format())
def get_ver_from_filename(self):
v = re.search('\.v(\d+)\.', self.bbdbfn)
return v.group(1) if v else None
class TestBBDBWrite(unittest.TestCase):
def setUp (self):
self.config = config
self.bbdbfn = os.path.join(CUR_DIR, "temp.bbdb")
self.reparse(self.bbdbfn)
def tearDown (self):
os.remove(self.bbdbfn)
pass
def reparse (self, bbfn=None):
if not bbfn:
bbfn = self.bbdbfn
self.bb = BBPIMDB(self.config, bbfn)
ms = self.bb.get_def_msgstore()
self.deff = ms.get_folder(ms.get_def_folder_name())
def test_basic (self):
con = BBContact(self.deff)
con.set_firstname('Test Basic Sri Venkata Sri Rama Subramanya Anjeneya Annapurna Sharma')
con.set_prefix('Mr.')
con.set_nickname('Karra')
con.set_gender('Male')
con.add_phone_mob(('Mobile', '+91 90084 88997'))
con.add_notes('And so it goes...')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
cons = self.deff.find_contacts_by_name(name='Venkata')
assert(len(cons) == 1)
con = cons[0]
assert(con.get_prefix() == "Mr.")
assert(con.get_gender() == "Male")
def test_nonexistent_contact (self):
cons = self.deff.find_contacts_by_name(name='Goof Ball')
assert(len(cons) == 0)
def test_unicode (self):
con = BBContact(self.deff)
con.set_firstname(u'Héctor')
con.set_lastname(r'Tarrido-Picart')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_1 (self):
con = BBContact(self.deff)
con.set_firstname(r'Test Quotes 1')
con.set_lastname(r'"Karra"')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_2 (self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes 2')
con.set_lastname("Ka\\\nrra")
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_3 (self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes 3')
con.set_lastname("\"Karra")
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_4 (self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes 4')
con.set_lastname("\\Karra")
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_terminating_backslash (self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes TB')
con.set_lastname('Karra\\')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
class TestBBDBWrite(unittest.TestCase):
def setUp(self):
self.config = config
self.bbdbfn = os.path.join(CUR_DIR, "temp.bbdb")
self.reparse(self.bbdbfn)
def tearDown(self):
os.remove(self.bbdbfn)
pass
def reparse(self, bbfn=None):
if not bbfn:
bbfn = self.bbdbfn
self.bb = BBPIMDB(self.config, bbfn)
ms = self.bb.get_def_msgstore()
self.deff = ms.get_folder(ms.get_def_folder_name())
def test_basic(self):
con = BBContact(self.deff)
con.set_firstname(
'Test Basic Sri Venkata Sri Rama Subramanya Anjeneya Annapurna Sharma'
)
con.set_prefix('Mr.')
con.set_nickname('Karra')
con.set_gender('Male')
con.add_phone_mob(('Mobile', '+91 90084 88997'))
con.add_notes('And so it goes...')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
cons = self.deff.find_contacts_by_name(name='Venkata')
assert (len(cons) == 1)
con = cons[0]
assert (con.get_prefix() == "Mr.")
assert (con.get_gender() == "Male")
def test_nonexistent_contact(self):
cons = self.deff.find_contacts_by_name(name='Goof Ball')
assert (len(cons) == 0)
def test_unicode(self):
con = BBContact(self.deff)
con.set_firstname(u'Héctor')
con.set_lastname(r'Tarrido-Picart')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_1(self):
con = BBContact(self.deff)
con.set_firstname(r'Test Quotes 1')
con.set_lastname(r'"Karra"')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_2(self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes 2')
con.set_lastname("Ka\\\nrra")
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_3(self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes 3')
con.set_lastname("\"Karra")
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_4(self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes 4')
con.set_lastname("\\Karra")
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
def test_quotes_terminating_backslash(self):
con = BBContact(self.deff)
con.set_firstname('Test Quotes TB')
con.set_lastname('Karra\\')
self.deff.add_contact(con)
self.deff.save()
self.reparse(self.bbdbfn)
