From ab8fdaa604b8caecfb9192ea7c6c77417bfbdd47 Mon Sep 17 00:00:00 2001
From: Aleksander Machniak <alec@alec.pl>
Date: Tue, 31 Jul 2012 04:27:37 -0400
Subject: [PATCH] Small code improvement

---
 tests/mailfunc.php |   11 +++++++----
 1 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/tests/mailfunc.php b/tests/mailfunc.php
index d8111c7..493ce94 100644
--- a/tests/mailfunc.php
+++ b/tests/mailfunc.php
@@ -16,8 +16,7 @@
     $GLOBALS['RCMAIL'] = $RCMAIL = rcmail::get_instance();
     $GLOBALS['OUTPUT'] = $OUTPUT = $RCMAIL->load_gui();
     $RCMAIL->action = 'autocomplete';
-    $RCMAIL->imap_init(false);
-    $IMAP = $RCMAIL->imap;
+    $RCMAIL->storage_init(false);
     
     require_once INSTALL_PATH . 'program/steps/mail/func.inc';
     
@@ -49,7 +48,7 @@
     $html = rcmail_html4inline(rcmail_print_body($part, array('safe' => false)), 'foo');
 
     $this->assertPattern('/src="'.$part->replaces['ex1.jpg'].'"/', $html, "Replace reference to inline image");
-    $this->assertPattern('#background="./program/blocked.gif"#', $html, "Replace external background image");
+    $this->assertPattern('#background="./program/resources/blocked.gif"#', $html, "Replace external background image");
     $this->assertNoPattern('/ex3.jpg/', $html, "No references to external images");
     $this->assertNoPattern('/<meta [^>]+>/', $html, "No meta tags allowed");
     //$this->assertNoPattern('/<style [^>]+>/', $html, "No style tags allowed");
@@ -92,7 +91,7 @@
   function test_html_xss2()
   {
     $part = $this->get_html_part('src/BID-26800.txt');
-    $washed = rcmail_print_body($part, array('safe' => true));
+    $washed = rcmail_html4inline(rcmail_print_body($part, array('safe' => true)), 'dabody', '', $attr, true);
 
     $this->assertNoPattern('/alert|expression|javascript|xss/', $washed, "Remove evil style blocks");
     $this->assertNoPattern('/font-style:italic/', $washed, "Allow valid styles");
@@ -166,5 +165,9 @@
     $this->assertPattern('|src="http://alec\.pl/dir/img1\.gif"|', $html, "URI base resolving [1]");
     $this->assertPattern('|src="http://alec\.pl/dir/img2\.gif"|', $html, "URI base resolving [2]");
     $this->assertPattern('|src="http://alec\.pl/img3\.gif"|', $html, "URI base resolving [3]");
+
+    // base resolving exceptions
+    $this->assertPattern('|src="cid:theCID"|', $html, "URI base resolving exception [1]");
+    $this->assertPattern('|src="http://other\.domain\.tld/img3\.gif"|', $html, "URI base resolving exception [2]");
   }
 }

--
Gitblit v1.9.1