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))
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()