tests/krb5: Add check_reply() method to check for AS or TGS reply