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

#Name        : test_bomtofile.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:

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

import tempfile

from shutil import rmtree

import logging

import unittest

from bomtofile import BOMWriter

_logger = logging.getLogger('test.bomtofile')

logging.basicConfig(level=logging.INFO)

class BomtofileTest(unittest.TestCase):

    """Verifiying bomtofile module"""

    def test_writeprojects(self):

        """Verifying writeprojects"""

        tmpdir = tempfile.mkdtemp()

        testProject = _emulateProject(("testProject1", "testProject2"))

        bomWriter = BOMWriter(None, "PythonUnitTesting", testProject, tmpdir)

        bomWriter.writeprojects()

        myFile = open(tmpdir+"/PythonUnitTesting_projects.txt",'r')

        contents = myFile.readlines()

        myFile.close()

        rmtree(tmpdir)

        assert len(contents) == 2

    def test_writebaselines(self):

        """Verifying writebaselines"""

        tmpdir = tempfile.mkdtemp()

        testProject = _emulateProject(("testProject1", "testProject2"))

        testSession = _emulateSession("testbaseline1")

        bomWriter = BOMWriter(testSession, "PythonUnitTesting", testProject, tmpdir)

        bomWriter.writebaselines()

        myFile = open(tmpdir+"/PythonUnitTesting_baselines.txt",'r')

        contents = myFile.readlines()

        myFile.close()

        rmtree(tmpdir)

        assert len(contents) == 4

    def test_writebaselines_withnodata(self):

        """Verifying writebaselines with no baseline"""

        tmpdir = tempfile.mkdtemp()

        testProject = _emulateProject(("testProject1", "testProject2"))

        testSession = _emulateSession(None)

        bomWriter = BOMWriter(testSession, "PythonUnitTesting", testProject, tmpdir)

        bomWriter.writebaselines()

        myFile = open(tmpdir+"/PythonUnitTesting_baselines.txt",'r')

        contents = myFile.readlines()

        myFile.close()

        rmtree(tmpdir)

        assert len(contents) == 4

    def test_writetasks(self):

        """Verifying writetasks"""

        tmpdir = tempfile.mkdtemp()

        testProject = _emulateProject(None, ("testTask1", "testTask2"))

        bomWriter = BOMWriter(None, "PythonUnitTesting", testProject, tmpdir)

        bomWriter.writetasks()

        myFile = open(tmpdir+"/PythonUnitTesting_tasks.txt",'r')

        contents = myFile.readlines()

        myFile.close()

        rmtree(tmpdir)

        assert len(contents) == 2

class _emulateProject():

    """Emulate a project"""

    def __init__(self, project = None, task = None):

        self.baseline = project

        self.task = task

# just emulate amara xml_properties dictionary, as the idea is 

# not to test xml file processing with amara 

        self.xml_properties = { 'task': None } 

class _emulateSession():

    """Emulate a session"""

    def __init__(self, baseline):

        self.baseline = baseline

    def create(self, project):

        return _emulateProject(self.baseline)