def test_successful_conversion(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key',
text='xlsx output')
pdf_fo = io.BytesIO(b'pdf content')
c = Client('fake_key')
with NamedTemporaryFile(suffix="test.pdf") as tf:
filename = tf.name
tf.write(b"Hello world")
tf.file.close()
filename_out = filename.replace(".pdf", ".xlsx")
try:
s = c.convert(filename, filename_out)
with open(filename_out) as fd:
self.assertEqual(fd.read(), "xlsx output")
finally:
try:
os.unlink(filename_out)
except OSError:
pass
def test_missing_format(self):
self.assertEqual(('foo.xlsx', 'xlsx-multiple'),
Client.ensure_format_ext('foo', None))
self.assertEqual(('foo.txt.xlsx', 'xlsx-multiple'),
Client.ensure_format_ext('foo.txt', None))
self.assertEqual(('foo.xlsx', 'xlsx-multiple'),
Client.ensure_format_ext('foo.xlsx', None))
def test_different_api_url(self):
with requests_mock.mock() as m:
m.post('http://example.com/api?key=fake_key', text='xlsx output')
pdf_fo = io.BytesIO(b'pdf content')
c = Client('fake_key', api_url='http://example.com/api')
s = c.dump(pdf_fo, 'csv')
self.assertEqual(b'xlsx output', consume(s))
def test_unmodified(self):
self.assertEqual(('foo.csv', 'csv'),
Client.ensure_format_ext('foo.csv', 'csv'))
self.assertEqual(('foo.xlsx', 'xlsx-multiple'),
Client.ensure_format_ext('foo.xlsx', 'xlsx-multiple'))
self.assertEqual(('foo.xlsx', 'xlsx-multiple'),
Client.ensure_format_ext('foo.xlsx', 'xlsx-multiple'))
self.assertEqual(('foo.xml', 'xml'),
Client.ensure_format_ext('foo.xml', 'xml'))
def test_successful_conversion(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key',
text='xlsx output')
pdf_fo = io.BytesIO(b'pdf content')
c = Client('fake_key')
s = c.dump(pdf_fo, 'csv')
self.assertEqual(b'xlsx output', consume(s))
def test_response_limit_exceeded(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key', status_code=402)
pdf_fo = io.BytesIO(b'pdf content')
c = Client('fake_key')
self.assertRaisesRegexp(APIException, 'Usage limit exceeded',
c.dump, pdf_fo)
def test_response_unauthorized(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=wrong_key', status_code=401)
pdf_fo = io.BytesIO(b'pdf content')
c = Client('wrong_key')
self.assertRaisesRegexp(APIException, 'Unauthorized API key',
c.dump, pdf_fo)
def test_invalid_format(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key', text='xlsx output')
pdf_fo = io.BytesIO(b'pdf content')
c = Client('fake_key')
self.assertRaisesRegexp(ValueError, 'Invalid output format', c.dump, pdf_fo, 'invalid_format')
def test_response_unknown_file_format(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key', status_code=403)
png_fo = io.BytesIO(b'png content')
c = Client('fake_key')
self.assertRaisesRegexp(APIException, 'Unknown format requested',
c.dump, png_fo)
def test_missing_api_key(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key', text='xlsx output')
pdf_fo = io.BytesIO(b'pdf content')
c = Client('')
self.assertRaisesRegexp(APIException, 'Invalid API key', c.dump, pdf_fo, 'csv')
def test_successful_conversion_bytes(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key', content=b'xlsx output')
with NamedTemporaryFile(suffix="test.pdf") as tf:
filename = tf.name
tf.write(b"Hello world")
tf.file.close()
output = Client('fake_key').convert(filename)
self.assertEqual(b'xlsx output', output)
def test_successful_conversion_string(self):
with requests_mock.mock() as m:
m.post('https://pdftables.com/api?key=fake_key', text='csv output')
with NamedTemporaryFile(suffix="test.pdf") as tf:
filename = tf.name
tf.write(b"Hello world")
tf.file.close()
output = Client('fake_key').convert(filename, out_format="csv")
self.assertEqual('csv output', output)
def test_remaining(self):
with requests_mock.mock() as m:
m.get('https://pdftables.com/api/remaining?key=fake_key', text='8584')
c = Client('fake_key')
self.assertEqual(c.remaining(), 8584)
def test_stdout(self):
self.assertEqual((None, 'xlsx-multiple'),
Client.ensure_format_ext(None, None))
self.assertEqual((None, 'csv'), Client.ensure_format_ext(None, 'csv'))
def test_incorrect_ext(self):
self.assertEqual(('foo.txt.csv', 'csv'),
Client.ensure_format_ext('foo.txt', 'csv'))
self.assertEqual(('foo.xlsx.csv', 'csv'),
Client.ensure_format_ext('foo.xlsx', 'csv'))
def test_missing_ext(self):
self.assertEqual(('foo.csv', 'csv'),
Client.ensure_format_ext('foo', 'csv'))
def test_wrong_format(self):
self.assertRaises(ValueError, Client.ensure_format_ext, 'foo.csv',
'txt')
self.assertRaises(ValueError, Client('key').dump, 'foo.pdf', 'txt')
