def test_simple(self):
"""
Ensures we can filter a simple dict
"""
json_grep = JSONGrep(['user'])
source = {'userID': 5, 'url' : 'http://www.google.com'}
result = json_grep.jgrep(encode(source))
self.assertEquals(result, {'userID': 5})
def test_file_jgrep(self):
"""
Ensures we can jgrep a file
"""
tmp_file = TemporaryFile('w+')
tmp_file.write('%s\n' % encode({'user' : 'john', 'ip' : '192.168.2.2', 'city' : 'philadelphia'}))
tmp_file.write('%s\n' % encode({'user' : 'jill', 'ip' : '10.0.0.1', 'city' : 'stockholm'}))
tmp_file.seek(0)
expected_output = [
{'user' : 'john', 'ip' : '192.168.2.2'},
{'user' : 'jill', 'ip' : '10.0.0.1'}]
json_grep = JSONGrep(['user', 'ip'])
output = [line for line in json_grep.jgrep_file(tmp_file)]
self.assertEquals(expected_output, output)