Update Samba build farm to new web site layout.

[build-farm.git] / buildfarm / tests / test_util.py

diff --git a/buildfarm/tests/test_util.py b/buildfarm/tests/test_util.py
index a727ee288e80953153513d3952621596cf2d7ce7..fbd8f2474f69a36fec220fa915b4feac80ac7b60 100755 (executable)
--- a/buildfarm/tests/test_util.py
+++ b/buildfarm/tests/test_util.py
@@ -40,7 +40,7 @@ class LoadTests(testtools.TestCase):
         f = os.fdopen(fd, 'w')
         f.write("""one
 two
-three\r
+three
 
 for
 """)
@@ -49,4 +49,21 @@ for
         self.assertEquals(4, len(l))
         self.assertEquals("three", l[2])
 
+class SambaWebFileLoadTest(testtools.TestCase):
+
+    def test_simple(self):
+        fd, name = tempfile.mkstemp()
+        self.addCleanup(os.remove, name)
+        f = os.fdopen(fd, 'w')
+        f.write('href="/samba/index.html"')
+        f.close()
+        l = util.SambaWebFileLoad(os.getcwd(),name)
+        self.assertEquals('href="http://www.samba.org/samba/index.html"', l)
+        fd1, name1 = tempfile.mkstemp()
+        self.addCleanup(os.remove, name1)
+        f1 = os.fdopen(fd1, 'w')
+        f1.write('<!--#include virtual="/samba/name2" -->')
+        f1.close()
+        l1 = util.SambaWebFileLoad(os.path.dirname(os.path.realpath("name1")),name1)
+        self.assertEquals('', l1)