44 lines
1.9 KiB
Erlang
44 lines
1.9 KiB
Erlang
-module(ip2region_xdb_test).
|
|
|
|
-include_lib("eunit/include/eunit.hrl").
|
|
-include("ip2region.hrl").
|
|
|
|
valid_v4_header_test() ->
|
|
HeaderBin = <<2:16/little, 1:16/little, 0:32/little,
|
|
256:32/little, 1024:32/little,
|
|
?IP_VERSION_4:16/little, 4:16/little>>,
|
|
{ok, Header} = ip2region_xdb:parse_header(HeaderBin),
|
|
?assertEqual(2, ip2region_xdb:header_version(Header)),
|
|
?assertEqual(1, ip2region_xdb:header_index_policy(Header)),
|
|
?assertEqual(256, ip2region_xdb:header_start_index_ptr(Header)),
|
|
?assertEqual(1024, ip2region_xdb:header_end_index_ptr(Header)),
|
|
?assertEqual(?IP_VERSION_4, ip2region_xdb:header_ip_version(Header)),
|
|
?assertEqual(4, ip2region_xdb:header_runtime_ptr_bytes(Header)).
|
|
|
|
valid_v6_header_test() ->
|
|
HeaderBin = <<3:16/little, 1:16/little, 0:32/little,
|
|
256:32/little, 1024:32/little,
|
|
?IP_VERSION_6:16/little, 4:16/little>>,
|
|
{ok, Header} = ip2region_xdb:parse_header(HeaderBin),
|
|
?assertEqual(3, ip2region_xdb:header_version(Header)),
|
|
?assertEqual(?IP_VERSION_6, ip2region_xdb:header_ip_version(Header)).
|
|
|
|
invalid_header_test() ->
|
|
?assertEqual({error, invalid_header}, ip2region_xdb:parse_header(<<0:128>>)).
|
|
|
|
table_helpers_test_() ->
|
|
[
|
|
?_assertEqual(?XDB_VECTOR_INDEX_V4, ip2region_xdb:vector_index_table(ipv4)),
|
|
?_assertEqual(?XDB_VECTOR_INDEX_V6, ip2region_xdb:vector_index_table(ipv6)),
|
|
?_assertEqual(?XDB_SEGMENT_INDEX_V4, ip2region_xdb:segment_index_table(ipv4)),
|
|
?_assertEqual(?XDB_SEGMENT_INDEX_V6, ip2region_xdb:segment_index_table(ipv6)),
|
|
?_assertEqual(?IP2REGION_CACHE_V4, ip2region_xdb:cache_table(ipv4)),
|
|
?_assertEqual(?IP2REGION_CACHE_V6, ip2region_xdb:cache_table(ipv6))
|
|
].
|
|
|
|
segment_index_size_test_() ->
|
|
[
|
|
?_assertEqual(14, ip2region_xdb:segment_index_size(ipv4)),
|
|
?_assertEqual(38, ip2region_xdb:segment_index_size(ipv6))
|
|
].
|