def _add_tab(self, peer_desc):
"""add tab with preview of given peer"""
tab_sizer = wx.BoxSizer(wx.VERTICAL)
preview_tab = wx.Panel(self.preview_notebook, -1)
match_preview = MyHtmlWindow(preview_tab, -1)
tab_sizer.Add(match_preview, 1, wx.EXPAND, 0)
view = HtmlView(peer_desc)
self.tabs[peer_desc.node_id] = preview_tab
preview_tab.preview = match_preview
preview_tab.pseudo = peer_desc.document.get_pseudo()
preview_tab.SetAutoLayout(True)
preview_tab.SetSizer(tab_sizer)
preview_tab.preview.SetPage(view.get_view())
tab_sizer.Fit(preview_tab)
tab_sizer.SetSizeHints(preview_tab)
self._update_tab(peer_desc)
def export_profile(self, path):
"""write profile in html format"""
html_file = open(path, "w")
html_view = HtmlView(self._desc)
html_file.write(html_view.get_view(True).encode(ENCODING))
def setUp(self):
"""override one in unittest.TestCase"""
self.desc = PeerDescriptor(PROFILE_TEST)
self.document = self.desc.document
self.view = HtmlView(self.desc)
class HtmlTest(unittest.TestCase):
"""test that all fields are correctly validated"""
def setUp(self):
"""override one in unittest.TestCase"""
self.desc = PeerDescriptor(PROFILE_TEST)
self.document = self.desc.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.get_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_files())))
# 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>"""% (doc.get_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_repository(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", PeerDescriptor(PROFILE_TEST).document)
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()