116 lines
3.5 KiB
C++
116 lines
3.5 KiB
C++
#include <cinttypes>
|
|
#include <cstdio>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <cmath>
|
|
#include <set>
|
|
#include <string_view>
|
|
|
|
#include "simdjson.h"
|
|
|
|
using namespace simdjson;
|
|
using namespace std;
|
|
|
|
#ifndef JSON_TEST_PATH
|
|
#define JSON_TEST_PATH "jsonexamples/twitter.json"
|
|
#endif
|
|
|
|
#define TEST_START() { cout << "Running " << __func__ << " ..." << endl; }
|
|
#define ASSERT_ERROR(ACTUAL, EXPECTED) if ((ACTUAL) != (EXPECTED)) { cerr << "FAIL: Unexpected error \"" << (ACTUAL) << "\" (expected \"" << (EXPECTED) << "\")" << endl; return false; }
|
|
#define TEST_FAIL(MESSAGE) { cerr << "FAIL: " << (MESSAGE) << endl; return false; }
|
|
#define TEST_SUCCEED() { return true; }
|
|
namespace parser_load {
|
|
const char * NONEXISTENT_FILE = "this_file_does_not_exit.json";
|
|
bool parser_load_capacity() {
|
|
TEST_START();
|
|
document::parser parser(1); // 1 byte max capacity
|
|
auto [doc, error] = parser.load(JSON_TEST_PATH);
|
|
ASSERT_ERROR(error, CAPACITY);
|
|
TEST_SUCCEED();
|
|
}
|
|
bool parser_load_many_capacity() {
|
|
TEST_START();
|
|
document::parser parser(1); // 1 byte max capacity
|
|
for (auto [doc, error] : parser.load_many(JSON_TEST_PATH)) {
|
|
ASSERT_ERROR(error, CAPACITY);
|
|
TEST_SUCCEED();
|
|
}
|
|
TEST_FAIL("No documents returned");
|
|
}
|
|
|
|
bool document_load_nonexistent() {
|
|
TEST_START();
|
|
auto [doc, error] = document::load(NONEXISTENT_FILE);
|
|
ASSERT_ERROR(error, IO_ERROR);
|
|
TEST_SUCCEED();
|
|
}
|
|
bool parser_load_nonexistent() {
|
|
TEST_START();
|
|
document::parser parser;
|
|
auto [doc, error] = parser.load(NONEXISTENT_FILE);
|
|
ASSERT_ERROR(error, IO_ERROR);
|
|
TEST_SUCCEED();
|
|
}
|
|
bool parser_load_many_nonexistent() {
|
|
TEST_START();
|
|
document::parser parser;
|
|
for (auto [doc, error] : parser.load_many(NONEXISTENT_FILE)) {
|
|
ASSERT_ERROR(error, IO_ERROR);
|
|
TEST_SUCCEED();
|
|
}
|
|
TEST_FAIL("No documents returned");
|
|
}
|
|
bool padded_string_load_nonexistent() {
|
|
TEST_START();
|
|
auto [str, error] = padded_string::load(NONEXISTENT_FILE);
|
|
ASSERT_ERROR(error, IO_ERROR);
|
|
TEST_SUCCEED();
|
|
}
|
|
|
|
bool document_load_chain() {
|
|
TEST_START();
|
|
auto [val, error] = document::load(NONEXISTENT_FILE)["foo"].as_uint64_t();
|
|
ASSERT_ERROR(error, IO_ERROR);
|
|
TEST_SUCCEED();
|
|
}
|
|
bool parser_load_chain() {
|
|
TEST_START();
|
|
document::parser parser;
|
|
auto [val, error] = parser.load(NONEXISTENT_FILE)["foo"].as_uint64_t();
|
|
ASSERT_ERROR(error, IO_ERROR);
|
|
TEST_SUCCEED();
|
|
}
|
|
bool parser_load_many_chain() {
|
|
TEST_START();
|
|
document::parser parser;
|
|
for (auto doc_result : parser.load_many(NONEXISTENT_FILE)) {
|
|
auto [val, error] = doc_result["foo"].as_uint64_t();
|
|
ASSERT_ERROR(error, IO_ERROR);
|
|
TEST_SUCCEED();
|
|
}
|
|
TEST_FAIL("No documents returned");
|
|
}
|
|
bool run() {
|
|
return parser_load_capacity() && parser_load_many_capacity()
|
|
&& parser_load_nonexistent() && parser_load_many_nonexistent() && document_load_nonexistent() && padded_string_load_nonexistent()
|
|
&& document_load_chain() && parser_load_chain() && parser_load_many_chain();
|
|
}
|
|
}
|
|
|
|
int main() {
|
|
// this is put here deliberately to check that the documentation is correct (README),
|
|
// should this fail to compile, you should update the documentation:
|
|
if (simdjson::active_implementation->name() == "unsupported") {
|
|
printf("unsupported CPU\n");
|
|
}
|
|
std::cout << "Running error tests." << std::endl;
|
|
if (!parser_load::run()) {
|
|
return EXIT_FAILURE;
|
|
}
|
|
std::cout << "Error tests are ok." << std::endl;
|
|
return EXIT_SUCCESS;
|
|
}
|