-from django.test import TestCase
-from django.test.client import Client
-from django.core.urlresolvers import reverse
-
-from django.utils import simplejson as json
-
-from django.contrib.auth.models import User
-
-import settings
-from os.path import join, dirname
-from StringIO import StringIO
-import shutil
-import tempfile
-
-REPO_TEMPLATES = join(dirname(__file__), 'data')
-
-def temprepo(name):
- from functools import wraps
-
- def decorator(func):
-
-
- @wraps(func)
- def decorated(self, *args, **kwargs):
- clean = False
- try:
- temp = tempfile.mkdtemp("-test", func.__name__)
- shutil.copytree(join(REPO_TEMPLATES, name), join(temp, 'repo'), False)
- settings.REPOSITORY_PATH = join(temp, 'repo')
- func(self, *args, **kwargs)
- clean = True
- finally:
- if not clean and self.response:
- print "RESULT", func.__name__, ">>>"
- print self.response.content
- print "<<<"
- else:
- shutil.rmtree(temp, True)
-
- settings.REPOSITORY_PATH = ''
-
- return decorated
- return decorator
-
-
-class SimpleTest(TestCase):
-
- def setUp(self):
- self.response = None
- u = User.objects.create_user('admin', 'test@localhost', 'admin')
- u.save()
-
- @temprepo('clean')
- def test_documents_get_anonymous(self):
- self.response = self.client.get( reverse("document_list_view") )
- self.assert_json_response({
- u'documents': [],
- })
-
- @temprepo('clean')
- def test_documents_get_with_login(self):
- self.assertTrue(self.client.login(username='admin', password='admin'))
-
- self.response = self.client.get( reverse("document_list_view") )
- self.assert_json_response({
- u'documents': [],
- })
-
- @temprepo('clean')
- def test_document_creation(self):
- self.assertTrue(self.client.login(username='admin', password='admin'))
-
- infile = tempfile.NamedTemporaryFile("w+")
- infile.write('012340123456789')
- infile.flush()
- infile.seek(0)
-
- self.response = self.client.post( reverse("document_list_view"),
- data = {
- 'bookname': 'testbook',
- 'ocr_file': infile,
- 'generate_dc': False,
- })
-
- r = self.assert_json_response( {
- 'url': reverse('document_view', args=['testbook']),
- 'name': 'testbook',
- # can't test revision number, 'cause it's random
- }, code=201)
-
- created_rev = r['revision']
- self.response = self.client.get(r['url'])
-
- result = self.assert_json_response({
- u'public_revision': created_rev,
- # u'size': 15,
- })
-
-
- @temprepo('simple')
- def test_document_meta_get_with_login(self):
- self.assertTrue(self.client.login(username='admin', password='admin'))
-
- self.response = self.client.get( reverse("document_list_view") )
- self.assert_json_response({
- # u'latest_rev': u'f94a263812dbe46a3a13d5209bb119988d0078d5',
- u'documents': [
- {u'url': u'/api/documents/sample', u'name': u'sample', u'parts': []},
- {u'url': u'/api/documents/sample_pl', u'name': u'sample_pl', u'parts': []}
- ],
- })
-
- self.response = self.client.get( \
- reverse("document_view", args=['sample']) )
-
- self.assert_json_response({
- #u'latest_shared_rev': u'f94a263812dbe46a3a13d5209bb119988d0078d5',
- #u'text_url': reverse("doctext_view", args=[u'sample']),
- #u'dc_url': reverse("docdc_view", args=[u'sample']),
- # u'parts_url': reverse("docparts_view", args=[u'sample']),
- u'name': u'sample',
- # u'size': 20,
- })
-
-
- @temprepo('simple')
- def test_document_text_with_login(self):
- self.assertTrue(self.client.login(username='admin', password='admin'))
-
- self.response = self.client.get( \
- reverse("document_view", args=['sample']) )
-
- resp = self.assert_json_response({
- #u'latest_shared_rev': u'f94a263812dbe46a3a13d5209bb119988d0078d5',
- #u'text_url': reverse("doctext_view", args=[u'sample']),
- #u'dc_url': reverse("docdc_view", args=[u'sample']),
- # u'parts_url': reverse("docparts_view", args=[u'sample']),
- u'name': u'sample',
- # u'size': 20,
- })
-
- self.response = self.client.get(resp['text_url'])
- self.assertEqual(self.response.status_code, 200)
- self.assertEqual(self.response.content, "Ala ma kota\n")
-
-
- @temprepo('simple')
- def test_document_text_update(self):
- self.assertTrue(self.client.login(username='admin', password='admin'))
- TEXT = u"Ala ma kota i psa"
-
- self.response = self.client.get(
- reverse("document_view", args=['sample']) )
-
- resp = self.assert_json_response()
-
- self.response = self.client.put(resp['text_url'], {'contents': TEXT })
- result = self.assert_json_response(must_have={u'document': u'sample'} )
-
- #self.response = self.client.get(result['url'])
- #self.assertEqual(self.response.content, TEXT)
-
- def assert_json_response(self, must_have={}, exclude=[], code=200):
- self.assertEqual(self.response.status_code, code)
- result = json.loads(self.response.content)
-
- for (k,v) in must_have.items():
- self.assertTrue(result.has_key(k), "Required field '%s' missing in response." % k)
- self.assertEqual(result[k], v)
-
- if exclude is True:
- for (k,v) in result.items():
- self.assertTrue(must_have.has_key(k))
- self.assertEqual(must_have[k], v)
-
- for key in exclude:
- self.assertFalse(result.has_key(key))
-
- return result
-
\ No newline at end of file
fnp / redakcja.git / blobdiff