mirror of
https://github.com/OPM/ResInsight.git
synced 2026-10-26 15:10:22 -05:00
JJ
6c39729e77
Add preferences for HPC/Batch scheduler Add Slurm and LSF support Add batch job monitor Support launcher specific user settings Use internal timer to refresh job state Add some more options to batch monitor Improve naming Add tests for job lists Change WSL handling for processes.
135 lines
5.8 KiB
C++
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" );
|
|
}
|