import io
import os
import re
import time

from unit.applications.lang.java import TestApplicationJava
from unit.option import option
from unit.utils import public_dir


class TestJavaApplication(TestApplicationJava):
    prerequisites = {'modules': {'java': 'all'}}

    def test_java_conf_error(self, temp_dir, skip_alert):
        skip_alert(
            r'realpath.*failed',
            r'failed to apply new conf',
            r'application setup failed',
        )
        assert 'error' in self.conf(
            {
                "listeners": {"*:7080": {"pass": "applications/app"}},
                "applications": {
                    "app": {
                        "type": self.get_application_type(),
                        "processes": 1,
                        "working_directory": f"{option.test_dir}/java/empty",
                        "webapp": f"{temp_dir}/java",
                        "unit_jars": f"{temp_dir}/no_such_dir",
                    }
                },
            }
        ), 'conf error'

    def test_java_war(self, temp_dir):
        self.load('empty_war')

        assert 'success' in self.conf(
            f'"{temp_dir}/java/empty.war"',
            '/config/applications/empty_war/webapp',
        ), 'configure war'

        assert self.get()['status'] == 200, 'war'

    def test_java_application_cookies(self):
        self.load('cookies')

        headers = self.get(
            headers={
                'Cookie': 'var1=val1; var2=val2',
                'Host': 'localhost',
                'Connection': 'close',
            }
        )['headers']

        assert headers['X-Cookie-1'] == 'val1', 'cookie 1'
        assert headers['X-Cookie-2'] == 'val2', 'cookie 2'

    def test_java_application_filter(self):
        self.load('filter')

        headers = self.get()['headers']

        assert headers['X-Filter-Before'] == '1', 'filter before'
        assert headers['X-Filter-After'] == '1', 'filter after'

        assert (
            self.get(url='/test')['headers']['X-Filter-After'] == '0'
        ), 'filter after 2'

    def test_java_application_get_variables(self):
        self.load('get_params')

        def check_header(header, expect):
            values = header.split(' ')[:-1]
            assert len(values) == len(expect)
            assert set(values) == set(expect)

        headers = self.get(url='/?var1=val1&var2=&var4=val4&var4=foo')[
            'headers'
        ]

        assert headers['X-Var-1'] == 'val1', 'GET variables'
        assert headers['X-Var-2'] == 'true', 'GET variables 2'
        assert headers['X-Var-3'] == 'false', 'GET variables 3'

        check_header(headers['X-Param-Names'], ['var4', 'var2', 'var1'])
        check_header(headers['X-Param-Values'], ['val4', 'foo'])
        check_header(
            headers['X-Param-Map'], ['var2=', 'var1=val1', 'var4=val4,foo']
        )

    def test_java_application_post_variables(self):
        self.load('post_params')

        headers = self.post(
            headers={
                'Content-Type': 'application/x-www-form-urlencoded',
                'Host': 'localhost',
                'Connection': 'close',
            },
            body='var1=val1&var2=',
        )['headers']

        assert headers['X-Var-1'] == 'val1', 'POST variables'
        assert headers['X-Var-2'] == 'true', 'POST variables 2'
        assert headers['X-Var-3'] == 'false', 'POST variables 3'

    def test_java_application_session(self):
        self.load('session')

        headers = self.get(url='/?var1=val1')['headers']
        session_id = headers['X-Session-Id']

        assert headers['X-Var-1'] == 'null', 'variable empty'
        assert headers['X-Session-New'] == 'true', 'session create'

        headers = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            },
            url='/?var1=val2',
        )['headers']

        assert headers['X-Var-1'] == 'val1', 'variable'
        assert headers['X-Session-New'] == 'false', 'session resume'
        assert session_id == headers['X-Session-Id'], 'session same id'

    def test_java_application_session_active(
        self, date_to_sec_epoch, sec_epoch
    ):
        self.load('session_inactive')

        resp = self.get(
            headers={
                'X-Interval': '4',
                'Host': 'localhost',
                'Connection': 'close',
            }
        )
        session_id = resp['headers']['X-Session-Id']

        assert resp['status'] == 200, 'session init'
        assert resp['headers']['X-Session-Interval'] == '4', 'session interval'
        assert (
            abs(
                date_to_sec_epoch(resp['headers']['X-Session-Last-Access-Time'])
                - sec_epoch
            )
            < 5
        ), 'session last access time'

        time.sleep(1)

        resp = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            }
        )

        assert resp['headers']['X-Session-Id'] == session_id, 'session active'

        session_id = resp['headers']['X-Session-Id']

        time.sleep(1)

        resp = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            }
        )

        assert resp['headers']['X-Session-Id'] == session_id, 'session active 2'

        time.sleep(2)

        resp = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            }
        )

        assert resp['headers']['X-Session-Id'] == session_id, 'session active 3'

    def test_java_application_session_inactive(self):
        self.load('session_inactive')

        resp = self.get(
            headers={
                'X-Interval': '1',
                'Host': 'localhost',
                'Connection': 'close',
            }
        )
        session_id = resp['headers']['X-Session-Id']

        time.sleep(3)

        resp = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            }
        )

        assert resp['headers']['X-Session-Id'] != session_id, 'session inactive'

    def test_java_application_session_invalidate(self):
        self.load('session_invalidate')

        resp = self.get()
        session_id = resp['headers']['X-Session-Id']

        resp = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            }
        )

        assert (
            resp['headers']['X-Session-Id'] != session_id
        ), 'session invalidate'

    def test_java_application_session_listeners(self):
        self.load('session_listeners')

        headers = self.get(url='/test?var1=val1')['headers']
        session_id = headers['X-Session-Id']

        assert headers['X-Session-Created'] == session_id, 'session create'
        assert headers['X-Attr-Added'] == 'var1=val1', 'attribute add'

        headers = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            },
            url='/?var1=val2',
        )['headers']

        assert session_id == headers['X-Session-Id'], 'session same id'
        assert headers['X-Attr-Replaced'] == 'var1=val1', 'attribute replace'

        headers = self.get(
            headers={
                'Host': 'localhost',
                'Cookie': f'JSESSIONID={session_id}',
                'Connection': 'close',
            },
            url='/',
        )['headers']

        assert session_id == headers['X-Session-Id'], 'session same id'
        assert headers['X-Attr-Removed'] == 'var1=val2', 'attribute remove'

    def test_java_application_jsp(self):
        self.load('jsp')

        headers = self.get(url='/index.jsp')['headers']

        assert headers['X-Unit-JSP'] == 'ok', 'JSP Ok header'

    def test_java_application_url_pattern(self):
        self.load('url_pattern')

        headers = self.get(url='/foo/bar/index.html')['headers']

        assert headers['X-Id'] == 'servlet1', '#1 Servlet1 request'
        assert (
            headers['X-Request-URI'] == '/foo/bar/index.html'
        ), '#1 request URI'
        assert headers['X-Servlet-Path'] == '/foo/bar', '#1 servlet path'
        assert headers['X-Path-Info'] == '/index.html', '#1 path info'

        headers = self.get(url='/foo/bar/index.bop')['headers']

        assert headers['X-Id'] == 'servlet1', '#2 Servlet1 request'
        assert (
            headers['X-Request-URI'] == '/foo/bar/index.bop'
        ), '#2 request URI'
        assert headers['X-Servlet-Path'] == '/foo/bar', '#2 servlet path'
        assert headers['X-Path-Info'] == '/index.bop', '#2 path info'

        headers = self.get(url='/baz')['headers']

        assert headers['X-Id'] == 'servlet2', '#3 Servlet2 request'
        assert headers['X-Request-URI'] == '/baz', '#3 request URI'
        assert headers['X-Servlet-Path'] == '/baz', '#3 servlet path'
        assert headers['X-Path-Info'] == 'null', '#3 path info'

        headers = self.get(url='/baz/index.html')['headers']

        assert headers['X-Id'] == 'servlet2', '#4 Servlet2 request'
        assert headers['X-Request-URI'] == '/baz/index.html', '#4 request URI'
        assert headers['X-Servlet-Path'] == '/baz', '#4 servlet path'
        assert headers['X-Path-Info'] == '/index.html', '#4 path info'

        headers = self.get(url='/catalog')['headers']

        assert headers['X-Id'] == 'servlet3', '#5 Servlet3 request'
        assert headers['X-Request-URI'] == '/catalog', '#5 request URI'
        assert headers['X-Servlet-Path'] == '/catalog', '#5 servlet path'
        assert headers['X-Path-Info'] == 'null', '#5 path info'

        headers = self.get(url='/catalog/index.html')['headers']

        assert headers['X-Id'] == 'default', '#6 default request'
        assert (
            headers['X-Request-URI'] == '/catalog/index.html'
        ), '#6 request URI'
        assert (
            headers['X-Servlet-Path'] == '/catalog/index.html'
        ), '#6 servlet path'
        assert headers['X-Path-Info'] == 'null', '#6 path info'

        headers = self.get(url='/catalog/racecar.bop')['headers']

        assert headers['X-Id'] == 'servlet4', '#7 servlet4 request'
        assert (
            headers['X-Request-URI'] == '/catalog/racecar.bop'
        ), '#7 request URI'
        assert (
            headers['X-Servlet-Path'] == '/catalog/racecar.bop'
        ), '#7 servlet path'
        assert headers['X-Path-Info'] == 'null', '#7 path info'

        headers = self.get(url='/index.bop')['headers']

        assert headers['X-Id'] == 'servlet4', '#8 servlet4 request'
        assert headers['X-Request-URI'] == '/index.bop', '#8 request URI'
        assert headers['X-Servlet-Path'] == '/index.bop', '#8 servlet path'
        assert headers['X-Path-Info'] == 'null', '#8 path info'

        headers = self.get(url='/foo/baz')['headers']

        assert headers['X-Id'] == 'servlet0', '#9 servlet0 request'
        assert headers['X-Request-URI'] == '/foo/baz', '#9 request URI'
        assert headers['X-Servlet-Path'] == '/foo', '#9 servlet path'
        assert headers['X-Path-Info'] == '/baz', '#9 path info'

        headers = self.get()['headers']

        assert headers['X-Id'] == 'default', '#10 default request'
        assert headers['X-Request-URI'] == '/', '#10 request URI'
        assert headers['X-Servlet-Path'] == '/', '#10 servlet path'
        assert headers['X-Path-Info'] == 'null', '#10 path info'

        headers = self.get(url='/index.bop/')['headers']

        assert headers['X-Id'] == 'default', '#11 default request'
        assert headers['X-Request-URI'] == '/index.bop/', '#11 request URI'
        assert headers['X-Servlet-Path'] == '/index.bop/', '#11 servlet path'
        assert headers['X-Path-Info'] == 'null', '#11 path info'

    def test_java_application_header(self):
        self.load('header')

        headers = self.get()['headers']

        assert headers['X-Set-Utf8-Value'] == '????', 'set Utf8 header value'
        assert headers['X-Set-Utf8-Name-???'] == 'x', 'set Utf8 header name'
        assert headers['X-Add-Utf8-Value'] == '????', 'add Utf8 header value'
        assert headers['X-Add-Utf8-Name-???'] == 'y', 'add Utf8 header name'
        assert headers['X-Add-Test'] == 'v1', 'add null header'
        assert 'X-Set-Test1' not in headers, 'set null header'
        assert headers['X-Set-Test2'] == '', 'set empty header'

    def test_java_application_content_type(self):
        self.load('content_type')

        headers = self.get(url='/1')['headers']

        assert (
            headers['Content-Type'] == 'text/plain;charset=utf-8'
        ), '#1 Content-Type header'
        assert (
            headers['X-Content-Type'] == 'text/plain;charset=utf-8'
        ), '#1 response Content-Type'
        assert headers['X-Character-Encoding'] == 'utf-8', '#1 response charset'

        headers = self.get(url='/2')['headers']

        assert (
            headers['Content-Type'] == 'text/plain;charset=iso-8859-1'
        ), '#2 Content-Type header'
        assert (
            headers['X-Content-Type'] == 'text/plain;charset=iso-8859-1'
        ), '#2 response Content-Type'
        assert (
            headers['X-Character-Encoding'] == 'iso-8859-1'
        ), '#2 response charset'

        headers = self.get(url='/3')['headers']

        assert (
            headers['Content-Type'] == 'text/plain;charset=windows-1251'
        ), '#3 Content-Type header'
        assert (
            headers['X-Content-Type'] == 'text/plain;charset=windows-1251'
        ), '#3 response Content-Type'
        assert (
            headers['X-Character-Encoding'] == 'windows-1251'
        ), '#3 response charset'

        headers = self.get(url='/4')['headers']

        assert (
            headers['Content-Type'] == 'text/plain;charset=windows-1251'
        ), '#4 Content-Type header'
        assert (
            headers['X-Content-Type'] == 'text/plain;charset=windows-1251'
        ), '#4 response Content-Type'
        assert (
            headers['X-Character-Encoding'] == 'windows-1251'
        ), '#4 response charset'

        headers = self.get(url='/5')['headers']

        assert (
            headers['Content-Type'] == 'text/plain;charset=iso-8859-1'
        ), '#5 Content-Type header'
        assert (
            headers['X-Content-Type'] == 'text/plain;charset=iso-8859-1'
        ), '#5 response Content-Type'
        assert (
            headers['X-Character-Encoding'] == 'iso-8859-1'
        ), '#5 response charset'

        headers = self.get(url='/6')['headers']

        assert 'Content-Type' not in headers, '#6 no Content-Type header'
        assert 'X-Content-Type' not in headers, '#6 no response Content-Type'
        assert headers['X-Character-Encoding'] == 'utf-8', '#6 response charset'

        headers = self.get(url='/7')['headers']

        assert (
            headers['Content-Type'] == 'text/plain;charset=utf-8'
        ), '#7 Content-Type header'
        assert (
            headers['X-Content-Type'] == 'text/plain;charset=utf-8'
        ), '#7 response Content-Type'
        assert headers['X-Character-Encoding'] == 'utf-8', '#7 response charset'

        headers = self.get(url='/8')['headers']

        assert (
            headers['Content-Type'] == 'text/html;charset=utf-8'
        ), '#8 Content-Type header'
        assert (
            headers['X-Content-Type'] == 'text/html;charset=utf-8'
        ), '#8 response Content-Type'
        assert headers['X-Character-Encoding'] == 'utf-8', '#8 response charset'

    def test_java_application_welcome_files(self):
        self.load('welcome_files')

        headers = self.get()['headers']

        resp = self.get(url='/dir1')

        assert resp['status'] == 302, 'dir redirect expected'

        resp = self.get(url='/dir1/')

        assert 'This is index.txt.' in resp['body'], 'dir1 index body'
        assert resp['headers']['X-TXT-Filter'] == '1', 'TXT Filter header'

        headers = self.get(url='/dir2/')['headers']

        assert headers['X-Unit-JSP'] == 'ok', 'JSP Ok header'
        assert headers['X-JSP-Filter'] == '1', 'JSP Filter header'

        headers = self.get(url='/dir3/')['headers']

        assert (
            headers['X-App-Servlet'] == '1'
        ), 'URL pattern overrides welcome file'

        headers = self.get(url='/dir4/')['headers']

        assert (
            'X-App-Servlet' not in headers
        ), 'Static welcome file served first'

        headers = self.get(url='/dir5/')['headers']

        assert (
            headers['X-App-Servlet'] == '1'
        ), 'Servlet for welcome file served when no static file found'

    def test_java_application_request_listeners(self):
        self.load('request_listeners')

        headers = self.get(url='/test1')['headers']

        assert (
            headers['X-Request-Initialized'] == '/test1'
        ), 'request initialized event'
        assert headers['X-Request-Destroyed'] == '', 'request destroyed event'
        assert headers['X-Attr-Added'] == '', 'attribute added event'
        assert headers['X-Attr-Removed'] == '', 'attribute removed event'
        assert headers['X-Attr-Replaced'] == '', 'attribute replaced event'

        headers = self.get(url='/test2?var1=1')['headers']

        assert (
            headers['X-Request-Initialized'] == '/test2'
        ), 'request initialized event'
        assert (
            headers['X-Request-Destroyed'] == '/test1'
        ), 'request destroyed event'
        assert headers['X-Attr-Added'] == 'var=1;', 'attribute added event'
        assert headers['X-Attr-Removed'] == 'var=1;', 'attribute removed event'
        assert headers['X-Attr-Replaced'] == '', 'attribute replaced event'

        headers = self.get(url='/test3?var1=1&var2=2')['headers']

        assert (
            headers['X-Request-Initialized'] == '/test3'
        ), 'request initialized event'
        assert (
            headers['X-Request-Destroyed'] == '/test2'
        ), 'request destroyed event'
        assert headers['X-Attr-Added'] == 'var=1;', 'attribute added event'
        assert headers['X-Attr-Removed'] == 'var=2;', 'attribute removed event'
        assert (
            headers['X-Attr-Replaced'] == 'var=1;'
        ), 'attribute replaced event'

        headers = self.get(url='/test4?var1=1&var2=2&var3=3')['headers']

        assert (
            headers['X-Request-Initialized'] == '/test4'
        ), 'request initialized event'
        assert (
            headers['X-Request-Destroyed'] == '/test3'
        ), 'request destroyed event'
        assert headers['X-Attr-Added'] == 'var=1;', 'attribute added event'
        assert headers['X-Attr-Removed'] == '', 'attribute removed event'
        assert (
            headers['X-Attr-Replaced'] == 'var=1;var=2;'
        ), 'attribute replaced event'

    def test_java_application_request_uri_forward(self):
        self.load('forward')

        resp = self.get(
            url='/fwd?uri=%2Fdata%2Ftest%3Furi%3Dnew_uri%26a%3D2%26b%3D3&a=1&c=4'
        )
        headers = resp['headers']

        assert (
            headers['X-REQUEST-Id'] == 'fwd'
        ), 'initial request servlet mapping'
        assert (
            headers['X-Forward-To'] == '/data/test?uri=new_uri&a=2&b=3'
        ), 'forwarding triggered'
        assert (
            headers['X-REQUEST-Param-uri'] == '/data/test?uri=new_uri&a=2&b=3'
        ), 'original uri parameter'
        assert headers['X-REQUEST-Param-a'] == '1', 'original a parameter'
        assert headers['X-REQUEST-Param-c'] == '4', 'original c parameter'

        assert (
            headers['X-FORWARD-Id'] == 'data'
        ), 'forward request servlet mapping'
        assert (
            headers['X-FORWARD-Request-URI'] == '/data/test'
        ), 'forward request uri'
        assert (
            headers['X-FORWARD-Servlet-Path'] == '/data'
        ), 'forward request servlet path'
        assert (
            headers['X-FORWARD-Path-Info'] == '/test'
        ), 'forward request path info'
        assert (
            headers['X-FORWARD-Query-String'] == 'uri=new_uri&a=2&b=3'
        ), 'forward request query string'
        assert (
            headers['X-FORWARD-Param-uri']
            == 'new_uri,/data/test?uri=new_uri&a=2&b=3'
        ), 'forward uri parameter'
        assert headers['X-FORWARD-Param-a'] == '2,1', 'forward a parameter'
        assert headers['X-FORWARD-Param-b'] == '3', 'forward b parameter'
        assert headers['X-FORWARD-Param-c'] == '4', 'forward c parameter'

        assert (
            headers['X-javax.servlet.forward.request_uri'] == '/fwd'
        ), 'original request uri'
        assert (
            headers['X-javax.servlet.forward.context_path'] == ''
        ), 'original request context path'
        assert (
            headers['X-javax.servlet.forward.servlet_path'] == '/fwd'
        ), 'original request servlet path'
        assert (
            headers['X-javax.servlet.forward.path_info'] == 'null'
        ), 'original request path info'
        assert (
            headers['X-javax.servlet.forward.query_string']
            == 'uri=%2Fdata%2Ftest%3Furi%3Dnew_uri%26a%3D2%26b%3D3&a=1&c=4'
        ), 'original request query'

        assert (
            'Before forwarding' not in resp['body']
        ), 'discarded data added before forward() call'
        assert (
            'X-After-Forwarding' not in headers
        ), 'cannot add headers after forward() call'
        assert (
            'After forwarding' not in resp['body']
        ), 'cannot add data after forward() call'

    def test_java_application_named_dispatcher_forward(self):
        self.load('forward')

        resp = self.get(url='/fwd?disp=name&uri=data')
        headers = resp['headers']

        assert (
            headers['X-REQUEST-Id'] == 'fwd'
        ), 'initial request servlet mapping'
        assert headers['X-Forward-To'] == 'data', 'forwarding triggered'

        assert (
            headers['X-FORWARD-Id'] == 'data'
        ), 'forward request servlet mapping'
        assert headers['X-FORWARD-Request-URI'] == '/fwd', 'forward request uri'
        assert (
            headers['X-FORWARD-Servlet-Path'] == '/fwd'
        ), 'forward request servlet path'
        assert (
            headers['X-FORWARD-Path-Info'] == 'null'
        ), 'forward request path info'
        assert (
            headers['X-FORWARD-Query-String'] == 'disp=name&uri=data'
        ), 'forward request query string'

        assert (
            headers['X-javax.servlet.forward.request_uri'] == 'null'
        ), 'original request uri'
        assert (
            headers['X-javax.servlet.forward.context_path'] == 'null'
        ), 'original request context path'
        assert (
            headers['X-javax.servlet.forward.servlet_path'] == 'null'
        ), 'original request servlet path'
        assert (
            headers['X-javax.servlet.forward.path_info'] == 'null'
        ), 'original request path info'
        assert (
            headers['X-javax.servlet.forward.query_string'] == 'null'
        ), 'original request query'

        assert (
            'Before forwarding' not in resp['body']
        ), 'discarded data added before forward() call'
        assert (
            'X-After-Forwarding' not in headers
        ), 'cannot add headers after forward() call'
        assert (
            'After forwarding' not in resp['body']
        ), 'cannot add data after forward() call'

    def test_java_application_request_uri_include(self):
        self.load('include')

        resp = self.get(url='/inc?uri=/data/test')
        headers = resp['headers']
        body = resp['body']

        assert (
            headers['X-REQUEST-Id'] == 'inc'
        ), 'initial request servlet mapping'
        assert headers['X-Include'] == '/data/test', 'including triggered'

        assert (
            'X-INCLUDE-Id' not in headers
        ), 'unable to add headers in include request'

        assert (
            'javax.servlet.include.request_uri:  /data/test' in body
        ), 'include request uri'
        # assert (
        #    'javax.servlet.include.context_path: ' in body
        # ) == True, 'include request context path'
        assert (
            'javax.servlet.include.servlet_path: /data' in body
        ), 'include request servlet path'
        assert (
            'javax.servlet.include.path_info:    /test' in body
        ), 'include request path info'
        assert (
            'javax.servlet.include.query_string: null' in body
        ), 'include request query'

        assert (
            'Before include' in body
        ), 'preserve data added before include() call'
        assert (
            headers['X-After-Include'] == 'you-should-see-this'
        ), 'add headers after include() call'
        assert 'After include' in body, 'add data after include() call'

    def test_java_application_named_dispatcher_include(self):
        self.load('include')

        resp = self.get(url='/inc?disp=name&uri=data')
        headers = resp['headers']
        body = resp['body']

        assert (
            headers['X-REQUEST-Id'] == 'inc'
        ), 'initial request servlet mapping'
        assert headers['X-Include'] == 'data', 'including triggered'

        assert (
            'X-INCLUDE-Id' not in headers
        ), 'unable to add headers in include request'

        assert (
            'javax.servlet.include.request_uri:  null' in body
        ), 'include request uri'
        # assert (
        #    'javax.servlet.include.context_path: null' in body
        # ) == True, 'include request context path'
        assert (
            'javax.servlet.include.servlet_path: null' in body
        ), 'include request servlet path'
        assert (
            'javax.servlet.include.path_info:    null' in body
        ), 'include request path info'
        assert (
            'javax.servlet.include.query_string: null' in body
        ), 'include request query'

        assert (
            'Before include' in body
        ), 'preserve data added before include() call'
        assert (
            headers['X-After-Include'] == 'you-should-see-this'
        ), 'add headers after include() call'
        assert 'After include' in body, 'add data after include() call'

    def test_java_application_path_translation(self):
        self.load('path_translation')

        headers = self.get(url='/pt/test?path=/')['headers']

        assert headers['X-Servlet-Path'] == '/pt', 'matched servlet path'
        assert headers['X-Path-Info'] == '/test', 'the rest of the path'
        assert (
            headers['X-Path-Translated']
            == f"{headers['X-Real-Path']}{headers['X-Path-Info']}"
        ), 'translated path is the app root + path info'
        assert headers['X-Resource-Paths'].endswith(
            '/WEB-INF/, /index.html]'
        ), 'app root directory content'
        assert (
            headers['X-Resource-As-Stream'] == 'null'
        ), 'no resource stream for root path'

        headers = self.get(url='/test?path=/none')['headers']

        assert headers['X-Servlet-Path'] == '/test', 'matched whole path'
        assert (
            headers['X-Path-Info'] == 'null'
        ), 'the rest of the path is null, whole path matched'
        assert (
            headers['X-Path-Translated'] == 'null'
        ), 'translated path is null because path info is null'
        assert headers['X-Real-Path'].endswith('/none'), 'read path is not null'
        assert headers['X-Resource-Paths'] == 'null', 'no resource found'
        assert headers['X-Resource-As-Stream'] == 'null', 'no resource stream'

    def test_java_application_query_string(self):
        self.load('query_string')

        assert (
            self.get(url='/?a=b')['headers']['X-Query-String'] == 'a=b'
        ), 'query string'

    def test_java_application_query_empty(self):
        self.load('query_string')

        assert (
            self.get(url='/?')['headers']['X-Query-String'] == ''
        ), 'query string empty'

    def test_java_application_query_absent(self):
        self.load('query_string')

        assert (
            self.get()['headers']['X-Query-String'] == 'null'
        ), 'query string absent'

    def test_java_application_empty(self):
        self.load('empty')

        assert self.get()['status'] == 200, 'empty'

    def test_java_application_keepalive_body(self):
        self.load('mirror')

        assert self.post()['status'] == 200, 'init'

        body = '0123456789' * 500
        (resp, sock) = self.post(
            headers={
                'Connection': 'keep-alive',
                'Content-Type': 'text/html',
                'Host': 'localhost',
            },
            start=True,
            body=body,
            read_timeout=1,
        )

        assert resp['body'] == body, 'keep-alive 1'

        body = '0123456789'
        resp = self.post(
            headers={
                'Connection': 'close',
                'Content-Type': 'text/html',
                'Host': 'localhost',
            },
            sock=sock,
            body=body,
        )

        assert resp['body'] == body, 'keep-alive 2'

    def test_java_application_http_10(self):
        self.load('empty')

        assert self.get(http_10=True)['status'] == 200, 'HTTP 1.0'

    def test_java_application_no_method(self):
        self.load('empty')

        assert self.post()['status'] == 405, 'no method'

    def test_java_application_get_header(self):
        self.load('get_header')

        assert (
            self.get(
                headers={
                    'X-Header': 'blah',
                    'Content-Type': 'text/html',
                    'Host': 'localhost',
                    'Connection': 'close',
                }
            )['headers']['X-Reply']
            == 'blah'
        ), 'get header'

    def test_java_application_get_header_empty(self):
        self.load('get_header')

        assert 'X-Reply' not in self.get()['headers'], 'get header empty'

    def test_java_application_get_headers(self):
        self.load('get_headers')

        headers = self.get(
            headers={
                'X-Header': ['blah', 'blah'],
                'Content-Type': 'text/html',
                'Host': 'localhost',
                'Connection': 'close',
            }
        )['headers']

        assert headers['X-Reply-0'] == 'blah', 'get headers'
        assert headers['X-Reply-1'] == 'blah', 'get headers 2'

    def test_java_application_get_headers_empty(self):
        self.load('get_headers')

        assert 'X-Reply-0' not in self.get()['headers'], 'get headers empty'

    def test_java_application_get_header_names(self):
        self.load('get_header_names')

        headers = self.get()['headers']

        assert re.search(
            r'(?:Host|Connection)', headers['X-Reply-0']
        ), 'get header names'
        assert re.search(
            r'(?:Host|Connection)', headers['X-Reply-1']
        ), 'get header names 2'
        assert (
            headers['X-Reply-0'] != headers['X-Reply-1']
        ), 'get header names not equal'

    def test_java_application_header_int(self):
        self.load('header_int')

        headers = self.get(
            headers={
                'X-Header': '2',
                'Content-Type': 'text/html',
                'Host': 'localhost',
                'Connection': 'close',
            }
        )['headers']

        assert headers['X-Set-Int'] == '1', 'set int header'
        assert headers['X-Get-Int'] == '2', 'get int header'

    def test_java_application_header_date(self):
        self.load('header_date')

        date = 'Fri, 15 Mar 2019 14:45:34 GMT'

        headers = self.get(
            headers={
                'X-Header': date,
                'Content-Type': 'text/html',
                'Host': 'localhost',
                'Connection': 'close',
            }
        )['headers']

        assert (
            headers['X-Set-Date'] == 'Thu, 01 Jan 1970 00:00:01 GMT'
        ), 'set date header'
        assert headers['X-Get-Date'] == date, 'get date header'

    def test_java_application_multipart(self, search_in_file, temp_dir):
        self.load('multipart')

        reldst = '/uploads'
        fulldst = f'{temp_dir}{reldst}'
        os.mkdir(fulldst)
        public_dir(fulldst)

        fields = {
            'file': {
                'filename': 'sample.txt',
                'type': 'text/plain',
                'data': io.StringIO('Data from sample file'),
            },
            'destination': fulldst,
            'upload': 'Upload',
        }

        encoded, content_type = self.multipart_encode(fields)

        preamble = 'Preamble. Should be ignored.'
        epilogue = 'Epilogue. Should be ignored.'
        body = "%s\r\n%s\r\n%s" % (preamble, encoded.decode(), epilogue)

        resp = self.post(
            headers={
                'Content-Type': content_type,
                'Host': 'localhost',
                'Connection': 'close',
            },
            body=body,
        )

        assert resp['status'] == 200, 'multipart status'
        assert re.search(r'sample\.txt created', resp['body']), 'multipart body'
        assert (
            search_in_file(
                r'^Data from sample file$', name=f'{reldst}/sample.txt'
            )
            is not None
        ), 'file created'

    def test_java_application_threads(self):
        self.load('threads')

        assert 'success' in self.conf(
            '4', 'applications/threads/threads'
        ), 'configure 4 threads'

        socks = []

        for _ in range(4):
            sock = self.get(
                headers={
                    'Host': 'localhost',
                    'X-Delay': '2',
                    'Connection': 'close',
                },
                no_recv=True,
            )

            socks.append(sock)

            time.sleep(0.25)  # required to avoid greedy request reading

        threads = set()

        for sock in socks:
            resp = self.recvall(sock).decode('utf-8')

            self.log_in(resp)

            resp = self._resp_to_dict(resp)

            assert resp['status'] == 200, 'status'

            threads.add(resp['headers']['X-Thread'])

            sock.close()

        assert len(socks) == len(threads), 'threads differs'