source: main/ikobacustom.skeleton/trunk/src/ikobacustom/skeleton/documents/tests/test_browser.py @ 12274

## $Id: test_browser.py 12269 2014-12-20 16:12:59Z henrik $
##
## Copyright (C) 2014 Uli Fouquet & Henrik Bettermann
## This program is free software; you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 2 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program; if not, write to the Free Software
## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
##
"""
Document browser and functional tests.
"""
import os
import grok
import datetime
from cStringIO import StringIO
from zope.component import queryUtility, getUtility
from zope.event import notify
from zope.interface.verify import verifyObject, verifyClass
from waeup.ikoba.documents.tests.test_batching import DocumentImportExportSetup
from ikobacustom.skeleton.documents.document import SkeletonDocument
from ikobacustom.skeleton.documents.export import SkeletonDocumentExporter
from ikobacustom.skeleton.documents.batching import SkeletonDocumentProcessor
from ikobacustom.skeleton.testing import FunctionalLayer


class DocumentImportExportTest(DocumentImportExportSetup):

    layer = FunctionalLayer

    def setup_for_export(self):
        document = SkeletonDocument()
        document.document_id = u'DOC1'
        document.title = u'My first document'
        self.app['documents'][document.document_id] = self.document = document
        self.outfile = os.path.join(self.workdir, 'myoutput.csv')
        return

    def test_export_reimport_all(self):
        # we can export all documents in a portal
        # set values we can expect in export file
        self.setup_for_export()
        exporter = SkeletonDocumentExporter()
        exporter.export_all(self.app, self.outfile)
        result = open(self.outfile, 'rb').read()
        self.assertEqual(result,
            'description,document_id,title,users_with_local_roles\r\n'
            ',DOC1,My first document,[]\r\n')
        # We can import the same file.
        processor = SkeletonDocumentProcessor()
        result = processor.doImport(
            self.outfile,
            ['description','document_id','title','users_with_local_roles'],
            mode='create')
        num, num_fail, finished_path, failed_path = result
        #content = open(failed_path).read()
        # The object exists.
        self.assertEqual(num_fail,1)
        # We remove the original document.
        del self.app['documents']['DOC1']
        result = processor.doImport(
            self.outfile,
            ['description','document_id','title','users_with_local_roles'],
            mode='create')
        num_succ, num_fail, finished_path, failed_path = result
        self.assertEqual(num_fail,1)
        # ... only if we add column
        open(self.outfile, 'wb').write(
            'description,document_id,title,users_with_local_roles,class_name\r\n'
            ',DOC1,My first title,[],SkeletonDocument\r\n')
        result = processor.doImport(
            self.outfile,
            ['description','document_id','title','users_with_local_roles','class_name'],
            mode='create')
        num_succ, num_fail, finished_path, failed_path = result
        self.assertEqual(num_fail,0)
        # We can import the same file in update mode.
        processor = SkeletonDocumentProcessor()
        result = processor.doImport(
            self.outfile,
            ['description','document_id','title','users_with_local_roles'],
            mode='update')
        num_succ, num_fail, finished_path, failed_path = result
        self.assertEqual(num_succ,1)
        self.assertEqual(num_fail,0)
        return