def testFixTrailers(self): ParseUtility.removeHeader(self._tables) ParseUtility.fixTrailers(self._setUp, self._tables) self._assertEquals( "<html><title>setup</title><body>" "s1<table><tr><td>SetUp</td></tr></table>s2" "s3<table><tr><td>SetUp</td></tr></table>", self._setUp) self._assertEquals( "s4<br>t1<table><tr><td>Test</td></tr></table>t2" "t3<table><tr><td>Test</td></tr></table>t4" "</body></html>", self._tables)
def testFixTrailers(self): ParseUtility.removeHeader(self._tables) ParseUtility.fixTrailers(self._setUp, self._tables) self._assertEquals("<html><title>setup</title><body>" "s1<table><tr><td>SetUp</td></tr></table>s2" "s3<table><tr><td>SetUp</td></tr></table>", self._setUp) self._assertEquals( "s4<br>t1<table><tr><td>Test</td></tr></table>t2" "t3<table><tr><td>Test</td></tr></table>t4" "</body></html>", self._tables)
def testFixTrailersNone(self): setUpHtml = ("<html><title>setup</title><body>" "<table><tr><td>SetUp</td></tr></table>" "</body></html>") setUp2 = Parse(setUpHtml) ParseUtility.fixTrailers(setUp2, self._tables) self._assertEquals( "<html><title>setup</title><body>" "<table><tr><td>SetUp</td></tr></table>", setUp2) self._assertEquals( "<html><title>table</title><body>" "t1<table><tr><td>Test</td></tr></table>t2" "t3<table><tr><td>Test</td></tr></table>t4" "</body></html>", self._tables)
def testFixTrailersNone(self): setUpHtml = ("<html><title>setup</title><body>" "<table><tr><td>SetUp</td></tr></table>" "</body></html>") setUp2 = Parse(setUpHtml) ParseUtility.fixTrailers(setUp2, self._tables) self._assertEquals("<html><title>setup</title><body>" "<table><tr><td>SetUp</td></tr></table>", setUp2) self._assertEquals("<html><title>table</title><body>" "t1<table><tr><td>Test</td></tr></table>t2" "t3<table><tr><td>Test</td></tr></table>t4" "</body></html>", self._tables)