def test_query_row_table_inventory(monkeypatch): row_table = inventory.RowTableInventory("invtesttable", ".foo.bar:") monkeypatch.setattr(row_table, "_get_raw_data", lambda only_sites, query: RAW_ROWS) monkeypatch.setattr(row_table, "_get_inv_data", lambda hostrow: INV_ROWS) rows, len_rows = row_table.query(None, [], None, None, None, []) for row in rows: assert set(row) == set(EXPECTED_INV_KEYS)
def test_query_row_table_inventory_add_columns(monkeypatch): row_table = inventory.RowTableInventory("invtesttable", ".foo.bar:") 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_ROWS) rows, _len_rows = row_table.query(view, ['host_foo'], "", None, None, []) for row in rows: assert set(row) == set(EXPECTED_INV_KEYS + ['host_foo'])