def setUpTestData(cls): # create user and guest group guest_group = Group.objects.create(name="guest") user_group = Group.objects.create(name="user") with open( os.path.join(os.path.dirname(__file__), "data", "api", "rdap_override.json"), "r", ) as fh: data = json.load(fh) cls.rdap_63311 = pdbinet.RdapAsn(data) cls.rdap_63311_no_name = pdbinet.RdapAsn(data) cls.rdap_63311_no_name._parse() cls.rdap_63311_no_name._parsed["org_name"] = None cls.rdap_63311_no_name._parsed["name"] = None cls.ticket = {} for ticket_name in [ "asnauto-9000001-org-net-created.txt", "asnauto-9000001-user-granted-ownership.txt", "asnauto-9000002-user-requested-ownership.txt", "asnauto-9000002-affiliated-user-requested-ownership.txt", ]: with open( os.path.join(os.path.dirname(__file__), "data", "deskpro", ticket_name), "r", ) as fh: cls.ticket[ticket_name] = fh.read() cls.base_org = models.Organization.objects.create( name="ASN Automation Tests") for username, email in [ ("user_a", "*****@*****.**"), ("user_b", "*****@*****.**"), ("user_c", "*****@*****.**"), ]: setattr( cls, username, models.User.objects.create_user(username, email, username), ) getattr(cls, username).set_password(username) cls.base_org.usergroup.user_set.add(getattr(cls, username)) user_group.user_set.add(getattr(cls, username))
def get_asn(self, asn): if asn in ASN_RANGE_OVERRIDE: return pdbinet.RdapAsn(self.override_result) elif pdbinet.asn_is_bogon(asn): return RdapLookup_get_asn(self, asn) else: raise pdbinet.RdapNotFoundError()
def get_asn(self, asn): if asn in ASN_RANGE_OVERRIDE: r = pdbinet.RdapAsn(self.override_result) r._parse() r._parsed["name"] = "AS%d" % asn r._parsed["org_name"] = "ORG AS%d" % asn return r else: raise pdbinet.RdapNotFoundError("Test Not Found")