Files
ResInsight/ApplicationLibCode/UnitTests/RiaHpcTools-Test.cpp
T
Jon Jenssen a8dc7dedae Preferences framework
Add preferences for HPC/Batch scheduler
Add slurm and LSF support
Add batch job monitor
Support launcher specific user settings
Use internal timer, not global view timer
Add some more options to batch monitor
Improve naming
Add tests for job lists
2026-10-19 12:54:19 +02:00

135 lines
5.8 KiB
C++

/////////////////////////////////////////////////////////////////////////////////
//
// Copyright (C) 2026 Equinor ASA
//
// ResInsight is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// ResInsight is distributed in the hope that it will be useful, but WITHOUT ANY
// WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.
//
// See the GNU General Public License at <http://www.gnu.org/licenses/gpl.html>
// for more details.
//
/////////////////////////////////////////////////////////////////////////////////
#include "gtest/gtest.h"
#include "RiaHpcTools.h"
#include <QStringList>
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RiaHpcTools, decodeSlurmQueues )
{
QStringList input;
input.append( "PARTITION AVAIL TIMELIMIT NODES STATE NODELIST" );
input.append( "normal* up infinite 1 drain* node-21-03" );
input.append( "normal* up infinite 70 mix node-20-[00-27],node-21-[00-02,04-31]" );
input.append( "normal* up infinite 1 idle node-21-32" );
input.append(
"extra up infinite 53 drain* node-22-[00,03,08-10,16-19,26,43],node-23-[00,16-19,21,42-43],node-24-[14,27,33]" );
input.append( "extra up infinite 1 drain node-25-37" );
input.append(
"extra up infinite 200 alloc node-22-[01-02,04-07,11-15,20-25],node-23-[15,20],node-24-[00-13,15-26,28-32,34-43]" );
input.append( "extra up infinite 51 idle node-22-[27-42],node-23-[01-14,22-41],node-25-43" );
auto queueNames = RiaHpcTools::decodeSlurmQueues( input );
ASSERT_EQ( queueNames.length(), 2 );
ASSERT_TRUE( queueNames.contains( "normal" ) );
ASSERT_TRUE( queueNames.contains( "extra" ) );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RiaHpcTools, decodeLsfQueues )
{
QStringList input;
input.append( "QUEUE_NAME PRIO STATUS MAX JL/U JL/P JL/H NJOBS PEND RUN SUSP RSV" );
input.append( "urgent 99 Open:Active - - - - 0 0 0 0 0" );
input.append( "wait 90 Open:Active - - - - 0 0 0 0 0" );
input.append( "extra 60 Open:Active - - - - 0 0 0 0 0" );
input.append( "super 60 Open:Active - - - - 0 0 0 0 0" );
input.append( "normal 50 Open:Active - - - - 0 0 0 0 0" );
auto queueNames = RiaHpcTools::decodeLsfQueues( input );
ASSERT_EQ( queueNames.length(), 5 );
ASSERT_TRUE( queueNames.contains( "normal" ) );
ASSERT_TRUE( queueNames.contains( "extra" ) );
ASSERT_TRUE( queueNames.contains( "super" ) );
ASSERT_TRUE( queueNames.contains( "wait" ) );
ASSERT_TRUE( queueNames.contains( "urgent" ) );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RiaHpcTools, decodeSlurmJobs )
{
QStringList input;
input.append( "80 RI_DROGON-2 0:51 RUNNING simulations compute_node" );
input.append( "79 RI_DROGON-1 0:54 RUNNING simulations compute_node" );
input.append( "78 RI_DROGON-0 0:57 RUNNING simulations compute_node" );
input.append( "77 RI_DROGON-3 1:00 RUNNING normal compute_node" );
auto jobs = RiaHpcTools::decodeSlurmJobs( input );
ASSERT_TRUE( jobs.size() == 4 );
ASSERT_TRUE( jobs[0].size() == 6 );
ASSERT_TRUE( jobs[1].size() == 6 );
ASSERT_TRUE( jobs[2].size() == 6 );
ASSERT_TRUE( jobs[3].size() == 6 );
ASSERT_TRUE( jobs[0][0] == "80" );
ASSERT_TRUE( jobs[1][0] == "79" );
ASSERT_TRUE( jobs[2][0] == "78" );
ASSERT_TRUE( jobs[3][0] == "77" );
}
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
TEST( RiaHpcTools, decodeLsfJobs )
{
QStringList input;
input.append( "612950 flow TEST-B 22 second(s) RUN mr 3*st-rsb04-04-01 " );
input.append( "612949 flow TEST-A 30 second(s) RUN mr 5*st-rsb03-03-08 " );
input.append( "612952 flow TEST-C 12 second(s) RUN jr 7*st-rsv17-17-16" );
input.append( "612951 flow TEST-D 15 second(s) RUN mr 6*st-rsb04-04-02 " );
auto jobs = RiaHpcTools::decodeLsfJobs( input );
ASSERT_TRUE( jobs.size() == 4 );
ASSERT_TRUE( jobs[0].size() == 6 );
ASSERT_TRUE( jobs[1].size() == 6 );
ASSERT_TRUE( jobs[2].size() == 6 );
ASSERT_TRUE( jobs[3].size() == 6 );
ASSERT_TRUE( jobs[0][0] == "612950" );
ASSERT_TRUE( jobs[1][0] == "612949" );
ASSERT_TRUE( jobs[2][0] == "612952" );
ASSERT_TRUE( jobs[3][0] == "612951" );
ASSERT_TRUE( jobs[0][1] == "flow TEST-B" );
ASSERT_TRUE( jobs[1][1] == "flow TEST-A" );
ASSERT_TRUE( jobs[2][1] == "flow TEST-C" );
ASSERT_TRUE( jobs[3][1] == "flow TEST-D" );
ASSERT_TRUE( jobs[0][4] == "mr" );
ASSERT_TRUE( jobs[1][4] == "mr" );
ASSERT_TRUE( jobs[2][4] == "jr" );
ASSERT_TRUE( jobs[3][4] == "mr" );
}