proxmox-perl-rs/pve-rs/test/resource_scheduling.pl
Wolfgang Bumiller 221be779c4 use Test::More for resource_scheduling tests
Signed-off-by: Wolfgang Bumiller <w.bumiller@proxmox.com>
2022-11-15 11:10:32 +01:00

57 lines
1.8 KiB
Perl
Executable File

#!/usr/bin/perl
use strict;
use warnings;
use Test::More;
# FIXME ensure that the just built library is loaded rather than the installed one and add a test
# target to pve-rs/Makefile afterwards. Issue is that the loader looks into an $PATH/auto directory,
# so it's not enough to use lib qw(../target/release)
# Also might be a good idea to test for existence of the files to avoid surprises if the directory
# structure changes in the future.
#use lib qw(..);
#use lib qw(../target/release);
use PVE::RS::ResourceScheduling::Static;
my $static = PVE::RS::ResourceScheduling::Static->new();
is(scalar($static->list_nodes()->@*), 0, 'node list empty');
$static->add_node("A", 10, 100_000_000_000);
is(scalar($static->list_nodes()->@*), 1, '1 node added');
$static->add_node("B", 20, 200_000_000_000);
is(scalar($static->list_nodes()->@*), 2, '2nd node');
$static->add_node("C", 30, 300_000_000_000);
is(scalar($static->list_nodes()->@*), 3, '3rd node');
$static->remove_node("C");
is(scalar($static->list_nodes()->@*), 2, '3rd removed should be 2');
ok($static->contains_node("A"), 'should contain a node A');
ok($static->contains_node("B"), 'should contain a node B');
ok(!$static->contains_node("C"), 'should not contain a node C');
my $service = {
maxcpu => 4,
maxmem => 20_000_000_000,
};
for (my $i = 0; $i < 15; $i++) {
my $score_list = $static->score_nodes_to_start_service($service);
# imitate HA manager
my $scores = { map { $_->[0] => -$_->[1] } $score_list->@* };
my @nodes = sort {
$scores->{$a} <=> $scores->{$b} || $a cmp $b
} keys $scores->%*;
if ($i % 3 == 2) {
is($nodes[0], "A", 'first should be A');
is($nodes[1], "B", 'second should be A');
} else {
is($nodes[0], "B", 'first should be B');
is($nodes[1], "A", 'second should be A');
}
$static->add_service_usage_to_node($nodes[0], $service);
}
done_testing();