From f512a56fafe2758ec237ac71743cec3570ca4240 Mon Sep 17 00:00:00 2001
From: Thomas LEVEIL <thomasleveil@gmail.com>
Date: Fri, 27 Jan 2017 02:01:57 +0100
Subject: [PATCH] TESTS: add test for custom per vhost configuration

---
 test2/test_custom/test_custom-per-vhost.py  | 19 +++++++++++++++++
 test2/test_custom/test_custom-per-vhost.yml | 23 +++++++++++++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 test2/test_custom/test_custom-per-vhost.py
 create mode 100644 test2/test_custom/test_custom-per-vhost.yml

diff --git a/test2/test_custom/test_custom-per-vhost.py b/test2/test_custom/test_custom-per-vhost.py
new file mode 100644
index 0000000..57c3bca
--- /dev/null
+++ b/test2/test_custom/test_custom-per-vhost.py
@@ -0,0 +1,19 @@
+import pytest
+
+def test_custom_conf_does_not_apply_to_unknown_vhost(docker_compose, nginxproxy):
+    r = nginxproxy.get("http://nginx-proxy/")
+    assert r.status_code == 503
+    assert "X-test" not in r.headers
+
+def test_custom_conf_applies_to_web1(docker_compose, nginxproxy):
+    r = nginxproxy.get("http://web1.nginx-proxy.local/port")
+    assert r.status_code == 200   
+    assert r.text == "answer from port 81\n"
+    assert "X-test" in r.headers
+    assert "f00" == r.headers["X-test"]
+
+def test_custom_conf_does_not_apply_to_web2(docker_compose, nginxproxy):
+    r = nginxproxy.get("http://web2.nginx-proxy.local/port")
+    assert r.status_code == 200   
+    assert r.text == "answer from port 82\n"
+    assert "X-test" not in r.headers
diff --git a/test2/test_custom/test_custom-per-vhost.yml b/test2/test_custom/test_custom-per-vhost.yml
new file mode 100644
index 0000000..a99da1d
--- /dev/null
+++ b/test2/test_custom/test_custom-per-vhost.yml
@@ -0,0 +1,23 @@
+version: '2'
+services:
+  nginx-proxy:
+    image: jwilder/nginx-proxy:test
+    volumes:
+      - /var/run/docker.sock:/tmp/docker.sock:ro
+      - ./my_custom_proxy_settings.conf:/etc/nginx/vhost.d/web1.nginx-proxy.local:ro
+
+  web1:
+    image: web 
+    expose:
+      - "81"
+    environment:
+      WEB_PORTS: 81
+      VIRTUAL_HOST: web1.nginx-proxy.local
+
+  web2:
+    image: web 
+    expose:
+      - "82"
+    environment:
+      WEB_PORTS: 82
+      VIRTUAL_HOST: web2.nginx-proxy.local
\ No newline at end of file