Michael Davidsaver
60 lines
1.4 KiB
C++
60 lines
1.4 KiB
C++
/**
|
|
* Copyright - See the COPYRIGHT that is included with this distribution.
|
|
* pvxs is distributed subject to a Software License Agreement found
|
|
* in file LICENSE that is included with this distribution.
|
|
*/
|
|
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <stdexcept>
|
|
#include <memory>
|
|
|
|
#include "capturestd.h"
|
|
|
|
#include <epicsUnitTest.h>
|
|
#include <epicsStdio.h>
|
|
|
|
static
|
|
std::string readFile(const std::string& filename)
|
|
{
|
|
std::ifstream t(filename.c_str());
|
|
std::stringstream buffer;
|
|
|
|
if (!t.is_open()) {
|
|
throw std::invalid_argument("Could not open filename " + filename);
|
|
}
|
|
|
|
buffer << t.rdbuf();
|
|
return buffer.str();
|
|
}
|
|
|
|
CaptureStd::CaptureStd(const std::function<void ()> &fn) {
|
|
std::shared_ptr<FILE>
|
|
out(fopen("testiocsh.out", "w+b"),
|
|
[](FILE* fp){
|
|
(void)fclose(fp);
|
|
}),
|
|
err(fopen("testiocsh.err", "w+b"),
|
|
[](FILE* fp){
|
|
(void)fclose(fp);
|
|
});
|
|
|
|
if(!out || !err)
|
|
testAbort("Unable to open/create testiocsh.out / .err");
|
|
epicsSetThreadStdout(out.get());
|
|
epicsSetThreadStderr(err.get());
|
|
|
|
try {
|
|
fn();
|
|
}catch(...){
|
|
epicsSetThreadStdout(nullptr);
|
|
epicsSetThreadStderr(nullptr);
|
|
throw;
|
|
}
|
|
out.reset();
|
|
err.reset();
|
|
|
|
m_out = readFile("testiocsh.out");
|
|
m_err = readFile("testiocsh.err");
|
|
}
|