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

from portage.tests import TestCase
from portage.dep import dep_getrepo

class DepGetRepo(TestCase):
	""" A simple testcase for isvalidatom
	"""

	def testDepGetRepo(self):

		repo_char = "::"
		repos = ( "a", "repo-name", "repo_name", "repo123", None )
		cpvs = ["sys-apps/portage"]
		versions = ["2.1.1","2.1-r1", None]
		uses = ["[use]", None]
		for cpv in cpvs:
			for version in versions:
				for use in uses:
					for repo in repos:
						pkg = cpv
						if version:
							pkg = '=' + pkg + '-' + version
						if repo is not None:
							pkg = pkg + repo_char + repo
						if use:
							pkg = pkg + use
						self.assertEqual( dep_getrepo( pkg ), repo )