Untitled

ADD_DNAT_RULE_PAYLOAD = {
    "name": "TEST DNAT RULE AZ1",
    "description": "Test dnat source nat",
    "from_zone": ["INTERNET"],
    "to_zone": ["TRANSIT"],
    "src_addrs": ["APPLICATIE-DIENSTEN-JH-P-AZ1-100.116.0.64_26"],
    "dst_addrs": ["APPLICATIE-FT-TENANT-01-USERFOR-USERFOR-AZ1-100.116.2.64_26"],
    "service": "tcp-8080",
    "src_translation_addr": "APPLICATIE-PICARD-12-TEST-RELEASE-12-TEST-AZ1-100.116.14.64_26",
    "tags": ["AZ1"],
    "group_tag": "AZ1",
    "location": "device-group",
    "device_group": "INTERNET",
    "nat_type": "dnat",
    "dst_port": 8080,
}


@pytest.mark.parametrize("nat_type", ["bidirectional", "snat", "dnat"])
@patch.object(PaloAltoClient, "_login", new=mock_palo_alto_login)
def test_add_rule_happy(nat_type):
    task = AddNatRuleTask()
    if nat_type == "dnat":
        rule_payload = ADD_DNAT_RULE_PAYLOAD
    else:
        rule_payload = ADD_RULE_PAYLOAD
    payload = {
        "session": SESSION,
        "params": deepcopy(rule_payload),
    }
    payload["params"]["nat_type"] = nat_type
    response = task.run(payload=payload)
    assert response.status.success
    assert response.session
    assert "Nat rule added" in response.status.message