1 files changed, 145 insertions, 0 deletions
diff --git a/mediagoblin/tests/test_util.py b/mediagoblin/tests/test_util.py
new file mode 100644
index 00000000..bc14f528
--- /dev/null
+++ b/mediagoblin/tests/test_util.py
@@ -0,0 +1,145 @@
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program 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 Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+import email
+
+from mediagoblin.tools import common, url, translate, mail, text, testing
+
+testing._activate_testing()
+
+
+def _import_component_testing_method(silly_string):
+    # Just for the sake of testing that our component importer works.
+    return u"'%s' is the silliest string I've ever seen" % silly_string
+
+
+def test_import_component():
+    imported_func = common.import_component(
+        'mediagoblin.tests.test_util:_import_component_testing_method')
+    result = imported_func('hooobaladoobala')
+    expected = u"'hooobaladoobala' is the silliest string I've ever seen"
+    assert result == expected
+
+
+def test_send_email():
+    mail._clear_test_inboxes()
+
+    # send the email
+    mail.send_email(
+        "sender@mediagoblin.example.org",
+        ["amanda@example.org", "akila@example.org"],
+        "Testing is so much fun!",
+        """HAYYY GUYS!
+
+I hope you like unit tests JUST AS MUCH AS I DO!""")
+
+    # check the main inbox
+    assert len(mail.EMAIL_TEST_INBOX) == 1
+    message = mail.EMAIL_TEST_INBOX.pop()
+    assert message['From'] == "sender@mediagoblin.example.org"
+    assert message['To'] == "amanda@example.org, akila@example.org"
+    assert message['Subject'] == "Testing is so much fun!"
+    assert message.get_payload(decode=True) == """HAYYY GUYS!
+
+I hope you like unit tests JUST AS MUCH AS I DO!"""
+
+    # Check everything that the FakeMhost.sendmail() method got is correct
+    assert len(mail.EMAIL_TEST_MBOX_INBOX) == 1
+    mbox_dict = mail.EMAIL_TEST_MBOX_INBOX.pop()
+    assert mbox_dict['from'] == "sender@mediagoblin.example.org"
+    assert mbox_dict['to'] == ["amanda@example.org", "akila@example.org"]
+    mbox_message = email.message_from_string(mbox_dict['message'])
+    assert mbox_message['From'] == "sender@mediagoblin.example.org"
+    assert mbox_message['To'] == "amanda@example.org, akila@example.org"
+    assert mbox_message['Subject'] == "Testing is so much fun!"
+    assert mbox_message.get_payload(decode=True) == """HAYYY GUYS!
+
+I hope you like unit tests JUST AS MUCH AS I DO!"""
+
+def test_slugify():
+    assert url.slugify(u'a walk in the park') == u'a-walk-in-the-park'
+    assert url.slugify(u'A Walk in the Park') == u'a-walk-in-the-park'
+    assert url.slugify(u'a  walk in the park') == u'a-walk-in-the-park'
+    assert url.slugify(u'a walk in-the-park') == u'a-walk-in-the-park'
+    assert url.slugify(u'a w@lk in the park?') == u'a-w-lk-in-the-park'
+    assert url.slugify(u'a walk in the par\u0107') == u'a-walk-in-the-parc'
+    assert url.slugify(u'\u00E0\u0042\u00E7\u010F\u00EB\u0066') == u'abcdef'
+
+def test_locale_to_lower_upper():
+    """
+    Test cc.i18n.util.locale_to_lower_upper()
+    """
+    assert translate.locale_to_lower_upper('en') == 'en'
+    assert translate.locale_to_lower_upper('en_US') == 'en_US'
+    assert translate.locale_to_lower_upper('en-us') == 'en_US'
+
+    # crazy renditions.  Useful?
+    assert translate.locale_to_lower_upper('en-US') == 'en_US'
+    assert translate.locale_to_lower_upper('en_us') == 'en_US'
+
+
+def test_locale_to_lower_lower():
+    """
+    Test cc.i18n.util.locale_to_lower_lower()
+    """
+    assert translate.locale_to_lower_lower('en') == 'en'
+    assert translate.locale_to_lower_lower('en_US') == 'en-us'
+    assert translate.locale_to_lower_lower('en-us') == 'en-us'
+
+    # crazy renditions.  Useful?
+    assert translate.locale_to_lower_lower('en-US') == 'en-us'
+    assert translate.locale_to_lower_lower('en_us') == 'en-us'
+
+
+def test_gettext_lazy_proxy():
+    from mediagoblin.tools.translate import lazy_pass_to_ugettext as _
+    from mediagoblin.tools.translate import pass_to_ugettext, set_thread_locale
+    proxy = _(u"Password")
+    orig = u"Password"
+
+    set_thread_locale("es")
+    p1 = unicode(proxy)
+    p1_should = pass_to_ugettext(orig)
+    assert p1_should != orig, "Test useless, string not translated"
+    assert p1 == p1_should
+
+    set_thread_locale("sv")
+    p2 = unicode(proxy)
+    p2_should = pass_to_ugettext(orig)
+    assert p2_should != orig, "Test broken, string not translated"
+    assert p2 == p2_should
+
+    assert p1_should != p2_should, "Test broken, same translated string"
+    assert p1 != p2
+
+
+def test_html_cleaner():
+    # Remove images
+    result = text.clean_html(
+        '<p>Hi everybody! '
+        '<img src="http://example.org/huge-purple-barney.png" /></p>\n'
+        '<p>:)</p>')
+    assert result == (
+        '<div>'
+        '<p>Hi everybody! </p>\n'
+        '<p>:)</p>'
+        '</div>')
+
+    # Remove evil javascript
+    result = text.clean_html(
+        '<p><a href="javascript:nasty_surprise">innocent link!</a></p>')
+    assert result == (
+        '<p><a href="">innocent link!</a></p>')