From 82943f228ff0879b7fdbf30f6dee6c46df31c126 Mon Sep 17 00:00:00 2001
From: Colin Grogan <grogan.colin@gmail.com>
Date: Tue, 12 Sep 2017 08:06:57 +0100
Subject: [PATCH] Added test for #1922

---
 tests/vibe.web.rest.1922/.gitignore   |   1 +
 tests/vibe.web.rest.1922/dub.json     |   7 ++
 tests/vibe.web.rest.1922/source/app.d | 104 ++++++++++++++++++++++++++
 3 files changed, 112 insertions(+)
 create mode 100644 tests/vibe.web.rest.1922/.gitignore
 create mode 100644 tests/vibe.web.rest.1922/dub.json
 create mode 100644 tests/vibe.web.rest.1922/source/app.d

diff --git a/tests/vibe.web.rest.1922/.gitignore b/tests/vibe.web.rest.1922/.gitignore
new file mode 100644
index 0000000000..922b81834d
--- /dev/null
+++ b/tests/vibe.web.rest.1922/.gitignore
@@ -0,0 +1 @@
+1922
diff --git a/tests/vibe.web.rest.1922/dub.json b/tests/vibe.web.rest.1922/dub.json
new file mode 100644
index 0000000000..df3fcbaf7c
--- /dev/null
+++ b/tests/vibe.web.rest.1922/dub.json
@@ -0,0 +1,7 @@
+{
+	"name": "1922",
+	"dependencies": {
+		"vibe-d": { "path": "../../" }
+	},
+	"versions": ["VibeDefaultMain"]
+}
diff --git a/tests/vibe.web.rest.1922/source/app.d b/tests/vibe.web.rest.1922/source/app.d
new file mode 100644
index 0000000000..97d79b8bd5
--- /dev/null
+++ b/tests/vibe.web.rest.1922/source/app.d
@@ -0,0 +1,104 @@
+import vibe.d;
+import std.datetime;
+import vibe.web.auth;
+
+shared static this()
+{
+	auto settings = new HTTPServerSettings;
+	// 10k + issue number -> Avoid bind errors
+	settings.port = 11922;
+	settings.bindAddresses = ["::1", "127.0.0.1"];
+	settings.sessionStore = new MemorySessionStore();
+
+	auto router = new URLRouter;
+	router.registerRestInterface(new AuthAPI);
+	listenHTTP(settings, router);
+
+	setTimer(1.seconds, {
+		scope(exit) exitEventLoop();
+
+		void test(string endpoint, string user, HTTPStatus expected = HTTPStatus.ok){
+			requestHTTP("http://127.0.0.1:11922"~endpoint, (scope req){
+				if(user !is null)
+					req.headers["AuthUser"] = user;
+			}, (scope res) {
+				assert(res.statusCode == expected, format("Unexpected status code for GET %s (%s): %s\n%s", endpoint, user, res.statusCode,res.readJson));
+			});
+		}
+
+		test("/non_auth_number?num=5", null);
+		test("/non_auth_number?num=5", "admin");
+		test("/auth_number?num=5", "admin");
+		test("/auth_number?num=5", null, HTTPStatus.forbidden);
+		test("/items/name?item=something", "admin");
+		test("/items/name?item=something", null, HTTPStatus.forbidden);
+	});
+}
+
+struct AuthInfo {
+	string name;
+}
+
+interface IItemAPI {
+	struct CollectionIndices {
+		string item;
+	}
+	string getName(string item, AuthInfo info);
+}
+
+@requiresAuth
+class ItemAPI : IItemAPI {
+	@anyAuth
+	string getName(string item, AuthInfo info){
+		return info.name ~ item;
+	}
+
+	@noRoute final mixin CreateAuthFunc;
+}
+
+@requiresAuth
+interface IAuthAPI {
+
+	@noAuth int getNonAuthNumber(int num);
+	@anyAuth int getAuthNumber(AuthInfo info, int num);
+	@anyAuth Collection!IItemAPI items();
+
+	@noRoute final mixin CreateAuthFunc;
+}
+
+class AuthAPI : IAuthAPI {
+	private IItemAPI m_items;
+	this(){
+		m_items = new ItemAPI;
+	}
+
+	Collection!IItemAPI items(){
+		return Collection!IItemAPI(m_items);
+	}
+
+	int getNonAuthNumber(int num){
+		return num;
+	}
+	int getAuthNumber(AuthInfo info, int num){
+		logInfo("Returning auth number for authorized user: %s", info.name);
+		return info.name.length.to!int * num;
+	}
+}
+
+auto getReq(HTTPServerRequest req, HTTPServerResponse _){
+	return req;
+}
+
+auto getRes(HTTPServerRequest _, HTTPServerResponse res){
+	return res;
+}
+
+mixin template CreateAuthFunc(){
+	AuthInfo authenticate(HTTPServerRequest req, HTTPServerResponse res){
+		AuthInfo ret;
+		if("AuthUser" in req.headers && req.headers["AuthUser"]=="admin"){
+			ret.name = "admin";
+		} else throw new HTTPStatusException(HTTPStatus.forbidden, "Forbidden");
+		return ret;
+	}
+}