# Copyright 2012 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2

from portage.tests import TestCase
from portage.tests.resolver.ResolverPlayground import ResolverPlayground, ResolverPlaygroundTestCase

class TargetRootTestCase(TestCase):

	def testTargetRoot(self):
		ebuilds = {
			"dev-lang/python-3.2": {
				"EAPI": "5-hdepend",
				"IUSE": "targetroot",
				"HDEPEND": "targetroot? ( ~dev-lang/python-3.2 )",
			},
			"dev-libs/A-1": {
				"EAPI": "4",
				"DEPEND": "dev-libs/B",
				"RDEPEND": "dev-libs/C",
			},
			"dev-libs/B-1": {},
			"dev-libs/C-1": {},
		}

		test_cases = (
			ResolverPlaygroundTestCase(
				["dev-lang/python"],
				options = {},
				success = True,
				mergelist = ["dev-lang/python-3.2", "dev-lang/python-3.2{targetroot}"]),
			ResolverPlaygroundTestCase(
				["dev-lang/python"],
				options = {"--root-deps": True},
				success = True,
				mergelist = ["dev-lang/python-3.2", "dev-lang/python-3.2{targetroot}"]),
			ResolverPlaygroundTestCase(
				["dev-lang/python"],
				options = {"--root-deps": "rdeps"},
				success = True,
				mergelist = ["dev-lang/python-3.2", "dev-lang/python-3.2{targetroot}"]),
			ResolverPlaygroundTestCase(
				["dev-libs/A"],
				options = {},
				ambiguous_merge_order = True,
				success = True,
				mergelist = [("dev-libs/B-1", "dev-libs/C-1{targetroot}"), "dev-libs/A-1{targetroot}"]),
			ResolverPlaygroundTestCase(
				["dev-libs/A"],
				options = {"--root-deps": True},
				ambiguous_merge_order = True,
				success = True,
				mergelist = [("dev-libs/B-1{targetroot}", "dev-libs/C-1{targetroot}"), "dev-libs/A-1{targetroot}"]),
			ResolverPlaygroundTestCase(
				["dev-libs/A"],
				options = {"--root-deps": "rdeps"},
				ambiguous_merge_order = True,
				success = True,
				mergelist = [("dev-libs/C-1{targetroot}"), "dev-libs/A-1{targetroot}"]),
		)

		playground = ResolverPlayground(ebuilds=ebuilds, targetroot=True,
			debug=False)
		try:
			for test_case in test_cases:
				playground.run_TestCase(test_case)
				self.assertEqual(test_case.test_success, True, test_case.fail_msg)
		finally:
			playground.cleanup()

		test_cases = (
			ResolverPlaygroundTestCase(
				["dev-lang/python"],
				options = {},
				success = True,
				mergelist = ["dev-lang/python-3.2"]),
		)

		playground = ResolverPlayground(ebuilds=ebuilds, targetroot=False,
			debug=False)
		try:
			for test_case in test_cases:
				playground.run_TestCase(test_case)
				self.assertEqual(test_case.test_success, True, test_case.fail_msg)
		finally:
			playground.cleanup()