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")