def test_host_inventory_set_with_inventory_mode(conn_args, mock_login): """ query_submitted = {"params": {"hostid": 10258, "inventory_mode": "1", "inventory": {"asset_tag": "jml3322", "type": "Xen"}}, "jsonrpc": "2.0", "id": 0, "auth": "a50d2c3030b9b73d7c28b5ebd89c044c", "method": "host.update"} """ module_return = {"hostids": [10258]} query_return = {"jsonrpc": "2.0", "result": {"hostids": [10258]}, "id": 0} with patch.object( zabbix, "_query", autospec=True, return_value=query_return ) as mock__query: assert ( zabbix.host_inventory_set( 10258, asset_tag="jml3322", type="Xen", inventory_mode="1", **conn_args ) == module_return ) mock__query.assert_called_with( "host.update", { "hostid": 10258, "inventory_mode": "1", "inventory": { "asset_tag": "jml3322", "type": "Xen", "url": "http://test.url", "auth": "1234", }, }, "http://test.url", "1234", )
def test_host_inventory_set(self): """ query_submitted = {"params": {"hostid": 10258, "inventory_mode": "0", "inventory": {"asset_tag": "jml3322", "type": "Xen"}}, "jsonrpc": "2.0", "id": 0, "auth": "a50d2c3030b9b73d7c28b5ebd89c044c", "method": "host.update"} """ module_return = {"hostids": [10258]} query_return = { "jsonrpc": "2.0", "result": { "hostids": [10258] }, "id": 0 } with patch.object(zabbix, "_query", return_value=query_return): with patch.object(zabbix, "_login", return_value=CONN_ARGS): self.assertEqual( zabbix.host_inventory_set(10258, asset_tag="jml3322", type="Xen", **CONN_ARGS), module_return, )