#============================================================================ 

#Name        : test_deltazip.py 

#Part of     : Helium 

#Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

#All rights reserved.

#This component and the accompanying materials are made available

#under the terms of the License "Eclipse Public License v1.0"

#which accompanies this distribution, and is available

#at the URL "http://www.eclipse.org/legal/epl-v10.html".

#

#Initial Contributors:

#Nokia Corporation - initial contribution.

#

#Contributors:

#

#Description:

#===============================================================================

from __future__ import with_statement

import unittest

import delta_zip

import logging

import os

import sys

import tempfile

class DeltaZipTest( unittest.TestCase ):

    """ Unit tests for the delta zip tool. """

    def setUp(self):

        self.cwd_backup = os.getcwd()

        self.logger = logging.getLogger('test.deltazip')

        self.root = os.environ['TEST_DATA']

        self.output = os.path.join(tempfile.gettempdir(), 'deltazip')

        self.output2 = os.path.join(tempfile.gettempdir(), 'deltazip2')

        logging.basicConfig(level=logging.INFO)

    def test_MD5SignatureBuilder(self):

        output = os.path.join(self.output2, 'md5_list.txt')

        md5output = os.path.join(self.output2, 'delta.md5')

        if os.path.exists(output):

            os.remove(output)

        sig = delta_zip.MD5SignatureBuilderEBS(self.root, 1, self.output2, '', output)

        sig.write_build_file()

        if sys.platform == 'win32':

            assert os.path.splitdrive(self.root)[0] + os.sep not in open(output).read()

        assert os.path.exists(output)

    def test_DeltaZipBuilder(self):

        if not os.path.exists(self.output):

            os.mkdir(self.output)

        md5output = os.path.join(self.output, 'delta.md5')

        oldmd5output = os.path.join(self.output, 'olddelta.md5')

        thisfile = os.path.abspath(__file__)

        md5string = """

Host:fasym014

Username:ssteiner

Date-Time:Fri Aug 17 08:47:40 2007

Version:0.02

Directory:z:\

FileList:z:\output/delta_zip\list_files.txt

Exclusion(s):

Inclusion(s):

----------------

%s TYPE=unknown format MD5=34dcda0d351c75e4942b55e1b2e2422f

        """ % thisfile

        tempoutput = open(md5output, 'w')

        tempoutput.write(md5string)

        tempoutput.close()

        md5string = """

Host:fasym014

Username:ssteiner

Date-Time:Fri Aug 17 08:47:40 2007

Version:0.02

Directory:z:\

FileList:z:\output/delta_zip\list_files.txt

Exclusion(s):

Inclusion(s):

----------------

%s TYPE=unknown format MD5=34dcda0d351c75e4942b55e1b2e2422g

        """ % thisfile

        tempoutput = open(oldmd5output, 'w')

        tempoutput.write(md5string)

        tempoutput.close()

        deltazipfile = os.path.join(self.output, 'delta.zip')

        deltaantfile = os.path.join(self.output, 'delta.ant.xml')

        deletefile = os.path.join(self.output, 'delta_zip_specialInstructions.xml')

        delta = delta_zip.DeltaZipBuilder(self.root, self.output, oldmd5output, md5output)

        delta.create_delta_zip(deltazipfile, deletefile, 1, deltaantfile)

    def test_changedFiles(self):

        dir1 = tempfile.mkdtemp()

        dir2 = tempfile.mkdtemp()

        with open(os.path.join(dir1, '1'), 'w') as f1:

            f1.write('Directory:%s\n' % self.root)

            f1.write('myfile1 TYPE=unknown format MD5=34dcda0d351c75e4942b55e1b2e2422g')

        with open(os.path.join(dir2, '2'), 'w') as f2:

            f2.write('Directory:%s\n' % self.root)

            f2.write('myfile1 TYPE=unknown format MD5=34dcda0d351c75e4542b55e1b2e2422g')

        assert delta_zip.changedFiles(dir1, dir2) == [os.path.join(self.root, 'myfile1')]

    def tearDown(self):

        """ Restore path """

        os.chdir(self.cwd_backup)