55 lines
1.8 KiB
Erlang
55 lines
1.8 KiB
Erlang
-module(xdb_test).
|
|
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
|
|
-define(IPV6_RESULT, "United States|Florida|Miami|Google LLC|US").
|
|
|
|
search_test_() ->
|
|
{ok, _} = ip2region_test_util:start_app(),
|
|
A = "中国|广东省|广州市|中国电信|CN",
|
|
Region0 = xdb:search("1.0.8.0"),
|
|
Region1 = xdb:search(<<"1.0.8.0">>),
|
|
Region2 = xdb:search({1,0,8,0}),
|
|
Region3 = xdb:search("xxx.0.8.0"),
|
|
[
|
|
?_assert(A =:= Region0),
|
|
?_assert(A =:= Region1),
|
|
?_assert(A =:= Region2),
|
|
?_assert({error, bad_ip_format} =:= Region3)
|
|
].
|
|
|
|
ipv6_search_test_() ->
|
|
{ok, _} = ip2region_test_util:start_app(),
|
|
[
|
|
?_assert(?IPV6_RESULT =:= xdb:search("2001:4860:4860::8888")),
|
|
?_assert(?IPV6_RESULT =:= xdb:search(<<"2001:4860:4860::8888">>)),
|
|
?_assert(?IPV6_RESULT =:= xdb:search({8193, 18528, 18528, 0, 0, 0, 0, 34952}))
|
|
].
|
|
|
|
invalid_search_test_() ->
|
|
{ok, _} = ip2region_test_util:start_app(),
|
|
[
|
|
?_assertEqual({error, bad_ip_format}, xdb:search("xxx.0.8.0")),
|
|
?_assertEqual({error, bad_ip_format}, xdb:search("::ggg")),
|
|
?_assertEqual({error, bad_ip_format}, xdb:search({1,2,3}))
|
|
].
|
|
|
|
ipv6_pool_not_configured_test() ->
|
|
{ok, _} = ip2region_test_util:start_app([{ipv4, "ip2region.xdb"}]),
|
|
?assertEqual({error, pool_not_configured}, xdb:search("2001:4860:4860::8888")).
|
|
|
|
xdb_version_mismatch_test() ->
|
|
{ok, _} = ip2region_test_util:start_app(),
|
|
V4File = filename:join([
|
|
ip2region_test_util:repo_root(), "binding", "erlang", "priv", "ip2region.xdb"
|
|
]),
|
|
error_logger:tty(false),
|
|
try
|
|
?assertEqual(
|
|
{error, {xdb_version_mismatch, ipv6, ipv4}},
|
|
ip2region_worker:start([{xdb_file, V4File}, {expected_version, ipv6}])
|
|
)
|
|
after
|
|
error_logger:tty(true)
|
|
end.
|