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
|
# testing: to be called by nosetests
import os
from ast import literal_eval
from stetl.etl import ETL
from stetl.util import ConfigSection, Util
from tests.stetl_test_case import StetlTestCase
class UtilTest(StetlTestCase):
"""Basic util tests"""
def setUp(self):
super(UtilTest, self).setUp()
def test_configsection_to_string(self):
cfg = {
'name': 'Stetl',
'password': 'something',
'paswoord': 'iets',
'token': 'abc123',
'user': 'John',
'username': 'Jane',
'gebruiker': 'Jan',
'ogrconn': 'PG:dbname=mydb host=myhost port=myport user=myuser password=mypassword active_schema=myschema',
'ogrconn_singlequotes': 'PG:dbname=\'mydb\' host=\'myhost\' port=\'myport\' user=\'myuser\' password=\'mypassword\' active_schema=\'myschema\'',
'ogrconn_doublequotes': 'PG:dbname="mydb" host="myhost" port="myport" user="myuser" password="mypassword" active_schema="myschema"',
'ogrconn_crazypwd1': 'PG:dbname=\'mydb\' host=\'myhost\' port=\'myport\' user=\'myuser\' password=\'my\\\'crazy\\"password\' active_schema=\'myschema\'',
'ogrconn_crazypwd2': 'PG:dbname="mydb" host="myhost" port="myport" user="myuser" password="my\\\'crazy\\"password" active_schema="myschema"',
'ogrconn_dkk': '"PG:dbname=mydb host=myhost port=myport user=myuser password=mypassword active_schema=myschema"',
}
obj = literal_eval(ConfigSection(cfg).to_string())
self.assertEqual('Stetl', obj['name'])
self.assertEqual('<hidden>', obj['password'])
self.assertEqual('<hidden>', obj['paswoord'])
self.assertEqual('<hidden>', obj['token'])
self.assertEqual('<hidden>', obj['user'])
self.assertEqual('<hidden>', obj['username'])
self.assertEqual('Jan', obj['gebruiker'])
self.assertEqual('PG:dbname=mydb host=myhost port=myport user=<hidden> password=<hidden> active_schema=myschema', obj['ogrconn'])
self.assertEqual('PG:dbname=\'mydb\' host=\'myhost\' port=\'myport\' user=\'<hidden>\' password=\'<hidden>\' active_schema=\'myschema\'', obj['ogrconn_singlequotes'])
self.assertEqual('PG:dbname="mydb" host="myhost" port="myport" user="<hidden>" password="<hidden>" active_schema="myschema"', obj['ogrconn_doublequotes'])
self.assertEqual('PG:dbname=\'mydb\' host=\'myhost\' port=\'myport\' user=\'<hidden>\' password=\'<hidden>\' active_schema=\'myschema\'', obj['ogrconn_crazypwd1'])
self.assertEqual('PG:dbname="mydb" host="myhost" port="myport" user="<hidden>" password="<hidden>" active_schema="myschema"', obj['ogrconn_crazypwd2'])
self.assertEqual('"PG:dbname=mydb host=myhost port=myport user=<hidden> password=<hidden> active_schema=myschema"', obj['ogrconn_dkk'])
def test_make_file_list_depth_search(self):
# Util.make_file_list
import sys
file_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'data/depth_search_test')
filename_pattern = 'dummy.gml'
# Test with depth_search enabled
depth_search = True
file_list = Util.make_file_list(file_path, None, filename_pattern, depth_search)
self.assertEqual(len(file_list), 2)
# Test with depth_search disabled
depth_search = False
file_list = Util.make_file_list(file_path, None, filename_pattern, depth_search)
self.assertEqual(len(file_list), 1)
|
