import io
import os
import re
import time
from unit.applications.lang.java import ApplicationJava
from unit.option import option
from unit.utils import public_dir
prerequisites = {'modules': {'java': 'all'}}
client = ApplicationJava()
def test_java_conf_error(temp_dir, skip_alert):
skip_alert(
r'realpath.*failed',
r'failed to apply new conf',
r'application setup failed',
)
assert 'error' in client.conf(
{
"listeners": {"*:7080": {"pass": "applications/app"}},
"applications": {
"app": {
"type": client.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(temp_dir):
client.load('empty_war')
assert 'success' in client.conf(
f'"{temp_dir}/java/empty.war"',
'/config/applications/empty_war/webapp',
), 'configure war'
assert client.get()['status'] == 200, 'war'
def test_java_application_cookies():
client.load('cookies')
headers = client.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():
client.load('filter')
headers = client.get()['headers']
assert headers['X-Filter-Before'] == '1', 'filter before'
assert headers['X-Filter-After'] == '1', 'filter after'
assert (
client.get(url='/test')['headers']['X-Filter-After'] == '0'
), 'filter after 2'
def test_java_application_get_variables():
client.load('get_params')
def check_header(header, expect):
values = header.split(' ')[:-1]
assert len(values) == len(expect)
assert set(values) == set(expect)
headers = client.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():
client.load('post_params')
headers = client.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():
client.load('session')
headers = client.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 = client.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(date_to_sec_epoch, sec_epoch):
client.load('session_inactive')
resp = client.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 = client.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 = client.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 = client.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():
client.load('session_inactive')
resp = client.get(
headers={
'X-Interval': '1',
'Host': 'localhost',
'Connection': 'close',
}
)
session_id = resp['headers']['X-Session-Id']
time.sleep(3)
resp = client.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():
client.load('session_invalidate')
resp = client.get()
session_id = resp['headers']['X-Session-Id']
resp = client.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():
client.load('session_listeners')
headers = client.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 = client.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 = client.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():
client.load('jsp')
headers = client.get(url='/index.jsp')['headers']
assert headers['X-Unit-JSP'] == 'ok', 'JSP Ok header'
def test_java_application_url_pattern():
client.load('url_pattern')
headers = client.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 = client.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 = client.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 = client.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 = client.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 = client.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 = client.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 = client.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 = client.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 = client.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 = client.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():
client.load('header')
headers = client.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():
client.load('content_type')
headers = client.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 = client.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 = client.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 = client.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 = client.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 = client.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 = client.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 = client.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():
client.load('welcome_files')
headers = client.get()['headers']
resp = client.get(url='/dir1')
assert resp['status'] == 302, 'dir redirect expected'
resp = client.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 = client.get(url='/dir2/')['headers']
assert headers['X-Unit-JSP'] == 'ok', 'JSP Ok header'
assert headers['X-JSP-Filter'] == '1', 'JSP Filter header'
headers = client.get(url='/dir3/')['headers']
assert headers['X-App-Servlet'] == '1', 'URL pattern overrides welcome file'
headers = client.get(url='/dir4/')['headers']
assert 'X-App-Servlet' not in headers, 'Static welcome file served first'
headers = client.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():
client.load('request_listeners')
headers = client.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 = client.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 = client.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 = client.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():
client.load('forward')
resp = client.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():
client.load('forward')
resp = client.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():
client.load('include')
resp = client.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():
client.load('include')
resp = client.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():
client.load('path_translation')
headers = client.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 = client.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():
client.load('query_string')
assert (
client.get(url='/?a=b')['headers']['X-Query-String'] == 'a=b'
), 'query string'
def test_java_application_query_empty():
client.load('query_string')
assert (
client.get(url='/?')['headers']['X-Query-String'] == ''
), 'query string empty'
def test_java_application_query_absent():
client.load('query_string')
assert (
client.get()['headers']['X-Query-String'] == 'null'
), 'query string absent'
def test_java_application_empty():
client.load('empty')
assert client.get()['status'] == 200, 'empty'
def test_java_application_keepalive_body():
client.load('mirror')
assert client.post()['status'] == 200, 'init'
body = '0123456789' * 500
(resp, sock) = client.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 = client.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():
client.load('empty')
assert client.get(http_10=True)['status'] == 200, 'HTTP 1.0'
def test_java_application_no_method():
client.load('empty')
assert client.post()['status'] == 405, 'no method'
def test_java_application_get_header():
client.load('get_header')
assert (
client.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():
client.load('get_header')
assert 'X-Reply' not in client.get()['headers'], 'get header empty'
def test_java_application_get_headers():
client.load('get_headers')
headers = client.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_many_headers():
client.load('get_headers')
value = '0123456789' * 10
headers = client.get(
headers={
'X-Header': [value] * 100,
'Content-Type': 'text/html',
'Host': 'localhost',
'Connection': 'close',
}
)['headers']
for i in range(0, 99):
assert headers[f'X-Reply-{i}'] == value, 'many headers'
def test_java_application_get_headers_empty():
client.load('get_headers')
assert 'X-Reply-0' not in client.get()['headers'], 'get headers empty'
def test_java_application_get_header_names():
client.load('get_header_names')
headers = client.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():
client.load('header_int')
headers = client.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():
client.load('header_date')
date = 'Fri, 15 Mar 2019 14:45:34 GMT'
headers = client.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(search_in_file, temp_dir):
client.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 = client.multipart_encode(fields)
preamble = 'Preamble. Should be ignored.'
epilogue = 'Epilogue. Should be ignored.'
body = f'{preamble}\r\n{encoded.decode()}\r\n{epilogue}'
resp = client.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():
client.load('threads')
assert 'success' in client.conf(
'4', 'applications/threads/threads'
), 'configure 4 threads'
socks = []
for _ in range(4):
sock = client.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 = client.recvall(sock).decode('utf-8')
client.log_in(resp)
resp = client._resp_to_dict(resp)
assert resp['status'] == 200, 'status'
threads.add(resp['headers']['X-Thread'])
sock.close()
assert len(socks) == len(threads), 'threads differs'
