data_file_unittests.cc 3.35 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
// Copyright (C) 2014 Internet Systems Consortium, Inc. ("ISC")
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
//
// THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
// AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
// OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
// PERFORMANCE OF THIS SOFTWARE.

#include <config.h>
#include <exceptions/exceptions.h>
#include <dhcpsrv/daemon.h>
#include <gtest/gtest.h>
#include <cc/data.h>
#include <fstream>

using namespace isc;
using namespace isc::data;

namespace {

/// @brief Test class for testing Deamon class
class DataFileTest : public ::testing::Test {
public:

    /// @brief writes specified text to a file
    ///
    /// That is an auxilliary funtion used in fileRead() tests.
    ///
    /// @param content text to be written to disk
    void writeFile(const std::string& content) {
        // Write sample content to disk
        unlink(TEMP_FILE);
        std::ofstream write_me(TEMP_FILE);
        EXPECT_TRUE(write_me.is_open());
        write_me << content;
        write_me.close();
    }

    /// destructor
    ~DataFileTest() {
        static_cast<void>(unlink(TEMP_FILE));
    }

    /// Name of the temporary file
    static const char* TEMP_FILE;
};

/// Temporary file name used in some tests
const char* DataFileTest::TEMP_FILE="temp-file.json";

// Test checks whether a text file can be read from disk.
TEST_F(DataFileTest, readFileMultiline) {

    const char* no_endline = "{ \"abc\": 123 }";
    const char* with_endline = "{\n \"abc\":\n 123\n }\n";

    // That's what we expect
    ElementPtr exp = Element::fromJSON(no_endline);

    // Write sample content to disk
    writeFile(no_endline);

    // Check that the read content is correct
    EXPECT_TRUE(exp->equals(*Element::fromJSONFile(TEMP_FILE)));

    // Write sample content to disk
    writeFile(with_endline);

    // Check that the read content is correct
    EXPECT_TRUE(exp->equals(*Element::fromJSONFile(TEMP_FILE)));
}

// Test checks whether comments in file are ignored as expected.
TEST_F(DataFileTest, readFileComments) {
    const char* commented_content = "# This is a comment\n"
        "{ \"abc\":\n"
        "# a comment comment\n"
        "1 }\n";

    // That's what we expect
    ElementPtr exp = Element::fromJSON("{ \"abc\": 1 }");

    // Write sample content to disk
    writeFile(commented_content);

    // Check that the read will fail (without comment elimination)
//    EXPECT_THROW(Element::fromJSONFile(TEMP_FILE), JSONError);

    // Check that the read content is correct (with comment elimination)
    EXPECT_NO_THROW(Element::fromJSONFile(TEMP_FILE, true));
    //EXPECT_TRUE(exp->equals(*Element::fromJSONFile(TEMP_FILE, true)));
}

// This test checks that missing file will generate an exception.
TEST_F(DataFileTest, readFileError) {

    // Check that the read content is correct
    EXPECT_THROW(Element::fromJSONFile("no-such-file.txt"), isc::InvalidOperation);
}

};