class BoxViewTestCase(unittest.TestCase):
def setUp(self):
self.api = BoxView('<box view api key>')
def test_initials(self):
# api key is required
self.assertRaises(ValueError, BoxView)
now = datetime.datetime.utcnow().replace(microsecond=0)
dtfiso = now.isoformat()
dfiso = now.date().isoformat()
self.assertEqual(dtfiso, format_date(dtfiso))
self.assertEqual(dtfiso, format_date(now))
self.assertEqual(dfiso, format_date(now.date()))
headers = {'Content-Type': 'text/plain'}
self.assertEqual('text/plain', get_mimetype_from_headers(headers))
headers = {'Content-Type': 'text/plain; charset=utf-8'}
self.assertEqual('text/plain', get_mimetype_from_headers(headers))
@patch.object(Session, 'request')
def test_crate_document_from_url(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.create_document(url=test_url, name='Test Document')
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
# url of file param is required
self.assertRaises(ValueError, self.api.create_document)
@patch.object(Session, 'request')
def test_create_document_from_url(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.create_document_from_url(test_url,
name='Test Document')
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_create_document_from_file(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_document)
mock_request.return_value = response
stream = six.BytesIO()
result = self.api.create_document_from_file(stream,
name='Test Document')
self.assertEqual(result, test_document)
result = self.api.create_document_from_file(__file__,
name='Test Document')
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_get_document(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.get_document(test_document['id'])
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_get_documents(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document_list)
mock_request.return_value = response
now = datetime.datetime.utcnow()
result = self.api.get_documents(limit=10, created_before=now)
self.assertIsNotNone(result)
self.assertEqual(result, test_document_list)
@patch.object(Session, 'request')
def test_update_document(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.update_document(test_document['id'],
name='TestDocument')
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_delete_document(self, mock_request):
response = Response()
response.status_code = 204
mock_request.return_value = response
self.api.delete_document(test_document['id'])
@patch.object(Session, 'request')
def test_get_document_content(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
response._content = 'test'
response.raw = six.StringIO('test')
mock_request.return_value = response
stream = six.StringIO()
mimetype = self.api.get_document_content(stream, test_document['id'])
self.assertEqual(stream.getvalue(), response._content)
self.assertEqual(mimetype, response.headers['Content-Type'])
stream = six.StringIO()
self.api.get_document_content(stream,
test_document['id'],
extension='.pdf')
self.assertEqual(stream.getvalue(), response._content)
stream = six.StringIO()
self.api.get_document_content(stream,
test_document['id'],
extension='.zip')
self.assertEqual(stream.getvalue(), response._content)
stream = six.StringIO()
# allowed only .zip and .pdf extensions
self.assertRaises(ValueError,
self.api.get_document_content,
stream,
test_document['id'],
extension='.docx')
@patch.object(Session, 'request')
def test_get_document_content_to_string(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
response._content = 'test'
response.raw = six.StringIO('test')
mock_request.return_value = response
doc_id = test_document['id']
result, mimetype = self.api.get_document_content_to_string(doc_id)
self.assertIsNotNone(result)
self.assertEqual(result, response._content)
self.assertEqual(mimetype, response.headers['Content-Type'])
@patch.object(Session, 'request')
def test_get_document_content_to_file(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
response._content = 'test'
response.raw = six.StringIO('test')
mock_request.return_value = response
filename = 'boxview.txt'
mimetype = self.api.get_document_content_to_file(filename,
test_document['id'])
self.assertEqual(mimetype, response.headers['Content-Type'])
self.assertTrue(os.path.exists(filename))
try:
os.remove(filename)
except OSError:
pass
@patch.object(Session, 'request')
def test_get_document_content_mimetype(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
mock_request.return_value = response
mimetype = self.api.get_document_content_mimetype(test_document['id'])
self.assertEqual(mimetype, response.headers['Content-Type'])
@patch.object(Session, 'request')
def test_create_session(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_session)
mock_request.return_value = response
expires_at = datetime.datetime.utcnow()
result = self.api.create_session(test_document['id'],
duration=600,
expires_at=expires_at)
self.assertIsNotNone(result)
self.assertEqual(result['id'], test_session['id'])
@patch.object(Session, 'request')
def test_ready_to_view(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.ready_to_view(test_document['id'])
self.assertIsNotNone(result)
self.assertTrue(bool(result))
response._content = json.dumps(dict(test_document, status='error'))
result = self.api.ready_to_view(test_document['id'])
self.assertFalse(bool(result))
@patch.object(Session, 'request')
def test_request_error(self, mock_request):
response = Response()
response.status_code = 401
response._content = 'Unauthorized'
response.reason = 'Unauthorized'
mock_request.return_value = response
self.assertRaises(BoxViewError,
self.api.get_document,
test_document['id'])
class BoxViewTestCase(unittest.TestCase):
def setUp(self):
self.api = BoxView('<box view api key>')
def test_initials(self):
# api key is required
self.assertRaises(ValueError, BoxView)
now = datetime.datetime.utcnow().replace(microsecond=0)
dtfiso = now.isoformat()
dfiso = now.date().isoformat()
self.assertEqual(dtfiso, format_date(dtfiso))
self.assertEqual(dtfiso, format_date(now))
self.assertEqual(dfiso, format_date(now.date()))
headers = {'Content-Type': 'text/plain'}
self.assertEqual('text/plain', get_mimetype_from_headers(headers))
headers = {'Content-Type': 'text/plain; charset=utf-8'}
self.assertEqual('text/plain', get_mimetype_from_headers(headers))
@patch.object(Session, 'request')
def test_crate_document_from_url(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.create_document(url=test_url, name='Test Document')
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
# url of file param is required
self.assertRaises(ValueError, self.api.create_document)
@patch.object(Session, 'request')
def test_create_document_from_url(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.create_document_from_url(test_url,
name='Test Document')
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_create_document_from_file(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_document)
mock_request.return_value = response
stream = six.BytesIO()
result = self.api.create_document_from_file(stream,
name='Test Document')
self.assertEqual(result, test_document)
result = self.api.create_document_from_file(__file__,
name='Test Document')
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_get_document(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.get_document(test_document['id'])
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_get_documents(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document_list)
mock_request.return_value = response
now = datetime.datetime.utcnow()
result = self.api.get_documents(limit=10, created_before=now)
self.assertIsNotNone(result)
self.assertEqual(result, test_document_list)
@patch.object(Session, 'request')
def test_update_document(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.update_document(test_document['id'],
name='TestDocument')
self.assertIsNotNone(result)
self.assertEqual(result, test_document)
@patch.object(Session, 'request')
def test_delete_document(self, mock_request):
response = Response()
response.status_code = 204
mock_request.return_value = response
self.api.delete_document(test_document['id'])
@patch.object(Session, 'request')
def test_get_document_content(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
response._content = 'test'
response.raw = six.StringIO('test')
mock_request.return_value = response
stream = six.StringIO()
mimetype = self.api.get_document_content(stream, test_document['id'])
self.assertEqual(stream.getvalue(), response._content)
self.assertEqual(mimetype, response.headers['Content-Type'])
stream = six.StringIO()
self.api.get_document_content(stream,
test_document['id'],
extension='.pdf')
self.assertEqual(stream.getvalue(), response._content)
stream = six.StringIO()
self.api.get_document_content(stream,
test_document['id'],
extension='.zip')
self.assertEqual(stream.getvalue(), response._content)
stream = six.StringIO()
# allowed only .zip and .pdf extensions
self.assertRaises(ValueError,
self.api.get_document_content,
stream,
test_document['id'],
extension='.docx')
@patch.object(Session, 'request')
def test_get_document_content_to_string(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
response._content = 'test'
response.raw = six.StringIO('test')
mock_request.return_value = response
doc_id = test_document['id']
result, mimetype = self.api.get_document_content_to_string(doc_id)
self.assertIsNotNone(result)
self.assertEqual(result, response._content)
self.assertEqual(mimetype, response.headers['Content-Type'])
@patch.object(Session, 'request')
def test_get_document_content_to_file(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
response._content = 'test'
response.raw = six.StringIO('test')
mock_request.return_value = response
filename = 'boxview.txt'
mimetype = self.api.get_document_content_to_file(
filename, test_document['id'])
self.assertEqual(mimetype, response.headers['Content-Type'])
self.assertTrue(os.path.exists(filename))
try:
os.remove(filename)
except OSError:
pass
@patch.object(Session, 'request')
def test_get_document_content_mimetype(self, mock_request):
response = Response()
response.status_code = 200
response.headers['Content-Type'] = 'text/plain'
mock_request.return_value = response
mimetype = self.api.get_document_content_mimetype(test_document['id'])
self.assertEqual(mimetype, response.headers['Content-Type'])
@patch.object(Session, 'request')
def test_create_session(self, mock_request):
response = Response()
response.status_code = 201
response._content = json.dumps(test_session)
mock_request.return_value = response
expires_at = datetime.datetime.utcnow()
result = self.api.create_session(test_document['id'],
duration=600,
expires_at=expires_at)
self.assertIsNotNone(result)
self.assertEqual(result['id'], test_session['id'])
@patch.object(Session, 'request')
def test_ready_to_view(self, mock_request):
response = Response()
response.status_code = 200
response._content = json.dumps(test_document)
mock_request.return_value = response
result = self.api.ready_to_view(test_document['id'])
self.assertIsNotNone(result)
self.assertTrue(bool(result))
response._content = json.dumps(dict(test_document, status='error'))
result = self.api.ready_to_view(test_document['id'])
self.assertFalse(bool(result))
@patch.object(Session, 'request')
def test_request_error(self, mock_request):
response = Response()
response.status_code = 401
response._content = 'Unauthorized'
response.reason = 'Unauthorized'
mock_request.return_value = response
self.assertRaises(BoxViewError, self.api.get_document,
test_document['id'])
