def test_update_tagged_item(self): site = Site.objects.create(name='Test Site', slug='test-site') site.tags.add("Foo", "Bar", "Baz") create_tags("New Tag") data = { 'tags': [ { "name": "Foo" }, { "name": "Bar" }, { "name": "New Tag" }, ] } self.add_permissions('dcim.change_site') url = reverse('dcim-api:site-detail', kwargs={'pk': site.pk}) response = self.client.patch(url, data, format='json', **self.header) self.assertHttpStatus(response, status.HTTP_200_OK) self.assertListEqual( sorted([t['name'] for t in response.data['tags']]), sorted([t['name'] for t in data['tags']])) site = Site.objects.get(pk=response.data['id']) self.assertListEqual(sorted([t.name for t in site.tags.all()]), sorted(["Foo", "Bar", "New Tag"]))
def setUpTestData(cls): tags = create_tags('Alpha', 'Bravo', 'Charlie') sites = [Site(name=f'Site {i}', slug=f'site-{i}') for i in range(1, 6)] Site.objects.bulk_create(sites) for site in sites: site.tags.add(*tags)
def setUpTestData(cls): contact_groups = ( ContactGroup(name='Contact Group 1', slug='contact-group-1'), ContactGroup(name='Contact Group 2', slug='contact-group-2'), ContactGroup(name='Contact Group 3', slug='contact-group-3'), ) for tenanantgroup in contact_groups: tenanantgroup.save() tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Contact Group X', 'slug': 'contact-group-x', 'description': 'A new contact group', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug,description", "Contact Group 4,contact-group-4,Fourth contact group", "Contact Group 5,contact-group-5,Fifth contact group", "Contact Group 6,contact-group-6,Sixth contact group", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): contact_groups = ( ContactGroup(name='Contact Group 1', slug='contact-group-1'), ContactGroup(name='Contact Group 2', slug='contact-group-2'), ) for contactgroup in contact_groups: contactgroup.save() Contact.objects.bulk_create([ Contact(name='Contact 1', group=contact_groups[0]), Contact(name='Contact 2', group=contact_groups[0]), Contact(name='Contact 3', group=contact_groups[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Contact X', 'group': contact_groups[1].pk, 'comments': 'Some comments', 'tags': [t.pk for t in tags], } cls.csv_data = ( "group,name", "Contact Group 1,Contact 4", "Contact Group 1,Contact 5", "Contact Group 1,Contact 6", ) cls.bulk_edit_data = { 'group': contact_groups[1].pk, }
def setUpTestData(cls): RIR.objects.bulk_create([ RIR(name='RIR 1', slug='rir-1'), RIR(name='RIR 2', slug='rir-2'), RIR(name='RIR 3', slug='rir-3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'RIR X', 'slug': 'rir-x', 'is_private': True, 'description': 'A new RIR', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug,description", "RIR 4,rir-4,Fourth RIR", "RIR 5,rir-5,Fifth RIR", "RIR 6,rir-6,Sixth RIR", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): groups = ( WirelessLANGroup(name='Wireless LAN Group 1', slug='wireless-lan-group-1'), WirelessLANGroup(name='Wireless LAN Group 2', slug='wireless-lan-group-2'), ) for group in groups: group.save() WirelessLAN.objects.bulk_create([ WirelessLAN(group=groups[0], ssid='WLAN1'), WirelessLAN(group=groups[0], ssid='WLAN2'), WirelessLAN(group=groups[0], ssid='WLAN3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'ssid': 'WLAN2', 'group': groups[1].pk, 'tags': [t.pk for t in tags], } cls.csv_data = ( "group,ssid", "Wireless LAN Group 2,WLAN4", "Wireless LAN Group 2,WLAN5", "Wireless LAN Group 2,WLAN6", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): tenants = ( Tenant(name='Tenant A', slug='tenant-a'), Tenant(name='Tenant B', slug='tenant-b'), ) Tenant.objects.bulk_create(tenants) tags = create_tags('Alpha', 'Bravo', 'Charlie') route_targets = ( RouteTarget(name='65000:1001', tenant=tenants[0]), RouteTarget(name='65000:1002', tenant=tenants[1]), RouteTarget(name='65000:1003'), ) RouteTarget.objects.bulk_create(route_targets) cls.form_data = { 'name': '65000:100', 'description': 'A new route target', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,tenant,description", "65000:1004,Tenant A,Foo", "65000:1005,Tenant B,Bar", "65000:1006,,No tenant", ) cls.bulk_edit_data = { 'tenant': tenants[1].pk, 'description': 'New description', }
def setUpTestData(cls): ClusterGroup.objects.bulk_create([ ClusterGroup(name='Cluster Group 1', slug='cluster-group-1'), ClusterGroup(name='Cluster Group 2', slug='cluster-group-2'), ClusterGroup(name='Cluster Group 3', slug='cluster-group-3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Cluster Group X', 'slug': 'cluster-group-x', 'description': 'A new cluster group', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug,description", "Cluster Group 4,cluster-group-4,Fourth cluster group", "Cluster Group 5,cluster-group-5,Fifth cluster group", "Cluster Group 6,cluster-group-6,Sixth cluster group", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): ClusterType.objects.bulk_create([ ClusterType(name='Cluster Type 1', slug='cluster-type-1'), ClusterType(name='Cluster Type 2', slug='cluster-type-2'), ClusterType(name='Cluster Type 3', slug='cluster-type-3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Cluster Type X', 'slug': 'cluster-type-x', 'description': 'A new cluster type', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug,description", "Cluster Type 4,cluster-type-4,Fourth cluster type", "Cluster Type 5,cluster-type-5,Fifth cluster type", "Cluster Type 6,cluster-type-6,Sixth cluster type", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): groups = ( WirelessLANGroup(name='Wireless LAN Group 1', slug='wireless-lan-group-1'), WirelessLANGroup(name='Wireless LAN Group 2', slug='wireless-lan-group-2'), WirelessLANGroup(name='Wireless LAN Group 3', slug='wireless-lan-group-3'), ) for group in groups: group.save() tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Wireless LAN Group X', 'slug': 'wireless-lan-group-x', 'parent': groups[2].pk, 'description': 'A new wireless LAN group', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug,description", "Wireles sLAN Group 4,wireless-lan-group-4,Fourth wireless LAN group", "Wireless LAN Group 5,wireless-lan-group-5,Fifth wireless LAN group", "Wireless LAN Group 6,wireless-lan-group-6,Sixth wireless LAN group", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): CircuitType.objects.bulk_create([ CircuitType(name='Circuit Type 1', slug='circuit-type-1'), CircuitType(name='Circuit Type 2', slug='circuit-type-2'), CircuitType(name='Circuit Type 3', slug='circuit-type-3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Circuit Type X', 'slug': 'circuit-type-x', 'description': 'A new circuit type', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug", "Circuit Type 4,circuit-type-4", "Circuit Type 5,circuit-type-5", "Circuit Type 6,circuit-type-6", ) cls.bulk_edit_data = { 'description': 'Foo', }
def setUpTestData(cls): ContactRole.objects.bulk_create([ ContactRole(name='Contact Role 1', slug='contact-role-1'), ContactRole(name='Contact Role 2', slug='contact-role-2'), ContactRole(name='Contact Role 3', slug='contact-role-3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Devie Role X', 'slug': 'contact-role-x', 'description': 'New contact role', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug", "Contact Role 4,contact-role-4", "Contact Role 5,contact-role-5", "Contact Role 6,contact-role-6", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): Role.objects.bulk_create([ Role(name='Role 1', slug='role-1'), Role(name='Role 2', slug='role-2'), Role(name='Role 3', slug='role-3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Role X', 'slug': 'role-x', 'weight': 200, 'description': 'A new role', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug,weight", "Role 4,role-4,1000", "Role 5,role-5,1000", "Role 6,role-6,1000", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): providers = ( Provider(name='Provider 1', slug='provider-1', asn=65001), Provider(name='Provider 2', slug='provider-2', asn=65002), ) Provider.objects.bulk_create(providers) circuittypes = ( CircuitType(name='Circuit Type 1', slug='circuit-type-1'), CircuitType(name='Circuit Type 2', slug='circuit-type-2'), ) CircuitType.objects.bulk_create(circuittypes) Circuit.objects.bulk_create([ Circuit(cid='Circuit 1', provider=providers[0], type=circuittypes[0]), Circuit(cid='Circuit 2', provider=providers[0], type=circuittypes[0]), Circuit(cid='Circuit 3', provider=providers[0], type=circuittypes[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'cid': 'Circuit X', 'provider': providers[1].pk, 'type': circuittypes[1].pk, 'status': CircuitStatusChoices.STATUS_DECOMMISSIONED, 'tenant': None, 'install_date': datetime.date(2020, 1, 1), 'commit_rate': 1000, 'description': 'A new circuit', 'comments': 'Some comments', 'tags': [t.pk for t in tags], } cls.csv_data = ( "cid,provider,type,status", "Circuit 4,Provider 1,Circuit Type 1,active", "Circuit 5,Provider 1,Circuit Type 1,active", "Circuit 6,Provider 1,Circuit Type 1,active", ) cls.bulk_edit_data = { 'provider': providers[1].pk, 'type': circuittypes[1].pk, 'status': CircuitStatusChoices.STATUS_DECOMMISSIONED, 'tenant': None, 'commit_rate': 2000, 'description': 'New description', 'comments': 'New comments', }
def setUpTestData(cls): sites = ( Site(name='Site 1', slug='site-1'), Site(name='Site 2', slug='site-2'), ) Site.objects.bulk_create(sites) vrfs = ( VRF(name='VRF 1', rd='65000:1'), VRF(name='VRF 2', rd='65000:2'), ) VRF.objects.bulk_create(vrfs) roles = ( Role(name='Role 1', slug='role-1'), Role(name='Role 2', slug='role-2'), ) Role.objects.bulk_create(roles) Prefix.objects.bulk_create([ Prefix(prefix=IPNetwork('10.1.0.0/16'), vrf=vrfs[0], site=sites[0], role=roles[0]), Prefix(prefix=IPNetwork('10.2.0.0/16'), vrf=vrfs[0], site=sites[0], role=roles[0]), Prefix(prefix=IPNetwork('10.3.0.0/16'), vrf=vrfs[0], site=sites[0], role=roles[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'prefix': IPNetwork('192.0.2.0/24'), 'site': sites[1].pk, 'vrf': vrfs[1].pk, 'tenant': None, 'vlan': None, 'status': PrefixStatusChoices.STATUS_RESERVED, 'role': roles[1].pk, 'is_pool': True, 'description': 'A new prefix', 'tags': [t.pk for t in tags], } cls.csv_data = ( "vrf,prefix,status", "VRF 1,10.4.0.0/16,active", "VRF 1,10.5.0.0/16,active", "VRF 1,10.6.0.0/16,active", ) cls.bulk_edit_data = { 'site': sites[1].pk, 'vrf': vrfs[1].pk, 'tenant': None, 'status': PrefixStatusChoices.STATUS_RESERVED, 'role': roles[1].pk, 'is_pool': False, 'description': 'New description', }
def setUpTestData(cls): sites = ( Site(name='Site 1', slug='site-1'), Site(name='Site 2', slug='site-2'), ) Site.objects.bulk_create(sites) vlangroups = ( VLANGroup(name='VLAN Group 1', slug='vlan-group-1', scope=sites[0]), VLANGroup(name='VLAN Group 2', slug='vlan-group-2', scope=sites[1]), ) VLANGroup.objects.bulk_create(vlangroups) roles = ( Role(name='Role 1', slug='role-1'), Role(name='Role 2', slug='role-2'), ) Role.objects.bulk_create(roles) VLAN.objects.bulk_create([ VLAN(group=vlangroups[0], vid=101, name='VLAN101', site=sites[0], role=roles[0]), VLAN(group=vlangroups[0], vid=102, name='VLAN102', site=sites[0], role=roles[0]), VLAN(group=vlangroups[0], vid=103, name='VLAN103', site=sites[0], role=roles[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'site': sites[1].pk, 'group': vlangroups[1].pk, 'vid': 999, 'name': 'VLAN999', 'tenant': None, 'status': VLANStatusChoices.STATUS_RESERVED, 'role': roles[1].pk, 'description': 'A new VLAN', 'tags': [t.pk for t in tags], } cls.csv_data = ( "vid,name,status", "104,VLAN104,active", "105,VLAN105,active", "106,VLAN106,active", ) cls.bulk_edit_data = { 'site': sites[1].pk, 'group': vlangroups[1].pk, 'tenant': None, 'status': VLANStatusChoices.STATUS_RESERVED, 'role': roles[1].pk, 'description': 'New description', }
def setUpTestData(cls): rirs = [ RIR.objects.create(name='RFC 6996', slug='rfc-6996', description='Private Use', is_private=True), RIR.objects.create(name='RFC 7300', slug='rfc-7300', description='IANA Use', is_private=True), ] sites = [ Site.objects.create(name='Site 1', slug='site-1'), Site.objects.create(name='Site 2', slug='site-2') ] tenants = [ Tenant.objects.create(name='Tenant 1', slug='tenant-1'), Tenant.objects.create(name='Tenant 2', slug='tenant-2'), ] asns = ( ASN(asn=64513, rir=rirs[0], tenant=tenants[0]), ASN(asn=65535, rir=rirs[1], tenant=tenants[1]), ASN(asn=4200000000, rir=rirs[0], tenant=tenants[0]), ASN(asn=4200002301, rir=rirs[1], tenant=tenants[1]), ) ASN.objects.bulk_create(asns) asns[0].sites.set([sites[0]]) asns[1].sites.set([sites[1]]) asns[2].sites.set([sites[0]]) asns[3].sites.set([sites[1]]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'asn': 64512, 'rir': rirs[0].pk, 'tenant': tenants[0].pk, 'site': sites[0].pk, 'description': 'A new ASN', } cls.csv_data = ( "asn,rir", "64533,RFC 6996", "64523,RFC 6996", "4200000002,RFC 6996", ) cls.bulk_edit_data = { 'rir': rirs[1].pk, 'description': 'Next description', }
def setUpTestData(cls): site = Site.objects.create(name='Site 1', slug='site-1') manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1') devicerole = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1') device = Device.objects.create(name='Device 1', site=site, device_type=devicetype, device_role=devicerole) Service.objects.bulk_create([ Service(device=device, name='Service 1', protocol=ServiceProtocolChoices.PROTOCOL_TCP, ports=[101]), Service(device=device, name='Service 2', protocol=ServiceProtocolChoices.PROTOCOL_TCP, ports=[102]), Service(device=device, name='Service 3', protocol=ServiceProtocolChoices.PROTOCOL_TCP, ports=[103]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'device': device.pk, 'virtual_machine': None, 'name': 'Service X', 'protocol': ServiceProtocolChoices.PROTOCOL_TCP, 'ports': '104,105', 'ipaddresses': [], 'description': 'A new service', 'tags': [t.pk for t in tags], } cls.csv_data = ( "device,name,protocol,ports,description", "Device 1,Service 1,tcp,1,First service", "Device 1,Service 2,tcp,2,Second service", "Device 1,Service 3,udp,3,Third service", ) cls.bulk_edit_data = { 'protocol': ServiceProtocolChoices.PROTOCOL_UDP, 'ports': '106,107', 'description': 'New description', }
def setUpTestData(cls): vrfs = ( VRF(name='VRF 1', rd='65000:1'), VRF(name='VRF 2', rd='65000:2'), ) VRF.objects.bulk_create(vrfs) roles = ( Role(name='Role 1', slug='role-1'), Role(name='Role 2', slug='role-2'), ) Role.objects.bulk_create(roles) ip_ranges = ( IPRange(start_address='192.168.0.10/24', end_address='192.168.0.100/24', size=91), IPRange(start_address='192.168.1.10/24', end_address='192.168.1.100/24', size=91), IPRange(start_address='192.168.2.10/24', end_address='192.168.2.100/24', size=91), IPRange(start_address='192.168.3.10/24', end_address='192.168.3.100/24', size=91), IPRange(start_address='192.168.4.10/24', end_address='192.168.4.100/24', size=91), ) IPRange.objects.bulk_create(ip_ranges) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'start_address': IPNetwork('192.0.5.10/24'), 'end_address': IPNetwork('192.0.5.100/24'), 'vrf': vrfs[1].pk, 'tenant': None, 'vlan': None, 'status': IPRangeStatusChoices.STATUS_RESERVED, 'role': roles[1].pk, 'is_pool': True, 'description': 'A new IP range', 'tags': [t.pk for t in tags], } cls.csv_data = ( "vrf,start_address,end_address,status", "VRF 1,10.1.0.1/16,10.1.9.254/16,active", "VRF 1,10.2.0.1/16,10.2.9.254/16,active", "VRF 1,10.3.0.1/16,10.3.9.254/16,active", ) cls.bulk_edit_data = { 'vrf': vrfs[1].pk, 'tenant': None, 'status': IPRangeStatusChoices.STATUS_RESERVED, 'role': roles[1].pk, 'description': 'New description', }
def setUpTestData(cls): sites = ( Site(name='Site 1', slug='site-1'), Site(name='Site 2', slug='site-2'), ) Site.objects.bulk_create(sites) clustergroups = ( ClusterGroup(name='Cluster Group 1', slug='cluster-group-1'), ClusterGroup(name='Cluster Group 2', slug='cluster-group-2'), ) ClusterGroup.objects.bulk_create(clustergroups) clustertypes = ( ClusterType(name='Cluster Type 1', slug='cluster-type-1'), ClusterType(name='Cluster Type 2', slug='cluster-type-2'), ) ClusterType.objects.bulk_create(clustertypes) Cluster.objects.bulk_create([ Cluster(name='Cluster 1', group=clustergroups[0], type=clustertypes[0], site=sites[0]), Cluster(name='Cluster 2', group=clustergroups[0], type=clustertypes[0], site=sites[0]), Cluster(name='Cluster 3', group=clustergroups[0], type=clustertypes[0], site=sites[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Cluster X', 'group': clustergroups[1].pk, 'type': clustertypes[1].pk, 'tenant': None, 'site': sites[1].pk, 'comments': 'Some comments', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,type", "Cluster 4,Cluster Type 1", "Cluster 5,Cluster Type 1", "Cluster 6,Cluster Type 1", ) cls.bulk_edit_data = { 'group': clustergroups[1].pk, 'type': clustertypes[1].pk, 'tenant': None, 'site': sites[1].pk, 'comments': 'New comments', }
def test_create_tagged_item(self): tags = create_tags("Foo", "Bar", "Baz") data = { 'name': 'Test Site', 'slug': 'test-site', 'tags': [t.pk for t in tags] } url = reverse('dcim-api:site-list') self.add_permissions('dcim.add_site') response = self.client.post(url, data, format='json', **self.header) self.assertHttpStatus(response, status.HTTP_201_CREATED) self.assertListEqual(sorted([t['id'] for t in response.data['tags']]), sorted(data['tags'])) site = Site.objects.get(pk=response.data['id']) self.assertListEqual(sorted([t.name for t in site.tags.all()]), sorted(["Foo", "Bar", "Baz"]))
def setUpTestData(cls): vrfs = ( VRF(name='VRF 1', rd='65000:1'), VRF(name='VRF 2', rd='65000:2'), ) VRF.objects.bulk_create(vrfs) IPAddress.objects.bulk_create([ IPAddress(address=IPNetwork('192.0.2.1/24'), vrf=vrfs[0]), IPAddress(address=IPNetwork('192.0.2.2/24'), vrf=vrfs[0]), IPAddress(address=IPNetwork('192.0.2.3/24'), vrf=vrfs[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'vrf': vrfs[1].pk, 'address': IPNetwork('192.0.2.99/24'), 'tenant': None, 'status': IPAddressStatusChoices.STATUS_RESERVED, 'role': IPAddressRoleChoices.ROLE_ANYCAST, 'nat_inside': None, 'dns_name': 'example', 'description': 'A new IP address', 'tags': [t.pk for t in tags], } cls.csv_data = ( "vrf,address,status", "VRF 1,192.0.2.4/24,active", "VRF 1,192.0.2.5/24,active", "VRF 1,192.0.2.6/24,active", ) cls.bulk_edit_data = { 'vrf': vrfs[1].pk, 'tenant': None, 'status': IPAddressStatusChoices.STATUS_RESERVED, 'role': IPAddressRoleChoices.ROLE_ANYCAST, 'dns_name': 'example', 'description': 'New description', }
def setUpTestData(cls): device = create_test_device('test-device') interfaces = [ Interface(device=device, name=f'radio{i}', type=InterfaceTypeChoices.TYPE_80211AC, rf_channel=WirelessChannelChoices.CHANNEL_5G_32, rf_channel_frequency=5160, rf_channel_width=20) for i in range(12) ] Interface.objects.bulk_create(interfaces) WirelessLink(interface_a=interfaces[0], interface_b=interfaces[1], ssid='LINK1').save() WirelessLink(interface_a=interfaces[2], interface_b=interfaces[3], ssid='LINK2').save() WirelessLink(interface_a=interfaces[4], interface_b=interfaces[5], ssid='LINK3').save() tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'interface_a': interfaces[6].pk, 'interface_b': interfaces[7].pk, 'status': LinkStatusChoices.STATUS_PLANNED, 'tags': [t.pk for t in tags], } cls.csv_data = ( "interface_a,interface_b,status", f"{interfaces[6].pk},{interfaces[7].pk},connected", f"{interfaces[8].pk},{interfaces[9].pk},connected", f"{interfaces[10].pk},{interfaces[11].pk},connected", ) cls.bulk_edit_data = { 'status': LinkStatusChoices.STATUS_PLANNED, }
def setUpTestData(cls): Provider.objects.bulk_create([ Provider(name='Provider 1', slug='provider-1', asn=65001), Provider(name='Provider 2', slug='provider-2', asn=65002), Provider(name='Provider 3', slug='provider-3', asn=65003), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Provider X', 'slug': 'provider-x', 'asn': 65123, 'account': '1234', 'portal_url': 'http://example.com/portal', 'noc_contact': '*****@*****.**', 'admin_contact': '*****@*****.**', 'comments': 'Another provider', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug", "Provider 4,provider-4", "Provider 5,provider-5", "Provider 6,provider-6", ) cls.bulk_edit_data = { 'asn': 65009, 'account': '5678', 'portal_url': 'http://example.com/portal2', 'noc_contact': '*****@*****.**', 'admin_contact': '*****@*****.**', 'comments': 'New comments', }
def setUpTestData(cls): FHRPGroup.objects.bulk_create(( FHRPGroup( protocol=FHRPGroupProtocolChoices.PROTOCOL_VRRP2, group_id=10, auth_type=FHRPGroupAuthTypeChoices.AUTHENTICATION_PLAINTEXT, auth_key='foobar123'), FHRPGroup(protocol=FHRPGroupProtocolChoices.PROTOCOL_VRRP3, group_id=20, auth_type=FHRPGroupAuthTypeChoices.AUTHENTICATION_MD5, auth_key='foobar123'), FHRPGroup(protocol=FHRPGroupProtocolChoices.PROTOCOL_HSRP, group_id=30), )) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'protocol': FHRPGroupProtocolChoices.PROTOCOL_VRRP2, 'group_id': 99, 'auth_type': FHRPGroupAuthTypeChoices.AUTHENTICATION_MD5, 'auth_key': 'abc123def456', 'description': 'Blah blah blah', 'tags': [t.pk for t in tags], } cls.csv_data = ( "protocol,group_id,auth_type,auth_key,description", "vrrp2,40,plaintext,foobar123,Foo", "vrrp3,50,md5,foobar123,Bar", "hsrp,60,,,", ) cls.bulk_edit_data = { 'protocol': FHRPGroupProtocolChoices.PROTOCOL_CARP, }
def setUpTestData(cls): tenants = ( Tenant(name='Tenant A', slug='tenant-a'), Tenant(name='Tenant B', slug='tenant-b'), ) Tenant.objects.bulk_create(tenants) VRF.objects.bulk_create([ VRF(name='VRF 1', rd='65000:1'), VRF(name='VRF 2', rd='65000:2'), VRF(name='VRF 3', rd='65000:3'), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'VRF X', 'rd': '65000:999', 'tenant': tenants[0].pk, 'enforce_unique': True, 'description': 'A new VRF', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name", "VRF 4", "VRF 5", "VRF 6", ) cls.bulk_edit_data = { 'tenant': tenants[1].pk, 'enforce_unique': False, 'description': 'New description', }
def setUpTestData(cls): tenant_groups = ( TenantGroup(name='Tenant Group 1', slug='tenant-group-1'), TenantGroup(name='Tenant Group 2', slug='tenant-group-2'), ) for tenanantgroup in tenant_groups: tenanantgroup.save() Tenant.objects.bulk_create([ Tenant(name='Tenant 1', slug='tenant-1', group=tenant_groups[0]), Tenant(name='Tenant 2', slug='tenant-2', group=tenant_groups[0]), Tenant(name='Tenant 3', slug='tenant-3', group=tenant_groups[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Tenant X', 'slug': 'tenant-x', 'group': tenant_groups[1].pk, 'description': 'A new tenant', 'comments': 'Some comments', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,slug", "Tenant 4,tenant-4", "Tenant 5,tenant-5", "Tenant 6,tenant-6", ) cls.bulk_edit_data = { 'group': tenant_groups[1].pk, }
def setUpTestData(cls): sites = ( Site(name='Site 1', slug='site-1'), Site(name='Site 2', slug='site-2'), ) Site.objects.bulk_create(sites) VLANGroup.objects.bulk_create([ VLANGroup(name='VLAN Group 1', slug='vlan-group-1', scope=sites[0]), VLANGroup(name='VLAN Group 2', slug='vlan-group-2', scope=sites[0]), VLANGroup(name='VLAN Group 3', slug='vlan-group-3', scope=sites[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'VLAN Group X', 'slug': 'vlan-group-x', 'description': 'A new VLAN group', 'tags': [t.pk for t in tags], } cls.csv_data = ( f"name,slug,scope_type,scope_id,description", f"VLAN Group 4,vlan-group-4,,,Fourth VLAN group", f"VLAN Group 5,vlan-group-5,dcim.site,{sites[0].pk},Fifth VLAN group", f"VLAN Group 6,vlan-group-6,dcim.site,{sites[1].pk},Sixth VLAN group", ) cls.bulk_edit_data = { 'description': 'New description', }
def setUpTestData(cls): providers = ( Provider(name='Provider 1', slug='provider-1'), Provider(name='Provider 2', slug='provider-2'), ) Provider.objects.bulk_create(providers) ProviderNetwork.objects.bulk_create([ ProviderNetwork(name='Provider Network 1', provider=providers[0]), ProviderNetwork(name='Provider Network 2', provider=providers[0]), ProviderNetwork(name='Provider Network 3', provider=providers[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'name': 'Provider Network X', 'provider': providers[1].pk, 'description': 'A new provider network', 'comments': 'Longer description goes here', 'tags': [t.pk for t in tags], } cls.csv_data = ( "name,provider,description", "Provider Network 4,Provider 1,Foo", "Provider Network 5,Provider 1,Bar", "Provider Network 6,Provider 1,Baz", ) cls.bulk_edit_data = { 'provider': providers[1].pk, 'description': 'New description', 'comments': 'New comments', }
def setUpTestData(cls): rirs = ( RIR(name='RIR 1', slug='rir-1'), RIR(name='RIR 2', slug='rir-2'), ) RIR.objects.bulk_create(rirs) Aggregate.objects.bulk_create([ Aggregate(prefix=IPNetwork('10.1.0.0/16'), rir=rirs[0]), Aggregate(prefix=IPNetwork('10.2.0.0/16'), rir=rirs[0]), Aggregate(prefix=IPNetwork('10.3.0.0/16'), rir=rirs[0]), ]) tags = create_tags('Alpha', 'Bravo', 'Charlie') cls.form_data = { 'prefix': IPNetwork('10.99.0.0/16'), 'rir': rirs[1].pk, 'date_added': datetime.date(2020, 1, 1), 'description': 'A new aggregate', 'tags': [t.pk for t in tags], } cls.csv_data = ( "prefix,rir", "10.4.0.0/16,RIR 1", "10.5.0.0/16,RIR 1", "10.6.0.0/16,RIR 1", ) cls.bulk_edit_data = { 'rir': rirs[1].pk, 'date_added': datetime.date(2020, 1, 1), 'description': 'New description', }