#!/usr/bin/python
#
# CDDL HEADER START
#
# The contents of this file are subject to the terms of the
# Common Development and Distribution License (the "License").
# You may not use this file except in compliance with the License.
#
# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
# or http://www.opensolaris.org/os/licensing.
# See the License for the specific language governing permissions
# and limitations under the License.
#
# When distributing Covered Code, include this CDDL HEADER in each
# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
# If applicable, add the following below this CDDL HEADER, with the
# fields enclosed by brackets "[]" replaced with your own identifying
# information: Portions Copyright [yyyy] [name of copyright owner]
#
# CDDL HEADER END
#

# Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
# Use is subject to license terms.

import unittest
import os
import shutil
import sys
import tempfile
import pkg.client.imageconfig as imageconfig

# Set the path so that modules above can be found
path_to_parent = os.path.join(os.path.dirname(__file__), "..")
sys.path.insert(0, path_to_parent)
import pkg5unittest

class TestImageConfig(pkg5unittest.Pkg5TestCase):
        def setUp(self):
                self.sample_dir = tempfile.mkdtemp()
                cfgfile = os.path.join(self.sample_dir, imageconfig.CFG_FILE)
                f = open(cfgfile, "w")

                f.write("""\
[policy]
Display-Copyrights: False

[property]
name = an image
                
[authority_sfbay.sun.com]
alias: zruty
prefix: sfbay.sun.com
origin: http://zruty.sfbay:10001/
mirrors:
ssl_key:
ssl_cert:
repo.collection_type: supplemental
repo.description: Lots of development packages here.
repo.legal_uris: ['http://zruty.sfbay:10001/legal.html', 'http://zruty.sfbay:10001/tos.html']
repo.name: zruty development repository
repo.refresh_seconds: 86400
repo.registered: True
repo.registration_uri: http://zruty.sfbay:10001/reg.html
repo.related_uris:
sort_policy: priority
""")
                f.close()
                self.ic = imageconfig.ImageConfig(self.sample_dir, "publisher")

        def tearDown(self):
                try:
                        shutil.rmtree(self.sample_dir)
                except:
                        pass

        def test_0_read(self):
                """Verify that read works and that values are read properly."""
                self.ic.read(self.sample_dir)

                pub = self.ic.publishers["sfbay.sun.com"]
                self.assertEqual(pub.alias, "zruty")
                repo = pub.selected_repository
                origin = repo.origins[0]
                self.assertEqual(origin.uri, "http://zruty.sfbay:10001/")
                self.assertEqual(origin.ssl_key, None)
                self.assertEqual(origin.ssl_cert, None)
                self.assertEqual(repo.collection_type, "supplemental")
                self.assertEqual(repo.description,
                    "Lots of development packages here.")
                self.assertEqual([u.uri for u in repo.legal_uris],
                    ["http://zruty.sfbay:10001/legal.html",
                    "http://zruty.sfbay:10001/tos.html"])
                self.assertEqual(repo.name, "zruty development repository")
                self.assertEqual(repo.refresh_seconds, 86400)
                self.assertEqual(repo.registered, True)
                self.assertEqual(repo.registration_uri, "http://zruty.sfbay:10001/reg.html")
                self.assertEqual(repo.related_uris, [])
                self.assertEqual(repo.sort_policy, "priority")
                # uuid should have been set even though it wasn't in the file
                self.assertNotEqual(pub.client_uuid, None)

        def test_1_unicode(self):
                self.ic.read(self.sample_dir)
                ustr = u'abc\u3041def'
                self.ic.properties['name'] = ustr
                newdir = tempfile.mkdtemp()
                self.ic.write(newdir)
                ic2 = imageconfig.ImageConfig(newdir, "publisher")
                ic2.read(newdir)
                ustr2 = ic2.properties['name']
                shutil.rmtree(newdir)
                self.assert_(ustr == ustr2)

        def test_2_missing_conffile(self):
                #
                #  See what happens if the conf file is missing.
                #
                shutil.rmtree(self.sample_dir)
                self.assertRaises(RuntimeError, self.ic.read, self.sample_dir)

        def test_3_reread(self):
                """Verify that the uuid determined during the first read is the
                same as the uuid in the second read."""
                self.ic.read(self.sample_dir)
                pub = self.ic.publishers["sfbay.sun.com"]
                uuid = pub.client_uuid

                ic2 = imageconfig.ImageConfig(self.sample_dir, "publisher")
                ic2.read(self.sample_dir)
                pub2 = ic2.publishers["sfbay.sun.com"]
                self.assertEqual(pub2.client_uuid, uuid)


# XXX more test cases needed.

if __name__ == "__main__":
        unittest.main()