test(erlang): expand xdb header and table helper coverage

This commit is contained in:
Alice39s 2026-06-28 05:10:07 +09:00
parent d1dda93abd
commit fac3f73193
No known key found for this signature in database
GPG Key ID: 0E0AA11C3939DDDD
1 changed files with 38 additions and 11 deletions

View File

@ -1,16 +1,43 @@
-module(ip2region_xdb_test). -module(ip2region_xdb_test).
-include_lib("eunit/include/eunit.hrl").
header_parse_test() -> -include_lib("eunit/include/eunit.hrl").
HeaderBin = <<3:16/little, 1:16/little, 123456:32/little, -include("ip2region.hrl").
1000:32/little, 2000:32/little, 6:16/little, 4:16/little,
0:((256-20)*8)>>, 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), {ok, Header} = ip2region_xdb:parse_header(HeaderBin),
?assertEqual(3, ip2region_xdb:header_version(Header)), ?assertEqual(3, ip2region_xdb:header_version(Header)),
?assertEqual(6, ip2region_xdb:header_ip_version(Header)), ?assertEqual(?IP_VERSION_6, ip2region_xdb:header_ip_version(Header)).
?assertEqual(1000, ip2region_xdb:header_start_index_ptr(Header)),
?assertEqual(2000, ip2region_xdb:header_end_index_ptr(Header)).
version_constants_test() -> invalid_header_test() ->
?assertEqual(14, ip2region_xdb:segment_index_size(ipv4)), ?assertEqual({error, invalid_header}, ip2region_xdb:parse_header(<<0:128>>)).
?assertEqual(38, ip2region_xdb:segment_index_size(ipv6)).
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))
].