From 41a89bbb5db73ea5ba43a58b834ddc6defe8d85f Mon Sep 17 00:00:00 2001
From: Jose Diaz-Gonzalez <email@josediazgonzalez.com>
Date: Wed, 7 Feb 2024 02:20:49 -0500
Subject: [PATCH] feat: add a test for ingress-nginx support

---
 tests/unit/scheduler-k3s.bats | 37 ++++++++++++++++++++++++++++++++---
 1 file changed, 34 insertions(+), 3 deletions(-)

diff --git a/tests/unit/scheduler-k3s.bats b/tests/unit/scheduler-k3s.bats
index ca503a258..182f489e7 100644
--- a/tests/unit/scheduler-k3s.bats
+++ b/tests/unit/scheduler-k3s.bats
@@ -49,7 +49,8 @@ install_k3s() {
   assert_success
   assert_output_contains "Login Succeeded"
 
-  run /bin/bash -c "dokku scheduler-k3s:initialize"
+  INGRESS_CLASS="${INGRESS_CLASS:-traefik}"
+  run /bin/bash -c "dokku scheduler-k3s:initialize --ingress-class $INGRESS_CLASS"
   echo "output: $output"
   echo "status: $status"
   assert_success
@@ -70,12 +71,42 @@ uninstall_k3s() {
   install_k3s
 }
 
-@test "(scheduler-k3s) deploy" {
+@test "(scheduler-k3s) deploy traefik" {
   if [[ -z "$DOCKERHUB_USERNAME" ]] || [[ -z "$DOCKERHUB_TOKEN" ]]; then
     skip "skipping due to missing docker.io credentials DOCKERHUB_USERNAME:DOCKERHUB_TOKEN"
   fi
 
-  install_k3s
+  INGRESS_CLASS=traefik install_k3s
+
+  run /bin/bash -c "dokku apps:create $TEST_APP"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "dokku domains:set $TEST_APP $TEST_APP.dokku.me"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "dokku git:sync --build $TEST_APP https://github.com/dokku/smoke-test-app.git"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "sleep 20"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  assert_http_localhost_response "http" "$TEST_APP.dokku.me" "80" "" "python/http.server"
+}
+
+@test "(scheduler-k3s) deploy nginx" {
+  if [[ -z "$DOCKERHUB_USERNAME" ]] || [[ -z "$DOCKERHUB_TOKEN" ]]; then
+    skip "skipping due to missing docker.io credentials DOCKERHUB_USERNAME:DOCKERHUB_TOKEN"
+  fi
+
+  INGRESS_CLASS=nginx install_k3s
 
   run /bin/bash -c "dokku apps:create $TEST_APP"
   echo "output: $output"