def test_empty_extra():
formatter = AppendJSONFormatter(limit_keys_to=('foo',))
record = get_record(extra={'bar': 'no'})
output = formatter.format(record)
# empty json should not lead to excess whitespace at the end of the line
assert not output.endswith(' ')
assert formatter.get_json(record) == ''
def test_limit_keys_to():
formatter = AppendJSONFormatter(limit_keys_to=('foo', 'bar'))
record = get_record(extra={'foo': 'yes', 'bar': 'yes', 'baz': 'no'})
output = formatter.format(record)
assert output.count('yes') == 2
assert output.count('no') == 0
data = json.loads(formatter.get_json(record))
assert data['foo'] == 'yes'
assert data['bar'] == 'yes'
assert 'baz' not in data
