class ProfileTest(unittest.TestCase):
"""test interactactions between profiles"""
def setUp(self):
"""override one in unittest.TestCase"""
# bruce
doc = FileDocument(PROFILE_BRUCE, PROFILE_DIRECTORY)
self.assert_(doc.load())
self.bruce_doc = CacheDocument(PROFILE_BRUCE, PROFILE_DIRECTORY)
self.bruce_doc.import_document(doc)
# demi
self.demi_doc = FileDocument(PROFILE_DEMI, PROFILE_DIRECTORY)
self.assert_(self.demi_doc.load())
def test_bruce(self):
"""import bruce data"""
self.assertEquals("Bruce", self.bruce_doc.get_firstname())
self.assertEquals("Willis", self.bruce_doc.get_lastname())
self.assertEquals("*****@*****.**", self.bruce_doc.get_email())
self.assertEquals({'City': u'', 'Country': u'',
'Favourite Book': u'', 'music': u'jazz',
'Favourite Movie': u'', 'Sport': u'', 'Studies': u'',
'film': u'Die Hard'},
self.bruce_doc.get_custom_attributes())
def test_peer_status(self):
self.assertEquals(self.demi_doc.has_peer(self.bruce_doc.pseudo), False)
self.demi_doc.fill_data((self.bruce_doc.pseudo, self.bruce_doc))
self.assertEquals(self.demi_doc.has_peer(self.bruce_doc.pseudo), True)
self.demi_doc.make_friend(self.bruce_doc.pseudo)
self.assertEquals(self.demi_doc.has_peer(self.bruce_doc.pseudo), True)
self.demi_doc.unmark_peer(self.bruce_doc.pseudo)
self.assertEquals(self.demi_doc.has_peer(self.bruce_doc.pseudo), False)
class HtmlTest(unittest.TestCase):
"""test that all fields are correctly validated"""
def setUp(self):
"""override one in unittest.TestCase"""
self.document = CacheDocument(PROFILE_TEST, PROFILE_DIRECTORY)
self.desc = PeerDescriptor(PROFILE_TEST, document=self.document)
self.view = HtmlView(self.desc)
def assert_template(self):
"""diff result from view with expeceted template"""
self.view.import_desc(self.desc)
result = difflib.ndiff([line.strip() for line in self.print_template().splitlines()],
[line.strip() for line in self.view.get_view(True).splitlines()],
charjunk=lambda x: x in [' ', '\t'])
result = list(result)
for index, line in enumerate(result):
if not line.startswith(" "):
print self.view.get_view()
print "*************"
print '\n'.join(result[index-2:index+10])
self.assert_(line.startswith(" "))
def print_template(self):
"""fill template with values"""
return TEMPLATE % (self.document.get_photo(),
self.document.pseudo,
self.document.get_title(),
self.document.get_firstname(),
self.document.get_lastname(),
self.document.get_email(),
self.print_custom(),
self.print_files(),
self.print_peers())
def print_custom(self):
html = ["<div><b>%s</b>: <span>%s</span></div>"% (key, value)
for key, value in self.document.get_custom_attributes().iteritems()]
return ''.join(html)
def print_files(self):
html = []
repos = self.document.get_repositories()
repos.sort()
for repo in repos:
html.append("""
<dt>%s</dt>
<dd>%d shared files</dd>
"""% (os.path.basename(repo), len(self.document.get_shared(repo))))
# concatenate all description of directories & return result
return ''.join(html)
def print_peers(self):
html = []
peers = self.document.get_peers()
keys = peers.keys()
keys.sort()
for key in keys:
peers_desc = peers[key]
doc = peers_desc.document
html.append("""<tr>
<td>%s</td>
<td>%s</td>
<td>%s</td>
</tr>"""% (peers_desc.pseudo,
peers_desc.state,
doc and doc.get_email() or "--"))
return ''.join(html)
# PERSONAL TAB
def test_personal(self):
"""personal info"""
self.document.set_title(u"Mr")
self.document.set_firstname(u"Bruce")
self.document.set_lastname(u"Willis")
self.document.set_photo(os.sep.join([images_dir(), u"profile_male.gif"]))
self.document.set_email(u"*****@*****.**")
self.assert_template()
# CUSTOM TAB
def test_custom(self):
self.document.add_custom_attributes([u"zic", u"jazz"])
self.document.add_custom_attributes([u"cinema", u"Die Hard"])
self.assert_template()
# FILE TAB
def test_files(self):
self.document.add_file(abspath("."))
self.document.expand_dir(abspath("data"))
self.document.share_files((abspath("data"), ["routage"], True))
self.document.share_file((abspath("data/emptydir"), True))
self.assert_template()
# OTHERS TAB
def test_ordered_peer(self):
self.document.set_peer((u"emb", PeerDescriptor(PSEUDO)))
self.document.set_peer((u"albert", PeerDescriptor(PSEUDO)))
self.document.set_peer((u"star", PeerDescriptor(PSEUDO)))
self.document.make_friend(u"emb")
self.document.make_friend(u"star")
self.document.make_friend(u"albert")
self.assert_template()
def test_adding_peer(self):
self.document.set_peer((u"nico", PeerDescriptor(PSEUDO)))
self.assert_template()
def test_filling_data(self):
self.document.fill_data((u"emb", FileDocument(PROFILE_TEST, PROFILE_DIRECTORY)))
self.assert_template()
def test_peers_status(self):
self.document.set_peer((u"nico", PeerDescriptor(PSEUDO)))
self.document.blacklist_peer(u"nico")
self.assert_template()