browser2: Handle server issues of postbin.org

And move tests outside.

weboob/tools/browser2/__init__.py

@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2012 Laurent Bachelier
+#
+# This file is part of weboob.
+#
+# weboob 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.
+#
+# weboob 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 weboob. If not, see <http://www.gnu.org/licenses/>.

weboob/tools/browser2/browser.py

@@ -17,6 +17,8 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with weboob. If not, see <http://www.gnu.org/licenses/>.
 
+from __future__ import absolute_import
+
 import requests
 from requests.status_codes import codes
 
@@ -203,47 +205,3 @@ class Browser(object):
             kwargs.setdefault('headers', {}).setdefault('Content-Length', '0')
         kwargs.setdefault('timeout', self.TIMEOUT)
         return self.session.request(*args, **kwargs)
-
-
-def test_base():
-    b = Browser()
-    r = b.location('http://httpbin.org/headers')
-    assert isinstance(r.text, unicode)
-    assert 'Firefox' in r.text
-    assert 'python' not in r.text
-    assert 'identity' not in r.text
-    assert b.url == 'http://httpbin.org/headers'
-
-
-def test_redirects():
-    b = Browser()
-    b.location('http://httpbin.org/redirect/1')
-    assert b.url == 'http://httpbin.org/get'
-
-
-def test_brokenpost():
-    """
-    Tests _fix_redirect()
-    """
-    b = Browser()
-    # postbin is picky with empty posts. that's good!
-    r = b.location('http://www.postbin.org/', {})
-    # ensures empty data (but not None) does a POST
-    assert r.request.method == 'POST'
-    # ensure we were redirected after submitting a post
-    assert len(r.url) >= len('http://www.postbin.org/')
-    # send a POST with data
-    b.location(r.url, {'hello': 'world'})
-    r = b.location(r.url + '/feed')
-    assert 'hello' in r.text
-    assert 'world' in r.text
-
-
-def test_weboob():
-    class BooBrowser(Browser):
-        PROFILE = Weboob('0.0')
-
-    b = BooBrowser()
-    r = b.location('http://httpbin.org/headers')
-    assert 'weboob/0.0' in r.text
-    assert 'identity' in r.text

weboob/tools/browser2/test.py

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2012 Laurent Bachelier
+#
+# This file is part of weboob.
+#
+# weboob 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.
+#
+# weboob 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 weboob. If not, see <http://www.gnu.org/licenses/>.
+
+from __future__ import absolute_import
+
+from .browser import Browser, Weboob
+
+import requests
+
+from nose.plugins.skip import SkipTest
+
+
+def test_base():
+    b = Browser()
+    r = b.location('http://httpbin.org/headers')
+    assert isinstance(r.text, unicode)
+    assert 'Firefox' in r.text
+    assert 'python' not in r.text
+    assert 'identity' not in r.text
+    assert b.url == 'http://httpbin.org/headers'
+
+
+def test_redirects():
+    b = Browser()
+    b.location('http://httpbin.org/redirect/1')
+    assert b.url == 'http://httpbin.org/get'
+
+
+def test_brokenpost():
+    """
+    Tests _fix_redirect()
+    """
+    try:
+        b = Browser()
+        # postbin is picky with empty posts. that's good!
+        r = b.location('http://www.postbin.org/', {})
+        # ensures empty data (but not None) does a POST
+        assert r.request.method == 'POST'
+        # ensure we were redirected after submitting a post
+        assert len(r.url) >= len('http://www.postbin.org/')
+        # send a POST with data
+        b.location(r.url, {'hello': 'world'})
+        r = b.location(r.url + '/feed')
+        assert 'hello' in r.text
+        assert 'world' in r.text
+    except requests.HTTPError, e:
+        if str(e).startswith('503 '):
+            raise SkipTest('Quota exceeded')
+        else:
+            raise
+
+
+def test_weboob():
+    class BooBrowser(Browser):
+        PROFILE = Weboob('0.0')
+
+    b = BooBrowser()
+    r = b.location('http://httpbin.org/headers')
+    assert 'weboob/0.0' in r.text
+    assert 'identity' in r.text