From 36ade1ecd0ef7df19215649ba93072b5b3ab4565 Mon Sep 17 00:00:00 2001
From: Jose Diaz-Gonzalez <email@josediazgonzalez.com>
Date: Thu, 14 Mar 2024 14:07:08 -0400
Subject: [PATCH] tests: add test for port publishing warning

---
 tests/unit/scheduler-docker-local.bats | 45 ++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)

diff --git a/tests/unit/scheduler-docker-local.bats b/tests/unit/scheduler-docker-local.bats
index ba3199a3b..573cbb819 100644
--- a/tests/unit/scheduler-docker-local.bats
+++ b/tests/unit/scheduler-docker-local.bats
@@ -119,3 +119,48 @@ teardown() {
   assert_success
   assert_output_contains "docker-init" 0
 }
+
+@test "(scheduler-docker-local) publish ports" {
+  run create_app
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run deploy_app
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+  assert_output_contains "Port publishing may not work as expected with multiple containers" 0
+  assert_output_contains "Port publishing may not work as expected with zero downtime" 0
+
+  run /bin/bash -c "dokku ps:scale $TEST_APP web=2"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run deploy_app
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+  assert_output_contains "Port publishing may not work as expected with multiple containers" 1
+  assert_output_contains "Port publishing may not work as expected with zero downtime" 0
+
+  run /bin/bash -c "dokku ps:scale --skip-deploy $TEST_APP web=2"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run deploy_app
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+  assert_output_contains "Port publishing may not work as expected with multiple containers" 0
+  assert_output_contains "Port publishing may not work as expected with zero downtime" 1
+
+  run /bin/bash -c "dokku checks:disable $TEST_APP"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+  assert_output_contains "Port publishing may not work as expected with multiple containers" 0
+  assert_output_contains "Port publishing may not work as expected with zero downtime" 0
+}