proxmox-mirrors/proxmox-perl-rs
1
0
Fork 0
mirror of https://git.proxmox.com/git/proxmox-perl-rs synced 2025-07-27 07:52:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

pve: test: resource scheduling: use dedicated functions for tests

Browse Source 
Signed-off-by: Fiona Ebner <f.ebner@proxmox.com>
This commit is contained in:
Fiona Ebner 2023-03-21 13:33:44 +01:00 committed by Thomas Lamprecht
parent 0648876946
commit 24702d4472
1 changed files with 48 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
pve-rs/test/resource_scheduling.pl
View File

@ -7,43 +7,54 @@ use Test::More;
use PVE::RS::ResourceScheduling::Static; use PVE::RS::ResourceScheduling::Static;
my $static = PVE::RS::ResourceScheduling::Static->new(); sub test_basic {
is(scalar($static->list_nodes()->@*), 0, 'node list empty'); my $static = PVE::RS::ResourceScheduling::Static->new();
$static->add_node("A", 10, 100_000_000_000); is(scalar($static->list_nodes()->@*), 0, 'node list empty');
is(scalar($static->list_nodes()->@*), 1, '1 node added'); $static->add_node("A", 10, 100_000_000_000);
$static->add_node("B", 20, 200_000_000_000); is(scalar($static->list_nodes()->@*), 1, '1 node added');
is(scalar($static->list_nodes()->@*), 2, '2nd node'); $static->add_node("B", 20, 200_000_000_000);
$static->add_node("C", 30, 300_000_000_000); is(scalar($static->list_nodes()->@*), 2, '2nd node');
is(scalar($static->list_nodes()->@*), 3, '3rd node'); $static->add_node("C", 30, 300_000_000_000);
$static->remove_node("C"); is(scalar($static->list_nodes()->@*), 3, '3rd node');
is(scalar($static->list_nodes()->@*), 2, '3rd removed should be 2'); $static->remove_node("C");
ok($static->contains_node("A"), 'should contain a node A'); is(scalar($static->list_nodes()->@*), 2, '3rd removed should be 2');
ok($static->contains_node("B"), 'should contain a node B'); ok($static->contains_node("A"), 'should contain a node A');
ok(!$static->contains_node("C"), 'should not contain a node C'); 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);
} }
sub test_balance {
my $static = PVE::RS::ResourceScheduling::Static->new();
$static->add_node("A", 10, 100_000_000_000);
$static->add_node("B", 20, 200_000_000_000);
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);
}
}
test_basic();
test_balance();
done_testing(); done_testing();