# Marshall: A framework for pluggable marshalling policies
# Copyright (C) 2004-2006 Enfold Systems, LLC
#
# 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
#
"""
$Id: test_export.py 2886 2004-08-25 03:51:04Z dreamcatcher $
"""

if __name__ == '__main__':
    execfile(os.path.join(sys.path[0], 'framework.py'))

import re
import difflib

# Load fixture
from Testing import ZopeTestCase
from Products.Archetypes.tests.atsitetestcase import ATSiteTestCase

from AccessControl.SecurityManagement import newSecurityManager

# Install our product
ZopeTestCase.installProduct('Marshall')
ZopeTestCase.installProduct('ATContentTypes')

portal_owner = 'portal_owner'

def normalize_tabs(s):
    s = re.sub(r"[ \t]+", " ", s)
    return s

def normalize_space(s):
    s =  re.sub(r"[\r\n]+", r'\r\n', s)
    return s

class BaseTest(ATSiteTestCase):
    """Base Test"""

    def loginPortalOwner(self):
        '''Use if - AND ONLY IF - you need to manipulate
        the portal object itself.
        '''
        uf = self.app.acl_users
        user = uf.getUserById(portal_owner)
        if not hasattr(user, 'aq_base'):
            user = user.__of__(uf)
        newSecurityManager(None, user)

    def compare(self, one, two):
        diff = difflib.ndiff(one.splitlines(), two.splitlines())
        diff = '\n'.join(list(diff))
        return diff

    def assertEqualsDiff(self, one, two, normalize=True):
        if normalize:
            one, two = normalize_tabs(one), normalize_tabs(two)
        one, two = normalize_space(one), normalize_space(two)
        self.failUnless(one.splitlines() == two.splitlines(),
                        self.compare(one, two))
