From: Alexandre Derumier Date: Sun, 7 Feb 2021 14:07:31 +0000 (+0100) Subject: tests: add subnets tests X-Git-Url: https://git.puffer.fish/?a=commitdiff_plain;h=5221635a52b90f955e1f895ac122615b8ecfa16e;p=matthieu%2Fpve-network.git tests: add subnets tests Signed-off-by: Alexandre Derumier --- diff --git a/PVE/Network/SDN/Subnets.pm b/PVE/Network/SDN/Subnets.pm index 0cbf5fd..56dcd28 100644 --- a/PVE/Network/SDN/Subnets.pm +++ b/PVE/Network/SDN/Subnets.pm @@ -251,9 +251,11 @@ sub add_ip { verify_dns_zone($reversednszone, $reversedns); if ($ipamid) { + my $ipam_cfg = PVE::Network::SDN::Ipams::config(); my $plugin_config = $ipam_cfg->{ids}->{$ipamid}; my $plugin = PVE::Network::SDN::Ipams::Plugin->lookup($plugin_config->{type}); + eval { $plugin->add_ip($plugin_config, $subnetid, $subnet, $ip, $hostname, $mac, $description); }; diff --git a/test/Makefile b/test/Makefile index ca05cf7..eedc4e0 100644 --- a/test/Makefile +++ b/test/Makefile @@ -1,6 +1,6 @@ all: test -test: test_zones test_ipams test_dns +test: test_zones test_ipams test_dns test_subnets test_zones: run_test_zones.pl ./run_test_zones.pl @@ -10,3 +10,6 @@ test_ipams: run_test_ipams.pl test_dns: run_test_dns.pl ./run_test_dns.pl + +test_subnets: run_test_subnets.pl + ./run_test_subnets.pl diff --git a/test/run_test_subnets.pl b/test/run_test_subnets.pl new file mode 100755 index 0000000..6ffa6a3 --- /dev/null +++ b/test/run_test_subnets.pl @@ -0,0 +1,273 @@ +#!/usr/bin/perl + +use strict; +use warnings; + +use lib qw(..); +use File::Slurp; + +use Test::More; +use Test::MockModule; + +use PVE::Network::SDN; +use PVE::Network::SDN::Zones; +use PVE::Network::SDN::Controllers; +use PVE::INotify; +use JSON; + +use Data::Dumper qw(Dumper); +$Data::Dumper::Sortkeys = 1; + +sub read_sdn_config { + my ($file) = @_; + # Read structure back in again + open my $in, '<', $file or die $!; + my $sdn_config; + { + local $/; # slurp mode + $sdn_config = eval <$in>; + } + close $in; + + return $sdn_config; +} + + +my @plugins = read_dir( './subnets/', prefix => 1 ) ; + +foreach my $path (@plugins) { + + my (undef, $testid) = split(/\//, $path); + + my $sdn_config = read_sdn_config ("$path/sdn_config"); + + + my $pve_sdn_subnets; + $pve_sdn_subnets = Test::MockModule->new('PVE::Network::SDN::Subnets'); + $pve_sdn_subnets->mock( + config => sub { + return $sdn_config->{subnets}; + }, + verify_dns_zone => sub { + return; + }, + add_dns_record => sub { + return; + } + ); + + + my $js = JSON->new; + $js->canonical(1); + + + #test params; + my $subnets = $sdn_config->{subnets}->{ids}; + my $subnetid = (keys %{$subnets})[0]; + my $subnet = PVE::Network::SDN::Subnets::sdn_subnets_config($sdn_config->{subnets}, $subnetid, 1); + + my $subnet_cidr = $subnet->{cidr}; + my $iplist = NetAddr::IP->new($subnet_cidr); + $iplist++; + my $ip = $iplist->canon(); + $iplist++; + my $ipnextfree = $iplist->canon(); + $iplist++; + my $ip2 = $iplist->canon(); + + my $ip3 = undef; + my $hostname = "myhostname"; + my $mac = "da:65:8f:18:9b:6f"; + my $description = "mydescription"; + my $is_gateway = 1; + my $ipamdb = {}; + + my $zone = $sdn_config->{zones}->{ids}->{"myzone"}; + + my $plugin = PVE::Network::SDN::Ipams::Plugin->lookup('pve'); + my $sdn_ipam_plugin = Test::MockModule->new($plugin); + $sdn_ipam_plugin->mock( + read_db => sub { + return $ipamdb; + }, + write_db => sub { + my ($cfg) = @_; + $ipamdb = $cfg; + } + ); + + ## add_subnet + my $test = "add_subnet"; + my $name = "$testid $test"; + my $result = undef; + my $expected = '{"zones":{"myzone":{"subnets":{"'.$subnet_cidr.'":{"ips":{}}}}}}'; + + eval { + $plugin->add_subnet(undef, $subnetid, $subnet, 1); + }; + + if ($@) { + fail($name); + } else { + $result = $js->encode($plugin->read_db()); + is ($result, $expected, $name); + } + + ## add_ip + $test = "add_ip"; + $name = "$testid $test"; + $result = undef; + $expected = '{"zones":{"myzone":{"subnets":{"'.$subnet_cidr.'":{"ips":{"'.$ip.'":{}}}}}}}'; + + eval { + PVE::Network::SDN::Subnets::add_ip($zone, $subnetid, $subnet, $ip, $hostname, $mac, $description); + }; + + if ($@) { + fail("$name : $@"); + } else { + $result = $js->encode($plugin->read_db()); + is ($result, $expected, $name); + } + + ## add_already_exist_ip + $test = "add_already_exist_ip"; + $name = "$testid $test"; + + eval { + PVE::Network::SDN::Subnets::add_ip($zone, $subnetid, $subnet, $ip, $hostname, $mac, $description); + }; + + if ($@) { + is (undef, undef, $name); + } else { + fail("$name : $@"); + } + + ## add_second_ip + $test = "add_second_ip"; + $name = "$testid $test"; + $result = undef; + $expected = '{"zones":{"myzone":{"subnets":{"'.$subnet_cidr.'":{"ips":{"'.$ip.'":{},"'.$ip2.'":{}}}}}}}'; + + eval { + PVE::Network::SDN::Subnets::add_ip($zone, $subnetid, $subnet, $ip2, $hostname, $mac, $description); + }; + + if ($@) { + fail("$name : $@"); + } else { + $result = $js->encode($plugin->read_db()); + is ($result, $expected, $name); + } + + + ## add_next_free + $test = "add_next_freeip"; + $name = "$testid $test"; + $result = undef; + $expected = '{"zones":{"myzone":{"subnets":{"'.$subnet_cidr.'":{"ips":{"'.$ip.'":{},"'.$ipnextfree.'":{},"'.$ip2.'":{}}}}}}}'; + + eval { + $ip3 = PVE::Network::SDN::Subnets::next_free_ip($zone, $subnetid, $subnet, $hostname, $mac, $description); + }; + + if ($@) { + fail("$name : $@"); + } else { + $result = $js->encode($plugin->read_db()); + is ($result, $expected, $name); + } + + ## del_ip + $test = "del_ip"; + $name = "$testid $test"; + $result = undef; + $expected = '{"zones":{"myzone":{"subnets":{"'.$subnet_cidr.'":{"ips":{"'.$ipnextfree.'":{},"'.$ip2.'":{}}}}}}}'; + + eval { + PVE::Network::SDN::Subnets::del_ip($zone, $subnetid, $subnet, $ip, $hostname); + }; + + if ($@) { + fail("$name : $@"); + } else { + $result = $js->encode($plugin->read_db()); + is ($result, $expected, $name); + } + + ## del_subnet_not_empty + $test = "del_subnet_not_empty"; + $name = "$testid $test"; + $result = undef; + $expected = undef; + + eval { + PVE::Network::SDN::Subnets::del_subnet($zone, $subnetid, $subnet); + }; + + if ($@) { + is ($result, $expected, $name); + } else { + fail("$name : $@"); + } + + + + ## add_ip_rollback_failing_dns + $test = "add_ip_rollback_failing_dns"; + + $pve_sdn_subnets->mock( + config => sub { + return $sdn_config->{subnets}; + }, + verify_dns_zone => sub { + return; + }, + add_dns_record => sub { + die "error add dns record"; + return; + } + ); + + $name = "$testid $test"; + $result = undef; + $expected = '{"zones":{"myzone":{"subnets":{"'.$subnet_cidr.'":{"ips":{"'.$ipnextfree.'":{},"'.$ip2.'":{}}}}}}}'; + + eval { + PVE::Network::SDN::Subnets::add_ip($zone, $subnetid, $subnet, $ip, $hostname, $mac, $description); + }; + + if ($@) { + $result = $js->encode($plugin->read_db()); + is ($result, $expected, $name); + } else { + fail("$name : $@"); + } + + + ## del_empty_subnet + $test = "del_empty_subnet"; + $name = "$testid $test"; + $result = undef; + $expected = '{"zones":{"myzone":{"subnets":{}}}}'; + + PVE::Network::SDN::Subnets::del_ip($zone, $subnetid, $subnet, $ip2, $hostname); + PVE::Network::SDN::Subnets::del_ip($zone, $subnetid, $subnet, $ip3, $hostname); + + eval { + PVE::Network::SDN::Subnets::del_subnet($zone, $subnetid, $subnet); + }; + + if ($@) { + fail("$name : $@"); + } else { + $result = $js->encode($plugin->read_db()); + is ($result, $expected, $name); + } + +} + +done_testing(); + + diff --git a/test/subnets/ipv4/ipam_config b/test/subnets/ipv4/ipam_config new file mode 100644 index 0000000..a33be30 --- /dev/null +++ b/test/subnets/ipv4/ipam_config @@ -0,0 +1,18 @@ +{ + 'ids' => { + 'phpipam' => { + 'url' => 'https://localhost/api/apiadmin', + 'type' => 'phpipam', + 'section' => 1, + 'token' => 'JPHkPSLB4O_XL-GQz4qtEFmNpx-99Htw' + }, + 'pve' => { + 'type' => 'pve' + }, + 'netbox' => { + 'token' => '0123456789abcdef0123456789abcdef01234567', + 'type' => 'netbox', + 'url' => 'http://localhost:8000/api' + } + }, +} diff --git a/test/subnets/ipv4/sdn_config b/test/subnets/ipv4/sdn_config new file mode 100644 index 0000000..72697d4 --- /dev/null +++ b/test/subnets/ipv4/sdn_config @@ -0,0 +1,20 @@ +{ + version => 1, + vnets => { + ids => { + myvnet => { type => "vnet", zone => "myzone" }, + }, + }, + + zones => { + ids => { myzone => { ipam => "pve", type =>"simple" } }, + }, + + subnets => { + ids => { 'myzone-10.0.0.0-24' => { + 'type' => 'subnet', + 'vnet' => 'myvnet', + } + } + } +} diff --git a/test/subnets/ipv6/ipam_config b/test/subnets/ipv6/ipam_config new file mode 100644 index 0000000..a33be30 --- /dev/null +++ b/test/subnets/ipv6/ipam_config @@ -0,0 +1,18 @@ +{ + 'ids' => { + 'phpipam' => { + 'url' => 'https://localhost/api/apiadmin', + 'type' => 'phpipam', + 'section' => 1, + 'token' => 'JPHkPSLB4O_XL-GQz4qtEFmNpx-99Htw' + }, + 'pve' => { + 'type' => 'pve' + }, + 'netbox' => { + 'token' => '0123456789abcdef0123456789abcdef01234567', + 'type' => 'netbox', + 'url' => 'http://localhost:8000/api' + } + }, +} diff --git a/test/subnets/ipv6/sdn_config b/test/subnets/ipv6/sdn_config new file mode 100644 index 0000000..618f234 --- /dev/null +++ b/test/subnets/ipv6/sdn_config @@ -0,0 +1,20 @@ +{ + version => 1, + vnets => { + ids => { + myvnet => { type => "vnet", zone => "myzone" }, + }, + }, + + zones => { + ids => { myzone => { ipam => "pve", type =>"simple" } }, + }, + + subnets => { + ids => { 'myzone-2a0a:1580:2000::-56' => { + 'type' => 'subnet', + 'vnet' => 'myvnet', + } + } + } +}