# copyright 2003-2015 LOGILAB S.A. (Paris, FRANCE), all rights reserved.# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr## This file is part of CubicWeb.## CubicWeb is free software: you can redistribute it and/or modify it under the# terms of the GNU Lesser General Public License as published by the Free# Software Foundation, either version 2.1 of the License, or (at your option)# any later version.## CubicWeb is distributed in the hope that it will be useful, but WITHOUT# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more# details.## You should have received a copy of the GNU Lesser General Public License along# with CubicWeb. If not, see <http://www.gnu.org/licenses/>.fromlogilab.common.testlibimportunittest_mainfromlogilab.mtconverterimporthtml_unescapefromcubicweb.devtools.htmlparserimportXMLValidatorfromcubicweb.devtools.testlibimportCubicWebTCclassTreeViewTC(CubicWebTC):deftest_treeview(self):withself.admin_access.repo_cnx()ascnx:ce=cnx.create_entityroot=ce('TreeNode',name=u'root')node=ce('TreeNode',name=u'node1',parent=root)ce('TreeNode',name=u'leaf1a',parent=node)ce('TreeNode',name=u'leaf1b',parent=node)node=ce('TreeNode',name=u'node2',parent=root)ce('TreeNode',name=u'leaf2a',parent=node)ce('TreeNode',name=u'leaf2b',parent=node)root_eid=root.eidcnx.commit()withself.admin_access.web_request()asreq:root=req.entity_from_eid(root_eid)valid=self.content_type_validators.get('text/html',XMLValidator)()page=valid.parse_string(root.view('tree',klass='oh-my-class'))uls=page.find_tag('ul',gettext=False)for_,attribinuls:self.assertEqual(attrib['class'],'oh-my-class')if__name__=='__main__':unittest_main()