def test_query_row_multi_table_inventory_add_columns(monkeypatch): sources = list(zip(["invtesttable1", "invtesttable2"], [".foo.bar:", "foo.baz:"])) row_table = inventory.RowMultiTableInventory(sources, ["sid"], []) 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_MULTI) rows, _len_rows = row_table.query(view, ['host_foo'], "", None, None, []) for row in rows: assert set(row) == set(EXPECTED_INV_MULTI_KEYS + ['host_foo'])
def test_query_row_multi_table_inventory(monkeypatch): sources = list( zip(["invtesttable1", "invtesttable2"], [".foo.bar:", "foo.baz:"])) row_table = inventory.RowMultiTableInventory(sources, ["sid"], []) monkeypatch.setattr(row_table, "_get_raw_data", lambda only_sites, query: RAW_ROWS) monkeypatch.setattr(row_table, "_get_inv_data", lambda hostrow: INV_ROWS_MULTI) rows = row_table.query(None, [], None, None, None, []) for row in rows: assert set(row) == set(EXPECTED_INV_MULTI_KEYS)