File: freeze-thaw-record-size.t

package info (click to toggle)
libmaxmind-db-writer-perl 0.300003-5
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 1,336 kB
  • sloc: ansic: 3,059; perl: 2,895; makefile: 5; sh: 4
file content (79 lines) | stat: -rw-r--r-- 1,930 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
use strict;
use warnings;

use Test::Fatal;
use Test::More;

use File::Temp qw( tempdir );
use MaxMind::DB::Writer::Tree;
use Net::Works::Network;

use Test::Requires {
    'MaxMind::DB::Reader' => 0.040000,
};

my $tempdir = tempdir( CLEANUP => 1 );

# make a frozen tree on disk with record size 24
{
    my $tree = MaxMind::DB::Writer::Tree->new(
        ip_version    => 4,
        record_size   => 24,
        database_type => 'Test',
        languages     => ['en'],
        description   => {
            en => 'Test Database',
        },
        map_key_type_callback => sub { },
    );

    $tree->insert_network(
        Net::Works::Network->new_from_string( string => '1.64.22.0/24', ),
        { answer => 42 },
    );

    $tree->insert_network(
        Net::Works::Network->new_from_string( string => '1.64.23.0/24', ),
        { ncc => 1701 },
    );

    $tree->freeze_tree("$tempdir/frozen");
}

# unfreeze the tree setting network size to 32, write it as an mmdb
{
    my $tree = MaxMind::DB::Writer::Tree->new_from_frozen_tree(
        filename              => "$tempdir/frozen",
        map_key_type_callback => sub { 'uint32' },
        record_size           => 32,
    );

    open my $fh, '>:raw', "$tempdir/mmdb" or die $!;
    $tree->write_tree($fh);
    close $fh or die $!;
}

# load the mmdb, check record size is 32 not 24
{
    ## no critic (Modules::RequireExplicitInclusion)
    my $mmdb = MaxMind::DB::Reader->new( file => "$tempdir/mmdb" );
    ## use critic
    my $metadata = $mmdb->metadata;

    is( $metadata->record_size, 32, 'record size' );

    # check that we can still read those IPs correctly
    is_deeply(
        $mmdb->record_for_address('1.64.22.123'),
        { answer => 42 },
        'ip address lookup check 1/2'
    );
    is_deeply(
        $mmdb->record_for_address('1.64.23.123'),
        { ncc => 1701 },
        'ip address lookup check 2/2'
    );
}

done_testing();