StdWriters.hpp

Go to the documentation of this file.

// @formatter:off
//
// Balau core C++ library
//
// Copyright (C) 2008 Bora Software (contact@borasoftware.com)
//
// Licensed under the Boost Software License - Version 1.0 - August 17th, 2003.
// See the LICENSE file for the full license text.
//


#ifndef COM_BORA_SOFTWARE__BALAU_TESTING__STD_WRITERS
#define COM_BORA_SOFTWARE__BALAU_TESTING__STD_WRITERS

#include <Balau/Resource/FileByteWriteResource.hpp>
#include <Balau/Resource/File.hpp>
#include <Balau/Resource/Uri.hpp>
#include <Balau/Testing/Assertions.hpp>
#include <Balau/Testing/Writers/TestWriter.hpp>
#include <Balau/Util/Vectors.hpp>

#include <mutex>

namespace Balau::Testing {

class StdOutTestWriter : public TestWriter {
    public: void writeString(const std::string & str) override {
        std::cout << str;
    }

    public: std::unique_ptr<TestWriter> clone() const override {
        return std::unique_ptr<TestWriter>(new StdOutTestWriter());
    }
};

class FileTestWriter : public TestWriter {
    public: explicit FileTestWriter(const Resource::File & file) : writeResource(file) {}

    public: void writeString(const std::string & str) override {
        std::lock_guard<std::mutex> lock(mutex);
        writeResource.writeStream() << str;
    }

    public: std::unique_ptr<TestWriter> clone() const override {
        return std::unique_ptr<TestWriter>(new FileTestWriter(writeResource.getFile()));
    }


    private: Resource::FileByteWriteResource writeResource;
    private: std::mutex mutex;
};

class OStreamTestWriter : public TestWriter {
    public: explicit OStreamTestWriter(std::ostream & stream_) : stream(stream_) {}

    public: void writeString(const std::string & str) override {
        stream << str;
    }

    public: std::unique_ptr<TestWriter> clone() const override {
        return std::unique_ptr<TestWriter>(new OStreamTestWriter(stream));
    }


    private: std::ostream & stream;
};

} // namespace Balau::Testing

#endif // COM_BORA_SOFTWARE__BALAU_TESTING__STD_WRITERS