##// END OF EJS Templates
pyro4: Add the custom header `X-RhodeCode-Backend` to the pyro4 backend responses....
pyro4: Add the custom header `X-RhodeCode-Backend` to the pyro4 backend responses. This custom header sets the SCM backend which is in use. It is used to identify VCS responses in the error handler. VCS responses are skipped during error handling.

File last commit:

r1:854a839a default
r848:2956cade default
Show More
test_ext_json.py
166 lines | 4.7 KiB | text/x-python | PythonLexer
project: added all source files and assets
r1 # -*- coding: utf-8 -*-
# Copyright (C) 2010-2016 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/
import datetime
import decimal
import io
import textwrap
import pytest
from rhodecode.lib.ext_json import json
from rhodecode.lib.ext_json import formatted_json
from pylons.i18n.translation import _, ungettext
class Timezone(datetime.tzinfo):
def __init__(self, hours):
self.hours = hours
def utcoffset(self, unused_dt):
return datetime.timedelta(hours=self.hours)
def test_dumps_set():
result = json.dumps(set((1, 2, 3)))
# We cannot infer what the order of result is going to be
result = json.loads(result)
assert isinstance(result, list)
assert [1, 2, 3] == sorted(result)
def test_dumps_decimal():
assert '"1.5"' == json.dumps(decimal.Decimal('1.5'))
def test_dumps_complex():
assert "[0.0, 1.0]" == json.dumps(1j)
assert "[1.0, 0.0]" == json.dumps(1 + 0j)
assert "[1.1, 1.2]" == json.dumps(1.1 + 1.2j)
def test_dumps_object_with_json_method():
class SerializableObject(object):
def __json__(self):
return 'foo'
assert '"foo"' == json.dumps(SerializableObject())
def test_dumps_object_with_json_attribute():
class SerializableObject(object):
__json__ = 'foo'
assert '"foo"' == json.dumps(SerializableObject())
def test_dumps_time():
assert '"03:14:15.926"' == json.dumps(datetime.time(3, 14, 15, 926535))
def test_dumps_time_no_microseconds():
assert '"03:14:15"' == json.dumps(datetime.time(3, 14, 15))
def test_dumps_time_with_timezone():
with pytest.raises(TypeError) as excinfo:
json.dumps(datetime.time(3, 14, 15, 926535, Timezone(0)))
error_msg = str(excinfo.value)
assert 'Time-zone aware times are not JSON serializable' in error_msg
def test_dumps_date():
assert '"1969-07-20"' == json.dumps(datetime.date(1969, 7, 20))
def test_dumps_datetime():
json_data = json.dumps(datetime.datetime(1969, 7, 20, 3, 14, 15, 926535))
assert '"1969-07-20T03:14:15.926"' == json_data
def test_dumps_datetime_no_microseconds():
json_data = json.dumps(datetime.datetime(1969, 7, 20, 3, 14, 15))
assert '"1969-07-20T03:14:15"' == json_data
def test_dumps_datetime_with_utc_timezone():
json_data = json.dumps(
datetime.datetime(1969, 7, 20, 3, 14, 15, 926535, Timezone(0)))
assert '"1969-07-20T03:14:15.926Z"' == json_data
def test_dumps_datetime_with_plus1_timezone():
json_data = json.dumps(
datetime.datetime(1969, 7, 20, 3, 14, 15, 926535, Timezone(1)))
assert '"1969-07-20T03:14:15.926+01:00"' == json_data
def test_dumps_unserializable_class():
unserializable_obj = object()
with pytest.raises(TypeError) as excinfo:
json.dumps(unserializable_obj)
assert repr(unserializable_obj) in str(excinfo.value)
assert 'is not JSON serializable' in str(excinfo.value)
def test_dump_is_like_dumps():
data = {
'decimal': decimal.Decimal('1.5'),
'set': set([1]), # Just one element to guarantee the order
'complex': 1 - 1j,
'datetime': datetime.datetime(1969, 7, 20, 3, 14, 15, 926535),
'time': datetime.time(3, 14, 15, 926535),
'date': datetime.date(1969, 7, 20),
}
json_buffer = io.BytesIO()
json.dump(data, json_buffer)
assert json.dumps(data) == json_buffer.getvalue()
def test_formatted_json():
data = {
'b': {'2': 2, '1': 1},
'a': {'3': 3, '4': 4},
}
expected_data = textwrap.dedent('''
{
"a": {
"3": 3,
"4": 4
},
"b": {
"1": 1,
"2": 2
}
}''').strip()
assert formatted_json(data) == expected_data
def test_pylons_lazy_translation_string(pylonsapp):
data = {'label': _('hello')}
data2 = {'label2': ungettext('singular', 'plural', 1)}
assert json.dumps(data) == '{"label": "hello"}'
assert json.dumps(data2) == '{"label2": "singular"}'