1 

     2 from logilab.common.testlib import TestCase, unittest_main

     3 

     4 from cubicweb.web.views.embedding import prefix_links

     5 

     6 class UILIBTC(TestCase):

     7 

     8 

     9     def test_prefix_links(self):

    10         """suppose we are embedding http://embedded.com/page1.html"""

    11         orig = ['<a href="http://www.perdu.com">perdu ?</a>',

    12         '<a href="http://embedded.com/page1.html">perdu ?</a>',

    13         '<a href="/page2.html">perdu ?</a>',

    14         '<a href="page3.html">perdu ?</a>',

    15         '<img src="http://www.perdu.com/img.png"/>',

    16         '<img src="/img.png"/>',

    17         '<img src="img.png"/>',

    18         ]

    19         expected = ['<a href="PREFIXhttp%3A%2F%2Fwww.perdu.com">perdu ?</a>',

    20         '<a href="PREFIXhttp%3A%2F%2Fembedded.com%2Fpage1.html">perdu ?</a>',

    21         '<a href="PREFIXhttp%3A%2F%2Fembedded.com%2Fpage2.html">perdu ?</a>',

    22         '<a href="PREFIXhttp%3A%2F%2Fembedded.com%2Fpage3.html">perdu ?</a>',

    23         '<img src="http://www.perdu.com/img.png"/>',

    24         '<img src="http://embedded.com/img.png"/>',

    25         '<img src="http://embedded.com/img.png"/>',

    26         ]

    27         for orig_a, expected_a in zip(orig, expected):

    28             got = prefix_links(orig_a, 'PREFIX', 'http://embedded.com/page1.html')

    29             self.assertEquals(got, expected_a)

    30 

    31 if __name__ == '__main__':

    32     unittest_main()

    33