コード例 #1
0
class TestQuotationProvider(unittest.TestCase):

    def setUp(self):
        self.provider = QuotationProvider(quotation_url='http://localhost/?')

    @patch('os.environ.get')
    def test_constructor_empty_quotation_env_var(self, mock_env_get):
        mock_env_get.return_value = ''

        try:
            QuotationProvider()
            self.fail("Should not be able to instantiate provider without mongo")
        except SettingsError as e:
            error = "You must supply quotation_url or set the QUOTATION_URL environment variable."
            self.assertEqual(error, '{0}'.format(e))

    @patch('os.environ.get')
    def test_constructor_with_quotation_env_var(self, mock_env_get):
        mock_env_get.return_value = 'http://localhost/?'

        provider = QuotationProvider()

        self.assertEqual('http://localhost/?', provider.BASE_URL)

    @patch('requests.get')
    def test_create_error_no_hashtags(self, mock_get):
        mock_get.return_value = MagicMock(status_code=500)

        quotation = self.provider.create({}, 140)

        self.assertEqual('No quotations found', quotation)
        mock_get.assert_called_with('http://localhost/?')

    @patch('requests.get')
    def test_create_error_with_hashtags(self, mock_get):
        mock_get.return_value = MagicMock(status_code=500)
        mention = {'entities': {'hashtags': [{'text': 'love'}, {'text': 'hate'}]}}

        quotation = self.provider.create(mention, 140)

        self.assertEqual('No quotations found matching #love #hate', quotation)
        mock_get.assert_called_with('http://localhost/?&search=love'
                                    '&search=hate')

    @patch('requests.get')
    def test_create_no_quotations(self, mock_get):
        mock_result = MagicMock(status_code=200)
        mock_result.json.return_value = {'results': []}
        mock_get.return_value = mock_result

        quotation = self.provider.create({}, 140)

        self.assertEqual('No quotations found', quotation)
        mock_get.assert_called_with('http://localhost/?')

    @patch('requests.get')
    def test_create_success(self, mock_get):
        mock_result = MagicMock(status_code=200)
        mock_result.json.return_value = {'results': [{
            'text': 'Here I stay',
            'author': 'Henrietta'
        }]}
        mock_get.return_value = mock_result

        message = self.provider.create({}, 140)

        self.assertEqual('Here I stay - Henrietta', message)
コード例 #2
0
 def setUp(self):
     self.provider = QuotationProvider(quotation_url='http://localhost/?')