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
|
from pymysql.optionfile import Parser
from unittest import TestCase
from pymysql._compat import PY2
try:
from cStringIO import StringIO
except ImportError:
from io import StringIO
__all__ = ['TestParser']
_cfg_file = (r"""
[default]
string = foo
quoted = "bar"
single_quoted = 'foobar'
""")
class TestParser(TestCase):
def test_string(self):
parser = Parser()
if PY2:
parser.readfp(StringIO(_cfg_file))
else:
parser.read_file(StringIO(_cfg_file))
self.assertEqual(parser.get("default", "string"), "foo")
self.assertEqual(parser.get("default", "quoted"), "bar")
self.assertEqual(parser.get("default", "single_quoted"), "foobar")
|
