def testAssertTrue(self):
     print("Testing axapi_auth")
     try:
         mitigator_base_url, signature = axapi_auth(a10_tps, username, password)
         print("base url: ", mitigator_base_url, "Signature: ", signature)
         axapi_action(mitigator_base_url+"/axapi/v3/logoff")
     
     except Exception as e:
         self.fail("Not authenticated")
    def testAssertTrue(self):
        try:
            print("Testing GET")
            mitigator_base_url, signature = axapi_auth(a10_tps, username, password)
            r = axapi_action(mitigator_base_url+"/axapi/v3/version/oper", method='GET', signature=signature)
            print(str(r))
            print("Testing POST")
            hostname_payload = {"hostname": {"value": hostname}}
            r = axapi_action(mitigator_base_url+"/axapi/v3/hostname", payload=hostname_payload, signature=signature)
            print(str(r))
            axapi_action(mitigator_base_url+"/axapi/v3/logoff")

        except Exception as e:
            self.fail("Failed")


client_ip_as_string=sys.argv[1]
data_direction=sys.argv[2]
pps_as_string=int(sys.argv[3])
action=sys.argv[4]

logger.info(" - " . join(sys.argv))


# A10 Mitigator Information
mitigator_ip = "192.168.199.152"
zone_name = client_ip_as_string + "_zone"
ip_addr = client_ip_as_string
mitigator_base_url, signature = axapi_auth(mitigator_ip, "admin", "a10")


if action == "unban":
    try: 
        r = axapi_action(mitigator_base_url+ddos_dst_zone_path+zone_name, method="DELETE", signature=signature)
    except Exception as e: 
        logger.info("Zone not removed in unban, may not exist. Result: " + str(e))

    # Commit config
    axapi_action(mitigator_base_url+write_mem_path, signature=signature)
    # Logoff
    axapi_action(mitigator_base_url+logoff_path, signature=signature)
 
    sys.exit(0)