|
@@ -20,7 +20,7 @@ import datetime
|
|
|
|
|
|
|
|
from nose.tools import assert_true, assert_equal, assert_not_equal
|
|
from nose.tools import assert_true, assert_equal, assert_not_equal
|
|
|
|
|
|
|
|
-from desktop.lib.python_util import CaseInsensitiveDict, force_dict_to_strings
|
|
|
|
|
|
|
+from desktop.lib.python_util import CaseInsensitiveDict, force_dict_to_strings, force_list_to_strings
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestPythonUtil(object):
|
|
class TestPythonUtil(object):
|
|
@@ -40,7 +40,20 @@ class TestPythonUtil(object):
|
|
|
transformed_dict = force_dict_to_strings(unicode_dict)
|
|
transformed_dict = force_dict_to_strings(unicode_dict)
|
|
|
assert_equal(string_dict, transformed_dict)
|
|
assert_equal(string_dict, transformed_dict)
|
|
|
|
|
|
|
|
|
|
+ # Embedded
|
|
|
unicode_dict = {u'test': {u'test': u'test'}}
|
|
unicode_dict = {u'test': {u'test': u'test'}}
|
|
|
string_dict = {'test': {'test': 'test'}}
|
|
string_dict = {'test': {'test': 'test'}}
|
|
|
transformed_dict = force_dict_to_strings(unicode_dict)
|
|
transformed_dict = force_dict_to_strings(unicode_dict)
|
|
|
assert_equal(string_dict, transformed_dict)
|
|
assert_equal(string_dict, transformed_dict)
|
|
|
|
|
+
|
|
|
|
|
+ # Embedded list
|
|
|
|
|
+ unicode_dict = {u'test': [{u'test': u'test'}]}
|
|
|
|
|
+ string_dict = {'test': [{'test': 'test'}]}
|
|
|
|
|
+ transformed_dict = force_dict_to_strings(unicode_dict)
|
|
|
|
|
+ assert_equal(string_dict, transformed_dict)
|
|
|
|
|
+
|
|
|
|
|
+ def test_force_list_to_strings(self):
|
|
|
|
|
+ unicode_list = [u'test', {u'test': u'test'}]
|
|
|
|
|
+ string_list = ['test', {'test': 'test'}]
|
|
|
|
|
+ transformed_list = force_list_to_strings(unicode_list)
|
|
|
|
|
+ assert_equal(string_list, transformed_list)
|