Produits Entr'ouvert » Fargo

import os

from django.core.management.base import BaseCommand

from django.core.files.base import ContentFile

from django.core.urlresolvers import reverse

from fargo.utils import make_url

from fargo.fargo.models import Document

from fargo.oauth2.models import OAuth2TempFile, OAuth2Client

from fargo.fargo.utils import cleanup

class Command(BaseCommand):

    help = 'Push documents inside fargo, returns URLs'

    def add_arguments(self, parser):

        parser.add_argument('--client-id', type=int)

        parser.add_argument('redirect_uri')

        parser.add_argument('paths', nargs='+')

    def handle(self, redirect_uri, paths, client_id, **options):

        client = OAuth2Client.objects.get(id=client_id)

        for path in paths:

            with open(path) as file_object:

                filename = os.path.basename(path)

                f = ContentFile(file_object.read(), name=filename)

                document = Document.objects.get_by_file(f)

                oauth2_document = OAuth2TempFile.objects.create(

                    client=client,

                    document=document,

                    filename=filename)

                uri = reverse('oauth2-put-document-authorize', args=[oauth2_document.pk])

                print 'https://localhost:8000' + make_url(uri, redirect_uri=redirect_uri)

import os

def test_command_put_document(db, capsys, app, john_doe):

    call_command('oauth2-create-client', 'test', 'https://example.com/')

    client = OAuth2Client.objects.get()

    path = os.path.join(os.path.dirname(__file__),  'pdf-sample.pdf')

    redirect_uri = 'https://example.com/'

    r = call_command('oauth2-put-document', '--client-id=%s' % client.pk, redirect_uri, path)

    out, err = capsys.readouterr()

    assert err == ''

    url = out.strip()

    response = app.get(url).follow()

    response.form.set('username', john_doe.username)

    response.form.set('password', john_doe.username)

    response = response.form.submit().follow()

    assert 'pdf-sample.pdf' in response

    import pdb

    pdb.set_trace()

    temp_file = OAuth2TempFile.objects.get()

    assert temp_file.uuid in response

    response = response.form.submit('accept')

    assert response['Location'] == redirect_uri

    assert UserDocument.objects.filter(user=john_doe, document=temp_file.document).exists()

    assert OAuth2TempFile.objects.count() == 0