class TestApi(unittest.TestCase):
def setUp(self):
"""Utility code shared among all tests."""
self.cr = ChimpRewriter('*****@*****.**', 'test_api_key', 'test_api_name')
def test_init(self):
"""Test initialization of ChimpRewriter.
ChimpRewriter is initialized on every test run and stored as self.sc.
We need to test for stored values if class was
initialized correctly.
"""
self.assertEquals(self.cr._email, '*****@*****.**')
self.assertEquals(self.cr._apikey, 'test_api_key')
self.assertEquals(self.cr._aid, 'test_api_name')
self.assertIsInstance(self.cr, ChimpRewriter)
@mock.patch('chimprewriter.urllib2')
def test_unique_variation_default_call(self, urllib2):
"""Test call of unique_variation() with default values."""
# mock response from ChimpRewriter
mocked_response = 'My cat is über cold.'
urllib2.urlopen.return_value.read.return_value = mocked_response
# test call
self.assertEquals(
self.cr.unique_variation(u'My cat is über cool.'),
u'My cat is über cold.',
)
@mock.patch('chimprewriter.urllib2')
def test_text_with_spintax_default_call(self, urllib2):
"""Test call of text_with_spintax_call() with default values."""
# mock response from ChimpRewriter
mocked_response = 'My cat is über {cold|cool}.'
urllib2.urlopen.return_value.read.return_value = mocked_response
# test call
self.assertEquals(
self.cr.text_with_spintax(u'My cat is über cool.'),
u'My cat is über {cold|cool}.',
)
@mock.patch('chimprewriter.urllib2')
def test_errors(self, urllib2):
mocked_response = 'failed:Credentials check result:InvalidEmail'
urllib2.urlopen.return_value.read.return_value = mocked_response
with self.assertRaises(ex.AuthenticationError):
self.cr._send_request(
'METHOD',
'Test text.',
ChimpRewriter.DEFAULT_PARAMS_SPIN
)
mocked_response = 'failed:Credentials check result:MaxQueriesReached'
urllib2.urlopen.return_value.read.return_value = mocked_response
with self.assertRaises(ex.QuotaLimitError):
self.cr._send_request(
'METHOD',
'Test text.',
ChimpRewriter.DEFAULT_PARAMS_SPIN
)
mocked_response = 'failed:Credentials check result:DatabaseFailure'
urllib2.urlopen.return_value.read.return_value = mocked_response
with self.assertRaises(ex.InternalError):
self.cr._send_request(
'METHOD',
'Test text.',
ChimpRewriter.DEFAULT_PARAMS_SPIN
)
mocked_response = 'failed:There are no words in your article!'
urllib2.urlopen.return_value.read.return_value = mocked_response
with self.assertRaises(ex.ArticleError):
self.cr._send_request(
'METHOD',
'Test text.',
ChimpRewriter.DEFAULT_PARAMS_SPIN
)
mocked_response = 'failed:Cars are foo!'
urllib2.urlopen.return_value.read.return_value = mocked_response
with self.assertRaises(ex.UnknownError):
self.cr._send_request(
'METHOD',
'Test text.',
ChimpRewriter.DEFAULT_PARAMS_SPIN
)
def setUp(self):
"""Utility code shared among all tests."""
self.cr = ChimpRewriter('*****@*****.**', 'test_api_key', 'test_api_name')
