encoding-method-test/test_get_webpage.py at master · mdruger/encoding-method-test · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
from mock import Mock, patch
from StringIO import StringIO
import unittest2
from urllib2 import HTTPError

from get_webpage import get_webform_content


class WhenReadingWebFormContent(unittest2.TestCase):

    webform_url = "http://www.google.com"

    def test_require_url(self):
        self.assertRaises(TypeError, get_webform_content)

    @patch("get_webpage.urlopen")
    def test_open_correct_webform(self, urlopen):
        urlopen.return_value = StringIO("")
        get_webform_content(self.webform_url)
        urlopen.assert_called_once_with(self.webform_url)

    @patch('get_webpage.massage_data')
    @patch("get_webpage.urlopen")
    def test_read_webform_content(self, urlopen, massage_data):
        get_webform_content(self.webform_url)
        urlopen(self.webform_url).read.assert_called_once_with()

    @patch('get_webpage.massage_data')
    @patch("get_webpage.urlopen")
    def test_return_converted_webform_content(self, urlopen, massage_data):
        massage_data.return_value = 'awesome'
        self.assertEqual(get_webform_content(self.webform_url), 'awesome')

    @patch("get_webpage.urlopen")
    def test_return_false_on_http_errors(self, urlopen):
        urlopen.side_effect = HTTPError('', '', '', '', StringIO(''))
        self.assertFalse(get_webform_content(self.webform_url))

    @patch('get_webpage.massage_data')
    @patch("get_webpage.urlopen")
    def test_decode_url_content_to_unicode(self, urlopen, massage_data):
        mock_url_content = Mock()
        urlopen(self.webform_url).read.return_value = mock_url_content
        get_webform_content(self.webform_url)
        mock_url_content.decode.assert_called_once_with('iso-8859-1')

    @patch('get_webpage.massage_data')
    @patch("get_webpage.urlopen")
    def test_call_massage_data_with_unicode_param(self, urlopen, massage_data):
        mock_url_content = Mock()
        urlopen(self.webform_url).read.return_value = mock_url_content
        get_webform_content(self.webform_url)
        massage_data.assert_called_with(mock_url_content.decode('iso-8859-1'))

    @patch('facebook_canvas.lib.webform_get.support_ssl_webforms')
    @patch('facebook_canvas.lib.webform_get.urlopen')
    def should_rtn_str_with_capital_thorn_in_latin1_to_unicode(self, urlopen,
                                                               ssl_webforms):
        mock_url_content = Mock()
        # DE is capital Thorn in latin1
        mock_url_content.return_value = 'awesome\xde'
        urlopen(self.webform_url).read.return_value = mock_url_content

        # 00de is capital Thorn in latin1 converted to unicode
        ssl_webforms.return_value = u'awesome\u00de'
        self.assertEqual(get_webform_content(98765),
                         mock_url_content.return_value.decode('iso-8859-1'))

    @patch('facebook_canvas.lib.webform_get.urlopen')
    def should_return_webform_content_as_valid_unicode(self, urlopen):
        urlopen.return_value = StringIO('webform_content')
        self.assertIsInstance(get_webform_content(12345), unicode)


if __name__ == '__main__':
    unittest2.main()