def test_row2dict_called(self):
"""
Verify that audit.row2dict is called when some element returned by
the searchQuery is no dictionary
"""
from linotp.lib.audit.iterator import AuditQuery
audit = MagicMock(spec=["searchQuery", "row2dict"])
audit.searchQuery.return_value = [None, {'key': 'value'}]
audit_query = AuditQuery({}, audit)
audit_iterator = iter(audit_query.get_query_result())
rows = 0
for row in audit_iterator:
audit_query.get_entry(row)
rows = rows + 1
assert rows == 2
audit.searchQuery.assert_called_once_with(
{},
rp_dict={
'sortname': None,
'sortorder': 'asc'
}
)
audit.row2dict.assert_called_once_with(None)
return
def test_row2dict_called(self):
"""
Verify that audit.row2dict is called when some element returned by
the searchQuery is no dictionary
"""
from linotp.lib.audit.iterator import AuditQuery
audit = MagicMock(spec=["searchQuery", "row2dict"])
audit.searchQuery.return_value = [None, {'key': 'value'}]
audit_query = AuditQuery({}, audit)
audit_iterator = iter(audit_query.get_query_result())
rows = 0
for row in audit_iterator:
audit_query.get_entry(row)
rows = rows + 1
self.assertEqual(rows, 2)
audit.searchQuery.assert_called_once_with(
{},
rp_dict={
'sortname': None,
'sortorder': None
}
)
audit.row2dict.assert_called_once_with(None)
return
