def testOutputRemovedColumn(self):
with patch('sys.stdout', new=StringIO()) as fake_out:
self.parsed_args.columns = 'key2'
_output(self.parsed_args, LISTOFDICTS)
self.assertNotIn('key1', fake_out.getvalue())
def testOutputJsonStdout(self):
with patch('sys.stdout', new=StringIO()) as fake_out:
_output(self.parsed_args, LISTOFDICTS)
assert json.loads(fake_out.getvalue()) == LISTOFDICTS
def testOutputNonList(self):
with patch('sys.stdout', new=StringIO()) as fake_out:
_output(self.parsed_args, LISTOFDICTS[0])
stdout = fake_out.getvalue()
self.assertIn('key1', stdout)
def testOutputEmptyResult(self):
with self.assertLogs(level='INFO'):
output = _output(self.parsed_args, [])
assert output is None