Aegisub/tests/tests/thesaurus.cpp

// Copyright (c) 2012, Thomas Goyne <plorkyeran@aegisub.org>
//
// Permission to use, copy, modify, and 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 THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR 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 <libaegisub/fs.h>
#include <libaegisub/thesaurus.h>

#include <main.h>
#include <util.h>

#include <fstream>

class lagi_thes : public libagi {
protected:
	std::string idx_path;
	std::string dat_path;

	void SetUp() override {
		// Use '\n' on all platforms, as we are mapping the file to memory as binary data, and readline can only handle '\n'
		const char endl = '\n';

		idx_path = "data/thes.idx";
		dat_path = "data/thes.dat";

		std::ofstream idx(idx_path.c_str(), std::ios_base::binary);
		std::ofstream dat(dat_path.c_str(), std::ios_base::binary);

		idx << "UTF-8" << endl;
		dat << "UTF-8" << endl;
		idx << 7 << endl; // entry count

		idx << "Word 1|" << dat.tellp() << endl;
		dat << "Word 1|1" << endl;
		dat << "(noun)|Word 1|Word 1A|Word 1B|Word 1C" << endl;

		idx << "Word 2|" << dat.tellp() << endl;
		dat << "Word 2|2" << endl;
		dat << "(adj)|Word 2|Word 2 adj" << endl;
		dat << "(noun)|Word 2|Word 2 noun" << endl;

		dat << "Unindexed Word|1" << endl;
		dat << "(adv)|Unindexed Word|Indexed Word" << endl;

		idx << "Word 3|" << dat.tellp() << endl;
		dat << "Word 3|1" << endl;
		dat << "(verb)|Not Word 3|Four" << endl;

		idx << "Too few fields" << endl;
		idx << "Too many fields|100|100" << endl;
		idx << "Not a number|foo" << endl;
		idx << "Out of range|" << dat.tellp() << endl;
		idx << "Further out of range|" << 1 + dat.tellp() << endl;
	}
};

TEST_F(lagi_thes, parse) {
	ASSERT_NO_THROW(agi::Thesaurus(dat_path, idx_path));
}

TEST_F(lagi_thes, word_1) {
	agi::Thesaurus thes(dat_path, idx_path);

	std::vector<agi::Thesaurus::Entry> entries;
	ASSERT_NO_THROW(entries = thes.Lookup("Word 1"));
	ASSERT_EQ(1, entries.size());
	ASSERT_EQ(3, entries[0].second.size());
	EXPECT_STREQ("(noun) Word 1", entries[0].first.c_str());
	EXPECT_STREQ("Word 1A", entries[0].second[0].c_str());
	EXPECT_STREQ("Word 1B", entries[0].second[1].c_str());
	EXPECT_STREQ("Word 1C", entries[0].second[2].c_str());
}

TEST_F(lagi_thes, word_2) {
	agi::Thesaurus thes(dat_path, idx_path);

	std::vector<agi::Thesaurus::Entry> entries;
	ASSERT_NO_THROW(entries = thes.Lookup("Word 2"));
	ASSERT_EQ(2, entries.size());
	ASSERT_EQ(1, entries[0].second.size());
	ASSERT_EQ(1, entries[1].second.size());
	EXPECT_STREQ("(adj) Word 2", entries[0].first.c_str());
	EXPECT_STREQ("(noun) Word 2", entries[1].first.c_str());
	EXPECT_STREQ("Word 2 adj", entries[0].second[0].c_str());
	EXPECT_STREQ("Word 2 noun", entries[1].second[0].c_str());
}

TEST_F(lagi_thes, word_3) {
	agi::Thesaurus thes(dat_path, idx_path);

	std::vector<agi::Thesaurus::Entry> entries;
	ASSERT_NO_THROW(entries = thes.Lookup("Word 3"));
	ASSERT_EQ(1, entries.size());
	ASSERT_EQ(1, entries[0].second.size());
	EXPECT_STREQ("(verb) Not Word 3", entries[0].first.c_str());
	EXPECT_STREQ("Four", entries[0].second[0].c_str());
}

TEST_F(lagi_thes, bad_word) {
	agi::Thesaurus thes(dat_path, idx_path);

	std::vector<agi::Thesaurus::Entry> entries;
	ASSERT_NO_THROW(entries = thes.Lookup("Nonexistent word"));
	EXPECT_EQ(0, entries.size());
}

TEST_F(lagi_thes, lookup_clears) {
	agi::Thesaurus thes(dat_path, idx_path);

	std::vector<agi::Thesaurus::Entry> entries;
	ASSERT_NO_THROW(entries = thes.Lookup("Word 1"));
	ASSERT_NO_THROW(entries = thes.Lookup("Word 2"));
	ASSERT_NO_THROW(entries = thes.Lookup("Word 3"));
	EXPECT_EQ(1, entries.size());
}

TEST_F(lagi_thes, malformed_index_lines) {
	agi::Thesaurus thes(dat_path, idx_path);

	std::vector<agi::Thesaurus::Entry> entries;
	ASSERT_NO_THROW(entries = thes.Lookup("Too few fields"));
	EXPECT_EQ(0, entries.size());
	ASSERT_NO_THROW(entries = thes.Lookup("Too many fields"));
	EXPECT_EQ(0, entries.size());
	ASSERT_NO_THROW(entries = thes.Lookup("Not a number"));
	EXPECT_EQ(0, entries.size());
	ASSERT_NO_THROW(entries = thes.Lookup("Out of range"));
	EXPECT_EQ(0, entries.size());
	ASSERT_NO_THROW(entries = thes.Lookup("Further out of range"));
	EXPECT_EQ(0, entries.size());
}

TEST_F(lagi_thes, unindexed_word) {
	agi::Thesaurus thes(dat_path, idx_path);

	std::vector<agi::Thesaurus::Entry> entries;
	ASSERT_NO_THROW(entries = thes.Lookup("Unindexed Word"));
	EXPECT_EQ(0, entries.size());
}