99 lines
2.8 KiB
C++
99 lines
2.8 KiB
C++
|
#include <stdio.h>
|
||
|
|
#include <string>
|
||
|
|
#include <chrono>
|
||
|
|
#include <fstream>
|
||
|
|
#include <sstream>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "thirdparty/microtest/microtest.h"
|
||
|
|
#include "SQLParser.h"
|
||
|
|
|
||
|
|
// Read all lines from the given file path. Skips comment lines.
|
||
|
|
std::vector<std::string> readlines(std::string path);
|
||
|
|
|
||
|
|
// Read the queries from all files that were supplied to the test
|
||
|
|
// through the -f argument. For all queries it is checked whether they
|
||
|
|
// can be parsed successfully.
|
||
|
|
TEST(AutoQueryFileTest) {
|
||
|
|
const std::vector<std::string>& args = mt::Runtime::args();
|
||
|
|
|
||
|
|
std::vector<std::string> query_files;
|
||
|
|
|
||
|
|
// Parse command line arguments to retrieve query files.
|
||
|
|
uint i = 1;
|
||
|
|
for (; i < args.size(); ++i) {
|
||
|
|
if (args[i] == "-f") {
|
||
|
|
query_files.push_back(args[++i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
// Read list of queries from all input files.
|
||
|
|
std::vector<std::string> lines;
|
||
|
|
for (std::string path : query_files) {
|
||
|
|
std::vector<std::string> tmp = readlines(path);
|
||
|
|
lines.insert(lines.end(), tmp.begin(), tmp.end());
|
||
|
|
}
|
||
|
|
|
||
|
|
// Execute queries.
|
||
|
|
size_t num_executed = 0;
|
||
|
|
size_t num_failed = 0;
|
||
|
|
for (std::string line : lines) {
|
||
|
|
bool expected_result = true;
|
||
|
|
std::string query = line;
|
||
|
|
|
||
|
|
// If a line starts with '!' parsing is expected to fail.
|
||
|
|
if (query.at(0) == '!') {
|
||
|
|
expected_result = false;
|
||
|
|
query = query.substr(1);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Measuring the parsing time.
|
||
|
|
std::chrono::time_point<std::chrono::system_clock> start, end;
|
||
|
|
start = std::chrono::system_clock::now();
|
||
|
|
|
||
|
|
// Parse the query.
|
||
|
|
hsql::SQLParserResult result;
|
||
|
|
hsql::SQLParser::parse(query, &result);
|
||
|
|
|
||
|
|
end = std::chrono::system_clock::now();
|
||
|
|
std::chrono::duration<double> elapsed_seconds = end - start;
|
||
|
|
double us = elapsed_seconds.count() * 1000 * 1000;
|
||
|
|
|
||
|
|
if (expected_result == result.isValid()) {
|
||
|
|
printf("\033[0;32m{ ok} (%.1fus)\033[0m %s\n", us, line.c_str());
|
||
|
|
} else {
|
||
|
|
printf("\033[0;31m{ failed}\033[0m\n");
|
||
|
|
printf("\t\033[0;31m%s (L%d:%d)\n\033[0m", result.errorMsg(), result.errorLine(), result.errorColumn());
|
||
|
|
printf("\t%s\n", line.c_str());
|
||
|
|
++num_failed;
|
||
|
|
}
|
||
|
|
++num_executed;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (num_failed == 0) {
|
||
|
|
printf("\033[0;32m{ ok} \033[0mAll %lu grammar tests completed successfully!\n", num_executed);
|
||
|
|
} else {
|
||
|
|
fprintf(stderr, "\033[0;31m{ failed} \033[0mSome grammar tests failed! %lu out of %lu tests failed!\n", num_failed, num_executed);
|
||
|
|
}
|
||
|
|
ASSERT_EQ(num_failed, 0);
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<std::string> readlines(std::string path) {
|
||
|
|
std::ifstream infile(path);
|
||
|
|
std::vector<std::string> lines;
|
||
|
|
std::string line;
|
||
|
|
while (std::getline(infile, line)) {
|
||
|
|
std::istringstream iss(line);
|
||
|
|
|
||
|
|
// Skip comments.
|
||
|
|
if (line[0] == '#' ||
|
||
|
|
(line[0] == '-' && line[1] == '-')) {
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
lines.push_back(line);
|
||
|
|
}
|
||
|
|
return lines;
|
||
|
|
}
|