Skip to content

Commit 55636d1

Browse files
Use std::function for simulationInitializer
1 parent 336646b commit 55636d1

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

flow/UnitTestRunner.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
#include <exception>
3838
#include <string>
3939
#include <string_view>
40+
#include <utility>
4041
#include <vector>
4142

4243
#ifndef _WIN32
@@ -343,7 +344,7 @@ Future<Void> stopNetworkAfter(Future<Void> what, std::string_view traceName, int
343344
} // namespace
344345

345346
UnitTestRunnerConfig::UnitTestRunnerConfig(std::string_view sourceSubDir, SimulationInitializer simulationInitializer)
346-
: sourceSubDir(sourceSubDir), simulationInitializer(simulationInitializer) {}
347+
: sourceSubDir(sourceSubDir), simulationInitializer(std::move(simulationInitializer)) {}
347348

348349
std::string_view UnitTestRunnerConfig::suiteName() const {
349350
return sourceSubDir;
@@ -358,7 +359,7 @@ std::string UnitTestRunnerConfig::traceName() const {
358359
}
359360

360361
bool UnitTestRunnerConfig::supportsSimulation() const {
361-
return simulationInitializer != nullptr;
362+
return static_cast<bool>(simulationInitializer);
362363
}
363364

364365
void UnitTestRunnerConfig::initializeSimulation() const {

flow/include/flow/UnitTestRunner.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,15 @@
2222
#define FLOW_UNIT_TEST_RUNNER_H
2323
#pragma once
2424

25+
#include <functional>
2526
#include <string>
2627
#include <string_view>
2728

2829
class UnitTestRunnerConfig {
2930
public:
30-
using SimulationInitializer = void (*)();
31+
using SimulationInitializer = std::function<void()>;
3132

32-
explicit UnitTestRunnerConfig(std::string_view sourceSubDir, SimulationInitializer simulationInitializer = nullptr);
33+
explicit UnitTestRunnerConfig(std::string_view sourceSubDir, SimulationInitializer simulationInitializer = {});
3334

3435
std::string_view suiteName() const;
3536
std::string dataDir() const;

0 commit comments

Comments
 (0)