#

# 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 "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: 

#

"""

Test the CPF root file parsing routines

"""

import zipfile

import unittest

import string

import sys,os

ROOT_PATH = os.path.dirname(os.path.abspath(__file__))

import __init__

from cone.public.exceptions import NotResource, StorageException

from cone.public import api

from cone.storage import filestorage

class TestFileResource(unittest.TestCase):    

    def setUp(self):

        self.spath = os.path.join(ROOT_PATH,"fileres_test")

        self.storage = filestorage.FileStorage(self.spath)

    def tearDown(self):

        self.storage.close()

    def test_open_and_write_resource(self):

        res = self.storage.open_resource("testwrite.txt","w")

        res.write("Testing writing func!")

        res.close()

        f = open(self.spath+"/testwrite.txt","r")

        data = f.read()

        self.assertEquals(data,"Testing writing func!")

        f.close()

        os.unlink(os.path.join(ROOT_PATH,"fileres_test/testwrite.txt"))

    def test_open_and_read_resource(self):

        res = self.storage.open_resource("testread.txt","r")

        resdata = res.read()

        res.close()

        f = open(self.spath+"/testread.txt","r")

        data = f.read()

        f.close()

        self.assertEquals(data,resdata)  

    def test_open_and_write_resource_and_trunk_it(self):

        res = self.storage.open_resource("testtrunk.txt","w")

        res.write("Testing writing func!")

        res.truncate()

        res.close()

        f = open(self.spath+"/testtrunk.txt","r")

        data = f.read()

        self.assertEquals(data,"")

        f.close()

        os.unlink(os.path.join(ROOT_PATH,"fileres_test/testtrunk.txt"))

    def test_get_size(self):

        res = self.storage.open_resource("testread.txt", "r")

        self.assertEquals(res.get_size(), 15)

        # Try a second time just in case

        self.assertEquals(res.get_size(), 15)

        res.close()

    def test_get_size_largefile(self):

        res = self.storage.open_resource("largefile.bin", "r")

        self.assertEquals(res.get_size(), 25000)

        # Try a second time just in case

        self.assertEquals(res.get_size(), 25000)

        res.close()

    def test_get_size_fails_in_write_mode(self):

        res = self.storage.open_resource("test_getsize.txt","w")

        res.write("Writing foobar")

        self.assertRaises(StorageException, res.get_size)

        res.close()

    def test_bmp_content_info_24bit(self):

        res = self.storage.open_resource("image-bmp-24bit.bmp", "r")

        content_info = res.get_content_info()

        self.assertEquals(api.BmpImageContentInfo, type(content_info))

        self.assertEquals(24, res.get_content_info().color_depth)

        self.assertEquals(24, content_info.color_depth)

        self.assertEquals('image', content_info.mimetype)

        self.assertEquals('bmp', content_info.mimesubtype)

        self.assertEquals('image/bmp', content_info.content_type)

        res.close()

    def test_bmp_content_info_8bit(self):

        res = self.storage.open_resource("image-bmp-8bit.bmp", "r")

        content_info = res.get_content_info()

        self.assertEquals(api.BmpImageContentInfo, type(content_info))

        self.assertEquals(8, res.get_content_info().color_depth)

        self.assertEquals(8, content_info.color_depth)

        self.assertEquals('image', content_info.mimetype)

        self.assertEquals('bmp', content_info.mimesubtype)

        self.assertEquals('image/bmp', content_info.content_type)

        res.close()

    def test_bmp_content_info_4bit(self):

        res = self.storage.open_resource("image-bmp-4bit.bmp", "r")

        content_info = res.get_content_info()

        self.assertEquals(api.BmpImageContentInfo, type(content_info))

        self.assertEquals(4, res.get_content_info().color_depth)

        self.assertEquals(4, content_info.color_depth)

        self.assertEquals('image', content_info.mimetype)

        self.assertEquals('bmp', content_info.mimesubtype)

        self.assertEquals('image/bmp', content_info.content_type)

        res.close()

    def test_bmp_content_info_1bit(self):

        res = self.storage.open_resource("image-bmp-1bit.bmp", "r")

        content_info = res.get_content_info()

        self.assertEquals(api.BmpImageContentInfo, type(content_info))

        self.assertEquals(1, res.get_content_info().color_depth)

        self.assertEquals(1, content_info.color_depth)

        self.assertEquals('image', content_info.mimetype)

        self.assertEquals('bmp', content_info.mimesubtype)

        self.assertEquals('image/bmp', content_info.content_type)

        res.close()

    def test_content_info_with_jpg(self):

        res = self.storage.open_resource("image-jpeg-1bit.jpg", "r")

        content_info = res.get_content_info()

        self.assertEquals('image/jpeg', content_info.content_type)

        self.assertEquals('image', content_info.mimetype)

        self.assertEquals('jpeg', content_info.mimesubtype)

        res.close()

    def test_content_info_invalid_bmp(self):

        res = self.storage.open_resource("invalid.bmp", "r")

        content_info = res.get_content_info()

        self.assertEquals('image/bmp', content_info.content_type)

        self.assertEquals('image', content_info.mimetype)

        self.assertEquals('bmp', content_info.mimesubtype)

        res.close()

    def test_get_content_info_and_read_data(self):

        res = self.storage.open_resource("testread.txt", "r")

        ci = res.get_content_info()

        self.assertEquals('text/plain', ci.content_type)

        data = res.read()

        self.assertEquals('foo bar test.\n', data)

        res.close()

if __name__ == '__main__':

      unittest.main()