2 files changed, 418 insertions, 0 deletions
diff --git a/v_windows/v/vlib/vweb/tests/vweb_test.v b/v_windows/v/vlib/vweb/tests/vweb_test.v
new file mode 100644
index 0000000..02908f4
--- /dev/null
+++ b/v_windows/v/vlib/vweb/tests/vweb_test.v
@@ -0,0 +1,300 @@
+import os
+import time
+import json
+import net
+import net.http
+import io
+
+const (
+	sport           = 12380
+	exit_after_time = 12000 // milliseconds
+	vexe            = os.getenv('VEXE')
+	vweb_logfile    = os.getenv('VWEB_LOGFILE')
+	vroot           = os.dir(vexe)
+	serverexe       = os.join_path(os.cache_dir(), 'vweb_test_server.exe')
+	tcp_r_timeout   = 30 * time.second
+	tcp_w_timeout   = 30 * time.second
+)
+
+// setup of vweb webserver
+fn testsuite_begin() {
+	os.chdir(vroot) or {}
+	if os.exists(serverexe) {
+		os.rm(serverexe) or {}
+	}
+}
+
+fn test_a_simple_vweb_app_can_be_compiled() {
+	// did_server_compile := os.system('$vexe -g -o $serverexe vlib/vweb/tests/vweb_test_server.v')
+	// TODO: find out why it does not compile with -usecache and -g
+	did_server_compile := os.system('$vexe -o $serverexe vlib/vweb/tests/vweb_test_server.v')
+	assert did_server_compile == 0
+	assert os.exists(serverexe)
+}
+
+fn test_a_simple_vweb_app_runs_in_the_background() {
+	mut suffix := ''
+	$if !windows {
+		suffix = ' > /dev/null &'
+	}
+	if vweb_logfile != '' {
+		suffix = ' 2>> $vweb_logfile >> $vweb_logfile &'
+	}
+	server_exec_cmd := '$serverexe $sport $exit_after_time $suffix'
+	$if debug_net_socket_client ? {
+		eprintln('running:\n$server_exec_cmd')
+	}
+	$if windows {
+		go os.system(server_exec_cmd)
+	} $else {
+		res := os.system(server_exec_cmd)
+		assert res == 0
+	}
+	$if macos {
+		time.sleep(1000 * time.millisecond)
+	} $else {
+		time.sleep(100 * time.millisecond)
+	}
+}
+
+// web client tests follow
+fn assert_common_headers(received string) {
+	assert received.starts_with('HTTP/1.1 200 OK\r\n')
+	assert received.contains('Server: VWeb\r\n')
+	assert received.contains('Content-Length:')
+	assert received.contains('Connection: close\r\n')
+}
+
+fn test_a_simple_tcp_client_can_connect_to_the_vweb_server() {
+	received := simple_tcp_client(path: '/') or {
+		assert err.msg == ''
+		return
+	}
+	assert_common_headers(received)
+	assert received.contains('Content-Type: text/plain')
+	assert received.contains('Content-Length: 15')
+	assert received.ends_with('Welcome to VWeb')
+}
+
+fn test_a_simple_tcp_client_simple_route() {
+	received := simple_tcp_client(path: '/simple') or {
+		assert err.msg == ''
+		return
+	}
+	assert_common_headers(received)
+	assert received.contains('Content-Type: text/plain')
+	assert received.contains('Content-Length: 15')
+	assert received.ends_with('A simple result')
+}
+
+fn test_a_simple_tcp_client_zero_content_length() {
+	// tests that sending a content-length header of 0 doesn't hang on a read timeout
+	watch := time.new_stopwatch(auto_start: true)
+	simple_tcp_client(path: '/', headers: 'Content-Length: 0\r\n\r\n') or {
+		assert err.msg == ''
+		return
+	}
+	assert watch.elapsed() < 1 * time.second
+}
+
+fn test_a_simple_tcp_client_html_page() {
+	received := simple_tcp_client(path: '/html_page') or {
+		assert err.msg == ''
+		return
+	}
+	assert_common_headers(received)
+	assert received.contains('Content-Type: text/html')
+	assert received.ends_with('<h1>ok</h1>')
+}
+
+// net.http client based tests follow:
+fn assert_common_http_headers(x http.Response) ? {
+	assert x.status() == .ok
+	assert x.header.get(.server) ? == 'VWeb'
+	assert x.header.get(.content_length) ?.int() > 0
+	assert x.header.get(.connection) ? == 'close'
+}
+
+fn test_http_client_index() ? {
+	x := http.get('http://127.0.0.1:$sport/') or { panic(err) }
+	assert_common_http_headers(x) ?
+	assert x.header.get(.content_type) ? == 'text/plain'
+	assert x.text == 'Welcome to VWeb'
+}
+
+fn test_http_client_404() ? {
+	url_404_list := [
+		'http://127.0.0.1:$sport/zxcnbnm',
+		'http://127.0.0.1:$sport/JHKAJA',
+		'http://127.0.0.1:$sport/unknown',
+	]
+	for url in url_404_list {
+		res := http.get(url) or { panic(err) }
+		assert res.status() == .not_found
+	}
+}
+
+fn test_http_client_simple() ? {
+	x := http.get('http://127.0.0.1:$sport/simple') or { panic(err) }
+	assert_common_http_headers(x) ?
+	assert x.header.get(.content_type) ? == 'text/plain'
+	assert x.text == 'A simple result'
+}
+
+fn test_http_client_html_page() ? {
+	x := http.get('http://127.0.0.1:$sport/html_page') or { panic(err) }
+	assert_common_http_headers(x) ?
+	assert x.header.get(.content_type) ? == 'text/html'
+	assert x.text == '<h1>ok</h1>'
+}
+
+fn test_http_client_settings_page() ? {
+	x := http.get('http://127.0.0.1:$sport/bilbo/settings') or { panic(err) }
+	assert_common_http_headers(x) ?
+	assert x.text == 'username: bilbo'
+	//
+	y := http.get('http://127.0.0.1:$sport/kent/settings') or { panic(err) }
+	assert_common_http_headers(y) ?
+	assert y.text == 'username: kent'
+}
+
+fn test_http_client_user_repo_settings_page() ? {
+	x := http.get('http://127.0.0.1:$sport/bilbo/gostamp/settings') or { panic(err) }
+	assert_common_http_headers(x) ?
+	assert x.text == 'username: bilbo | repository: gostamp'
+	//
+	y := http.get('http://127.0.0.1:$sport/kent/golang/settings') or { panic(err) }
+	assert_common_http_headers(y) ?
+	assert y.text == 'username: kent | repository: golang'
+	//
+	z := http.get('http://127.0.0.1:$sport/missing/golang/settings') or { panic(err) }
+	assert z.status() == .not_found
+}
+
+struct User {
+	name string
+	age  int
+}
+
+fn test_http_client_json_post() ? {
+	ouser := User{
+		name: 'Bilbo'
+		age: 123
+	}
+	json_for_ouser := json.encode(ouser)
+	mut x := http.post_json('http://127.0.0.1:$sport/json_echo', json_for_ouser) or { panic(err) }
+	$if debug_net_socket_client ? {
+		eprintln('/json_echo endpoint response: $x')
+	}
+	assert x.header.get(.content_type) ? == 'application/json'
+	assert x.text == json_for_ouser
+	nuser := json.decode(User, x.text) or { User{} }
+	assert '$ouser' == '$nuser'
+	//
+	x = http.post_json('http://127.0.0.1:$sport/json', json_for_ouser) or { panic(err) }
+	$if debug_net_socket_client ? {
+		eprintln('/json endpoint response: $x')
+	}
+	assert x.header.get(.content_type) ? == 'application/json'
+	assert x.text == json_for_ouser
+	nuser2 := json.decode(User, x.text) or { User{} }
+	assert '$ouser' == '$nuser2'
+}
+
+fn test_http_client_multipart_form_data() ? {
+	boundary := '6844a625b1f0b299'
+	name := 'foo'
+	ct := 'multipart/form-data; boundary=------------------------$boundary'
+	contents := 'baz buzz'
+	data := "--------------------------$boundary
+Content-Disposition: form-data; name=\"$name\"
+
+$contents
+--------------------------$boundary--
+"
+	mut x := http.fetch(
+		url: 'http://127.0.0.1:$sport/form_echo'
+		method: .post
+		header: http.new_header(
+			key: .content_type
+			value: ct
+		)
+		data: data
+	) ?
+	$if debug_net_socket_client ? {
+		eprintln('/form_echo endpoint response: $x')
+	}
+	assert x.text == contents
+}
+
+fn test_http_client_shutdown_does_not_work_without_a_cookie() {
+	x := http.get('http://127.0.0.1:$sport/shutdown') or {
+		assert err.msg == ''
+		return
+	}
+	assert x.status() == .not_found
+	assert x.text == '404 Not Found'
+}
+
+fn testsuite_end() {
+	// This test is guaranteed to be called last.
+	// It sends a request to the server to shutdown.
+	x := http.fetch(
+		url: 'http://127.0.0.1:$sport/shutdown'
+		method: .get
+		cookies: {
+			'skey': 'superman'
+		}
+	) or {
+		assert err.msg == ''
+		return
+	}
+	assert x.status() == .ok
+	assert x.text == 'good bye'
+}
+
+// utility code:
+struct SimpleTcpClientConfig {
+	retries int    = 20
+	host    string = 'static.dev'
+	path    string = '/'
+	agent   string = 'v/net.tcp.v'
+	headers string = '\r\n'
+	content string
+}
+
+fn simple_tcp_client(config SimpleTcpClientConfig) ?string {
+	mut client := &net.TcpConn(0)
+	mut tries := 0
+	for tries < config.retries {
+		tries++
+		client = net.dial_tcp('127.0.0.1:$sport') or {
+			if tries > config.retries {
+				return err
+			}
+			time.sleep(100 * time.millisecond)
+			continue
+		}
+		break
+	}
+	client.set_read_timeout(tcp_r_timeout)
+	client.set_write_timeout(tcp_w_timeout)
+	defer {
+		client.close() or {}
+	}
+	message := 'GET $config.path HTTP/1.1
+Host: $config.host
+User-Agent: $config.agent
+Accept: */*
+$config.headers
+$config.content'
+	$if debug_net_socket_client ? {
+		eprintln('sending:\n$message')
+	}
+	client.write(message.bytes()) ?
+	read := io.read_all(reader: client) ?
+	$if debug_net_socket_client ? {
+		eprintln('received:\n$read')
+	}
+	return read.bytestr()
+}
diff --git a/v_windows/v/vlib/vweb/tests/vweb_test_server.v b/v_windows/v/vlib/vweb/tests/vweb_test_server.v
new file mode 100644
index 0000000..4dfeb7d
--- /dev/null
+++ b/v_windows/v/vlib/vweb/tests/vweb_test_server.v
@@ -0,0 +1,118 @@
+module main
+
+import os
+import vweb
+import time
+
+const (
+	known_users = ['bilbo', 'kent']
+)
+
+struct App {
+	vweb.Context
+	port          int
+	timeout       int
+	global_config shared Config
+}
+
+struct Config {
+	max_ping int
+}
+
+fn exit_after_timeout(timeout_in_ms int) {
+	time.sleep(timeout_in_ms * time.millisecond)
+	// eprintln('webserver is exiting ...')
+	exit(0)
+}
+
+fn main() {
+	if os.args.len != 3 {
+		panic('Usage: `vweb_test_server.exe PORT TIMEOUT_IN_MILLISECONDS`')
+	}
+	http_port := os.args[1].int()
+	assert http_port > 0
+	timeout := os.args[2].int()
+	assert timeout > 0
+	go exit_after_timeout(timeout)
+	//
+	shared config := &Config{
+		max_ping: 50
+	}
+	app := &App{
+		port: http_port
+		timeout: timeout
+		global_config: config
+	}
+	eprintln('>> webserver: started on http://127.0.0.1:$app.port/ , with maximum runtime of $app.timeout milliseconds.')
+	// vweb.run<App>(mut app, http_port)
+	vweb.run(app, http_port)
+}
+
+// pub fn (mut app App) init_server() {
+//}
+
+pub fn (mut app App) index() vweb.Result {
+	assert app.global_config.max_ping == 50
+	return app.text('Welcome to VWeb')
+}
+
+pub fn (mut app App) simple() vweb.Result {
+	return app.text('A simple result')
+}
+
+pub fn (mut app App) html_page() vweb.Result {
+	return app.html('<h1>ok</h1>')
+}
+
+// the following serve custom routes
+['/:user/settings']
+pub fn (mut app App) settings(username string) vweb.Result {
+	if username !in known_users {
+		return app.not_found()
+	}
+	return app.html('username: $username')
+}
+
+['/:user/:repo/settings']
+pub fn (mut app App) user_repo_settings(username string, repository string) vweb.Result {
+	if username !in known_users {
+		return app.not_found()
+	}
+	return app.html('username: $username | repository: $repository')
+}
+
+['/json_echo'; post]
+pub fn (mut app App) json_echo() vweb.Result {
+	// eprintln('>>>>> received http request at /json_echo is: $app.req')
+	app.set_content_type(app.req.header.get(.content_type) or { '' })
+	return app.ok(app.req.data)
+}
+
+['/form_echo'; post]
+pub fn (mut app App) form_echo() vweb.Result {
+	app.set_content_type(app.req.header.get(.content_type) or { '' })
+	return app.ok(app.form['foo'])
+}
+
+// Make sure [post] works without the path
+[post]
+pub fn (mut app App) json() vweb.Result {
+	// eprintln('>>>>> received http request at /json is: $app.req')
+	app.set_content_type(app.req.header.get(.content_type) or { '' })
+	return app.ok(app.req.data)
+}
+
+pub fn (mut app App) shutdown() vweb.Result {
+	session_key := app.get_cookie('skey') or { return app.not_found() }
+	if session_key != 'superman' {
+		return app.not_found()
+	}
+	go app.gracefull_exit()
+	return app.ok('good bye')
+}
+
+fn (mut app App) gracefull_exit() {
+	eprintln('>> webserver: gracefull_exit')
+	time.sleep(100 * time.millisecond)
+	exit(0)
+}