#include "../src/search.h"

std::map<int, std::string> prompt;

void test(xdb::search_t& s, const std::string& ip, const std::string& region) {
    if (s.search(ip) != region)
        xdb::log_exit("测试失败, ip " + ip + ", region " + region);
}

void test_ipv4(int policy) {
    std::cout << "测试 IPv4 " << prompt[policy];

    xdb::search_t s("../../data/ip2region_v4.xdb", xdb::ipv4, policy);
    test(s, "0.0.0.0", "0|0|内网IP|内网IP");
    test(s, "1.2.3.4", "美国|华盛顿|0|谷歌");
    test(s, "255.255.255.255", "0|0|内网IP|内网IP");

    std::cout << " 成功" << std::endl;
}

void test_ipv6(int policy) {
    std::cout << "测试 IPv6 " << prompt[policy];

    xdb::search_t s("../../data/ip2region_v6.xdb", xdb::ipv6, policy);
    test(s, "::1", "");
    test(s, "2001:200:124::", "日本|东京都|千代田区|专线用户");

    std::cout << " 成功" << std::endl;
}

int main() {
    prompt[xdb::policy_file]    = "  不缓存:";
    prompt[xdb::policy_vector]  = "部分缓存:";
    prompt[xdb::policy_content] = "全部缓存:";

    test_ipv4(xdb::policy_file);
    test_ipv4(xdb::policy_vector);
    test_ipv4(xdb::policy_content);

    test_ipv6(xdb::policy_file);
    test_ipv6(xdb::policy_vector);
    test_ipv6(xdb::policy_content);
    return 0;
}