#!/usr/bin/perl

# (C) Dmitry Volyntsev
# (C) Nginx, Inc.

# Tests for stream njs module, js_periodic directive.

###############################################################################

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/http rewrite stream/)
	->write_file_expand('nginx.conf', <<'EOF');

%%TEST_GLOBALS%%

daemon off;
worker_processes 4;

events {
}

worker_shutdown_timeout 100ms;

stream {
    %%TEST_GLOBALS_STREAM%%

    js_import test.js;

    js_shared_dict_zone zone=nums:32k type=number;
    js_shared_dict_zone zone=strings:32k;
    js_shared_dict_zone zone=workers:32k type=number;

    js_set $js_set  test.js_set;
    js_var $js_var  JS-VAR;
    map _ $map_var {
        default "MAP-VAR";
    }

    server {
        listen       127.0.0.1:8081;

        js_periodic test.tick interval=30ms jitter=1ms;
        js_periodic test.timer interval=1s worker_affinity=all;
        js_periodic test.overrun interval=30ms;
        js_periodic test.affinity interval=50ms worker_affinity=0101;
        js_periodic test.vars interval=10s;

        js_periodic test.tick_exception interval=1s;
        js_periodic test.timer_exception interval=1s;
        js_periodic test.timeout_exception interval=30ms;

        js_preread test.test;

        proxy_pass 127.0.0.1:8090;
    }
}

EOF

$t->write_file('test.js', <<EOF);
    function affinity() {
        ngx.shared.workers.set(ngx.worker_id, 1);
    }

    function js_set() {
        return 'JS-SET';
    }

    async function overrun() {
        setTimeout(() => {}, 100000);
    }

    function tick() {
        ngx.shared.nums.incr('tick', 1);
    }

    function tick_exception() {
        throw new Error("EXCEPTION");
    }

    function timer() {
        if (ngx.worker_id != 0) {
            return;
        }

        setTimeout(() => {ngx.shared.nums.set('timer', 1)}, 10);
    }

    function timer_exception() {
        setTimeout(() => {ngx.log(ngx.ERR, 'should not be seen')}, 10);
        throw new Error("EXCEPTION");
    }

    function timeout_exception() {
        setTimeout(() => {
            var v = ngx.shared.nums.get('timeout_exception') || 0;

            if (v == 0) {
                ngx.shared.nums.set('timeout_exception', 1);
                throw new Error("EXCEPTION");
                return;
            }

            ngx.shared.nums.incr('timeout_exception', 1);
        }, 1);
    }

    function vars(s) {
        var v = s.variables;
        ngx.shared.strings.set('vars',
                               `\${v.js_var}|\${v.js_set}|\${v.map_var}`);
    }

    function test(s) {
        s.on('upload', function (data) {
            if (data.length > 0) {
                switch (data) {
                case 'affinity':
                    if (ngx.shared.workers.keys().toSorted().toString()
                        == '1,3')
                    {
                        s.done();
                        return;
                    }

                    break;

                case 'tick':
                    if (ngx.shared.nums.get('tick') >= 3) {
                        s.done();
                        return;
                    }

                    break;

                case 'timeout_exception':
                    if (ngx.shared.nums.get('timeout_exception') >= 2) {
                        s.done();
                        return;
                    }

                    break;

                case 'timer':
                    if (ngx.shared.nums.get('timer') == 1) {
                        s.done();
                        return;
                    }

                    break;

                case 'vars':
                    var vars = ngx.shared.strings.get('vars');
                    if (vars === 'JS-VAR|JS-SET|MAP-VAR') {
                        s.done();
                        return;
                    }

                    break;

                default:
                    throw new Error(`Unknown test "\${data}"`);
                }

                throw new Error(`Test "\${data}" failed`);
            }
        });
    }

    export default { affinity, js_set,  overrun, test, tick, tick_exception,
                     timer, timer_exception, timeout_exception, vars };
EOF

$t->run_daemon(\&stream_daemon, port(8090));
$t->try_run('no js_periodic');
$t->plan(6);
$t->waitforsocket('127.0.0.1:' . port(8090));

###############################################################################

select undef, undef, undef, 0.2;

is(stream('127.0.0.1:' . port(8081))->io('affinity'), 'affinity',
	'affinity test');
is(stream('127.0.0.1:' . port(8081))->io('tick'), 'tick', '3x tick test');
is(stream('127.0.0.1:' . port(8081))->io('timer'), 'timer', 'timer test');
is(stream('127.0.0.1:' . port(8081))->io('timeout_exception'),
	'timeout_exception', 'timeout exception test');
is(stream('127.0.0.1:' . port(8081))->io('vars'), 'vars', 'vars test');

$t->stop();

unlike($t->read_file('error.log'), qr/\[error\].*should not be seen/,
	'check for not discadred events');

###############################################################################

sub stream_daemon {
	my $server = IO::Socket::INET->new(
		Proto => 'tcp',
		LocalAddr => '127.0.0.1:' . port(8090),
		Listen => 5,
		Reuse => 1
	)
		or die "Can't create listening socket: $!\n";

	local $SIG{PIPE} = 'IGNORE';

	while (my $client = $server->accept()) {
		$client->autoflush(1);

		log2c("(new connection $client)");

		$client->sysread(my $buffer, 65536) or next;

		log2i("$client $buffer");

		log2o("$client $buffer");

		$client->syswrite($buffer);

		close $client;
	}
}

sub log2i { Test::Nginx::log_core('|| <<', @_); }
sub log2o { Test::Nginx::log_core('|| >>', @_); }
sub log2c { Test::Nginx::log_core('||', @_); }

###############################################################################