Mercurial > hg > nginx-tests
view stream_geoip.t @ 1427:eb1d883305ea
Tests: avoid edge cases in upstream random two test.
Unavailable servers contribute to the number of attempts, if selected,
before the balancer would fall back to the default round-robin method.
This means that it's quite possible to get server with more connections.
To facilitate with selecting two alive servers, down server was removed
from the upstream configuration at the cost of slightly worse coverage.
author | Sergey Kandaurov <pluknet@nginx.com> |
---|---|
date | Thu, 10 Jan 2019 17:42:34 +0300 |
parents | 3fc6817cd84a |
children | f3ba4c74de31 |
line wrap: on
line source
#!/usr/bin/perl # (C) Andrey Zelenkov # (C) Nginx, Inc. # Tests for stream geoip module. ############################################################################### use warnings; use strict; use Test::More; use Socket qw/ $CRLF /; BEGIN { use FindBin; chdir($FindBin::Bin); } use lib 'lib'; use Test::Nginx; use Test::Nginx::Stream qw/ stream /; ############################################################################### select STDERR; $| = 1; select STDOUT; $| = 1; my $t = Test::Nginx->new()->has(qw/stream stream_geoip stream_return/) ->has('stream_realip'); $t->write_file_expand('nginx.conf', <<'EOF'); %%TEST_GLOBALS%% daemon off; events { } stream { set_real_ip_from 127.0.0.1/32; geoip_country %%TESTDIR%%/country.dat; geoip_city %%TESTDIR%%/city.dat; geoip_org %%TESTDIR%%/org.dat; server { listen 127.0.0.1:8080 proxy_protocol; return "country_code:$geoip_country_code country_code3:$geoip_country_code3 country_name:$geoip_country_name area_code:$geoip_area_code city_continent_code:$geoip_city_continent_code city_country_code:$geoip_city_country_code city_country_code3:$geoip_city_country_code3 city_country_name:$geoip_city_country_name dma_code:$geoip_dma_code latitude:$geoip_latitude longitude:$geoip_longitude region:$geoip_region region_name:$geoip_region_name city:$geoip_city postal_code:$geoip_postal_code org:$geoip_org"; } } EOF my $d = $t->testdir(); # country database: # # "10.0.0.1","10.0.0.1","RU","Russian Federation" # "2001:db8::","2001:db8::","US","United States" my $data = ''; for my $i (0 .. 156) { # skip to offset 32 if 1st bit set in ipv6 address wins $data .= pack_node($i + 1) . pack_node(32), next if $i == 2; # otherwise default to RU $data .= pack_node(0xffffb9) . pack_node(0xffff00), next if $i == 31; # continue checking bits set in ipv6 address $data .= pack_node(0xffff00) . pack_node($i + 1), next if grep $_ == $i, (44, 49, 50, 52, 53, 55, 56, 57); # last bit set in ipv6 address $data .= pack_node(0xffffe1) . pack_node(0xffff00), next if $i == 156; $data .= pack_node($i + 1) . pack_node(0xffff00); } $data .= chr(0x00) x 3; $data .= chr(0xFF) x 3; $data .= chr(12); $t->write_file('country.dat', $data); # city database: # # "167772161","167772161","RU","48","Moscow","119034","55.7543",37.6202",, $data = ''; for my $i (0 .. 31) { $data .= pack_node(32) . pack_node($i + 1), next if $i == 4 or $i == 6; $data .= pack_node(32) . pack_node($i + 2), next if $i == 31; $data .= pack_node($i + 1) . pack_node(32); } $data .= chr(42); $data .= chr(185); $data .= pack('Z*', 48); $data .= pack('Z*', 'Moscow'); $data .= pack('Z*', 119034); $data .= pack_node(int((55.7543 + 180) * 10000)); $data .= pack_node(int((37.6202 + 180) * 10000)); $data .= chr(0) x 3; $data .= chr(0xFF) x 3; $data .= chr(2); $data .= pack_node(32); $t->write_file('city.dat', $data); # organization database: # # "167772161","167772161","Nginx" $data = ''; for my $i (0 .. 31) { $data .= pack_org(32) . pack_org($i + 1), next if $i == 4 or $i == 6; $data .= pack_org(32) . pack_org($i + 2), next if $i == 31; $data .= pack_org($i + 1) . pack_org(32); } $data .= chr(42); $data .= pack('Z*', 'Nginx'); $data .= chr(0xFF) x 3; $data .= chr(5); $data .= pack_node(32); $t->write_file('org.dat', $data); $t->try_run('no inet6 support')->plan(20); ############################################################################### my %data = stream_pp('10.0.0.1') =~ /(\w+):(.*)/g; is($data{country_code}, 'RU', 'geoip country code'); is($data{country_code3}, 'RUS', 'geoip country code 3'); is($data{country_name}, 'Russian Federation', 'geoip country name'); is($data{area_code}, 0, 'geoip area code'); is($data{city_continent_code}, 'EU', 'geoip city continent code'); is($data{city_country_code}, 'RU', 'geoip city country code'); is($data{city_country_code3}, 'RUS', 'geoip city country code 3'); is($data{city_country_name}, 'Russian Federation', 'geoip city country name'); is($data{dma_code}, 0, 'geoip dma code'); is($data{latitude}, 55.7543, 'geoip latitude'); is($data{longitude}, 37.6202, 'geoip longitude'); is($data{region}, 48, 'geoip region'); is($data{region_name}, 'Moscow City', 'geoip region name'); is($data{city}, 'Moscow', 'geoip city'); is($data{postal_code}, 119034, 'geoip postal code'); is($data{org}, 'Nginx', 'geoip org'); like(stream_pp('::ffff:10.0.0.1'), qr/org:Nginx/, 'geoip ipv6 ipv4-mapped'); %data = stream_pp('2001:db8::') =~ /(\w+):(.*)/g; is($data{country_code}, 'US', 'geoip ipv6 country code'); is($data{country_code3}, 'USA', 'geoip ipv6 country code 3'); is($data{country_name}, 'United States', 'geoip ipv6 country name'); ############################################################################### sub stream_pp { my ($ip) = @_; my $type = ($ip =~ ':' ? 'TCP6' : 'TCP4'); return stream('127.0.0.1:' . port(8080)) ->io("PROXY $type $ip 127.0.0.1 8080 8080${CRLF}"); } sub pack_node { substr pack('V', shift), 0, 3; } sub pack_org { pack('V', shift); } ###############################################################################