def test_query_row_table_inventory_history_add_columns(monkeypatch):
row_table = inventory.RowTableInventoryHistory()
view = View("", {}, {})
monkeypatch.setattr(row_table, "_get_raw_data", lambda only_sites, query: RAW_ROWS2)
monkeypatch.setattr(row_table, "_get_inv_data", lambda hostrow: INV_HIST_ROWS)
rows, _len_rows = row_table.query(view, ['host_foo'], "", None, None, [])
for row in rows:
assert set(row) == set(EXPECTED_INV_HIST_KEYS + ['host_foo'])
def test_query_row_table_inventory_history_unknown_columns(monkeypatch):
row_table = inventory.RowTableInventoryHistory()
monkeypatch.setattr(row_table, "_get_raw_data",
lambda only_sites, query: RAW_ROWS)
monkeypatch.setattr(row_table, "_get_inv_data",
lambda hostrow: INV_HIST_ROWS)
rows, len_rows = row_table.query(None, ['foo'], None, None, None, [])
for row in rows:
assert set(row) == set(EXPECTED_INV_HIST_KEYS)