python-formalchemy 1.4.2-1 / usr / lib / python2.7 / dist-packages / formalchemy / tests / test_binary.py

import cgi
import shutil
import tempfile
from StringIO import StringIO

from formalchemy.fields import FileFieldRenderer
from formalchemy.ext import fsblob
from formalchemy.tests import *
from webtest import TestApp, selenium
from webob import multidict

BOUNDARY='testdata'
ENVIRON = {
        'REQUEST_METHOD':'POST',
        'CONTENT_TYPE': 'multipart/form-data;boundary="%s"' % BOUNDARY
        }
TEST_DATA = '''--%s
Content-Disposition: form-data; name="Binaries--file"; filename="test.js"
Content-Type: application/x-javascript

var test = null;

--%s--
''' % (BOUNDARY, BOUNDARY)
EMPTY_DATA = '''--%s
Content-Disposition: form-data; name="Binaries--file"; filename=""
Content-Type: application/x-javascript

--%s--
''' % (BOUNDARY, BOUNDARY)
REMOVE_DATA = '''--%s
Content-Disposition: form-data; name="Binaries--file--remove"
1
--%s
Content-Disposition: form-data; name="Binaries--file"; filename=""
Content-Type: application/x-javascript

--%s--
''' % (BOUNDARY, BOUNDARY, BOUNDARY)


def get_fields(data):
    return multidict.MultiDict.from_fieldstorage(cgi.FieldStorage(fp=StringIO(data), environ=ENVIRON))

def test_binary():
    r"""

    Notice that those tests assume that the FileFieldRenderer work with LargeBinary type
    *and* String type if you only want to store file path in your DB.

    Configure a fieldset with a file field

        >>> fs = FieldSet(Three)
        >>> record = fs.model
        >>> fs.configure(include=[fs.bar.with_renderer(FileFieldRenderer)])
        >>> isinstance(fs.bar.renderer, FileFieldRenderer)
        True

    At creation time only the input field is rendered

        >>> print fs.render()
        <div>
         <label class="field_opt" for="Three--bar">
          Bar
         </label>
         <input id="Three--bar" name="Three--bar" type="file" />
        </div>
        <script type="text/javascript">
         //<![CDATA[
        document.getElementById("Three--bar").focus();
        //]]>
        </script>

    If the field has a value then we add a check box to remove it

        >>> record.bar = '/path/to/file'
        >>> print fs.render()
        <div>
         <label class="field_opt" for="Three--bar">
          Bar
         </label>
         <input id="Three--bar" name="Three--bar" type="file" />
         <input id="Three--bar--remove" name="Three--bar--remove" type="checkbox" value="1" />
         <label for="Three--bar--remove">
          Remove
         </label>
        </div>
        <script type="text/javascript">
         //<![CDATA[
        document.getElementById("Three--bar").focus();
        //]]>
        </script>

    Now submit form with empty value

        >>> fs.rebind(data={'Three--bar':''})
        >>> fs.validate()
        True
        >>> fs.sync()

    The field value does not change

        >>> print record.bar
        /path/to/file

    Try to remove it by checking the checkbox

        >>> fs.rebind(data={'Three--bar':'', 'Three--bar--remove':'1'})
        >>> fs.validate()
        True
        >>> fs.sync()

    The field value is removed

        >>> print record.bar
        <BLANKLINE>

    Also check that this work with cgi.FieldStorage

        >>> fs = FieldSet(Binaries)
        >>> record = fs.model

    We need test data

        >>> data = get_fields(TEST_DATA)
        >>> print data.getone('Binaries--file')
        FieldStorage(u'Binaries--file', u'test.js')

        >>> fs.rebind(data=data)
        >>> if fs.validate(): fs.sync()

    We get the file, yeah.

        >>> print record.file
        var test = null;
        <BLANKLINE>

    Now submit form with empty value

        >>> data = get_fields(EMPTY_DATA)
        >>> fs.rebind(data=data)
        >>> if fs.validate(): fs.sync()

    The field value dos not change

        >>> print record.file
        var test = null;
        <BLANKLINE>

    Remove file

        >>> data = get_fields(REMOVE_DATA)
        >>> fs.rebind(data=data)
        >>> if fs.validate(): fs.sync()

    The field value is now empty

        >>> print record.file
        <BLANKLINE>

    See what append in read only mode

        >>> record.file = 'e'*1000
        >>> print fs.file.render_readonly()
        1 KB

        >>> record.file = 'e'*1000*1024
        >>> print fs.file.render_readonly()
        1000.00 KB

        >>> record.file = 'e'*2*1024*1024
        >>> print fs.file.render_readonly()
        2.00 MB

    """

class BlobTestCase(unittest.TestCase):

    renderer = fsblob.FileFieldRenderer

    def setUp(self):
        self.wd = tempfile.mkdtemp()
        self.binary = Three()
        self.fs = FieldSet(Three)
        self.fs.configure(include=[self.fs.bar, self.fs.foo])
        self.fs.foo.set(renderer=self.renderer.new(
                        storage_path=self.wd,
                        url_prefix='/media'))
        self.app = TestApp(application(self.binary, self.fs))

    def test_file(self):
        resp = self.app.get('/')
        resp.mustcontain('type="file"')
        resp = self.app.post('/', {"Three--bar":'bar'},
                             upload_files=[('Three--foo', 'foo.txt', 'data')])
        resp = self.app.get('/')
        resp.mustcontain('<a href="/media/', '/foo.txt">', 'foo.txt (1 KB)')
        self.assert_(self.binary.foo.endswith('foo.txt'), repr(self.binary.foo))
        resp = self.app.get('/')
        resp.mustcontain('name="Three--foo--remove"',
                         '<a href="/media/', '/foo.txt">', 'foo.txt (1 KB)')

        # no change
        form = resp.form
        resp = form.submit()
        resp.mustcontain('<a href="/media/', '/foo.txt">', 'foo.txt (1 KB)')
        self.assert_(self.binary.foo.endswith('foo.txt'), repr(self.binary.foo))

        # remove file
        resp = self.app.get('/')
        form = resp.form
        form['Three--foo--remove'] = '1'
        resp = form.submit()
        self.assert_(self.binary.foo == '', repr(self.binary.foo))
        resp.mustcontain(no='foo.txt (1 KB)')

    def tearDown(self):
        shutil.rmtree(self.wd)