summaryrefslogtreecommitdiffstats
path: root/pym/portage/tests/unicode/test_string_format.py
blob: fb6e8e02ed6622997820c29d320b5047aae762b8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
# Copyright 2010 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2

import sys

from portage import _encodings, _unicode_decode
from portage.exception import PortageException
from portage.tests import TestCase
from _emerge.DependencyArg import DependencyArg
from _emerge.UseFlagDisplay import UseFlagDisplay

if sys.hexversion >= 0x3000000:
	basestring = str

STR_IS_UNICODE = sys.hexversion >= 0x3000000

class StringFormatTestCase(TestCase):
	"""
	Test that string formatting works correctly in the current interpretter,
	which may be either python2 or python3.
	"""

	# In order to get some unicode test strings in a way that works in
	# both python2 and python3, write them here as byte strings and
	# decode them before use. This assumes _encodings['content'] is
	# utf_8.

	unicode_strings = (
		b'\xE2\x80\x98',
		b'\xE2\x80\x99',
	)

	def testDependencyArg(self):

		self.assertEqual(_encodings['content'], 'utf_8')

		for arg_bytes in self.unicode_strings:
			arg_unicode = _unicode_decode(arg_bytes, encoding=_encodings['content'])
			dependency_arg = DependencyArg(arg=arg_unicode)

			# Force unicode format string so that __unicode__() is
			# called in python2.
			formatted_str = _unicode_decode("%s") % (dependency_arg,)
			self.assertEqual(formatted_str, arg_unicode)

			if STR_IS_UNICODE:

				# Test the __str__ method which returns unicode in python3
				formatted_str = "%s" % (dependency_arg,)
				self.assertEqual(formatted_str, arg_unicode)

			else:

				# Test the __str__ method which returns encoded bytes in python2
				formatted_bytes = "%s" % (dependency_arg,)
				self.assertEqual(formatted_bytes, arg_bytes)

	def testPortageException(self):

		self.assertEqual(_encodings['content'], 'utf_8')

		for arg_bytes in self.unicode_strings:
			arg_unicode = _unicode_decode(arg_bytes, encoding=_encodings['content'])
			e = PortageException(arg_unicode)

			# Force unicode format string so that __unicode__() is
			# called in python2.
			formatted_str = _unicode_decode("%s") % (e,)
			self.assertEqual(formatted_str, arg_unicode)

			if STR_IS_UNICODE:

				# Test the __str__ method which returns unicode in python3
				formatted_str = "%s" % (e,)
				self.assertEqual(formatted_str, arg_unicode)

			else:

				# Test the __str__ method which returns encoded bytes in python2
				formatted_bytes = "%s" % (e,)
				self.assertEqual(formatted_bytes, arg_bytes)

	def testUseFlagDisplay(self):

		self.assertEqual(_encodings['content'], 'utf_8')

		for enabled in (True, False):
			for forced in (True, False):
				for arg_bytes in self.unicode_strings:
					arg_unicode = _unicode_decode(arg_bytes, encoding=_encodings['content'])
					e = UseFlagDisplay(arg_unicode, enabled, forced)

					# Force unicode format string so that __unicode__() is
					# called in python2.
					formatted_str = _unicode_decode("%s") % (e,)
					self.assertEqual(isinstance(formatted_str, basestring), True)

					if STR_IS_UNICODE:

						# Test the __str__ method which returns unicode in python3
						formatted_str = "%s" % (e,)
						self.assertEqual(isinstance(formatted_str, str), True)

					else:

						# Test the __str__ method which returns encoded bytes in python2
						formatted_bytes = "%s" % (e,)
						self.assertEqual(isinstance(formatted_bytes, bytes), True)