From 1a66102e3043c3e0f5540c0527fb96982c5eeaff Mon Sep 17 00:00:00 2001
From: Jose Diaz-Gonzalez <email@josediazgonzalez.com>
Date: Thu, 7 Jan 2021 02:54:10 -0500
Subject: [PATCH] feat: add tests for allow-host and public-key commands

---
 tests/unit/git_3.bats | 77 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100644 tests/unit/git_3.bats

diff --git a/tests/unit/git_3.bats b/tests/unit/git_3.bats
new file mode 100644
index 000000000..dd6a99b5f
--- /dev/null
+++ b/tests/unit/git_3.bats
@@ -0,0 +1,77 @@
+#!/usr/bin/env bats
+
+load test_helper
+
+setup() {
+  global_setup
+  touch /home/dokku/.ssh/known_hosts
+  chown dokku:dokku /home/dokku/.ssh/known_hosts
+}
+
+teardown() {
+  rm -f /home/dokku/.ssh/id_rsa.pub || true
+  destroy_app
+  global_teardown
+}
+
+@test "(git) git:allow-host" {
+  run /bin/bash -c "dokku git:allow-host"
+  echo "output: $output"
+  echo "status: $status"
+  assert_failure
+
+  run /bin/bash -c "cat /home/dokku/.ssh/known_hosts | wc -l"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+  start_lines=$output
+
+  run /bin/bash -c "dokku git:allow-host github.com"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "cat /home/dokku/.ssh/known_hosts | wc -l"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "test -f /home/dokku/.ssh/known_hosts"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "cat /home/dokku/.ssh/known_hosts | wc -l"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+  assert_equal "$output" "$((start_lines + 1))"
+
+  run /bin/bash -c "dokku git:allow-host github.com"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "cat /home/dokku/.ssh/known_hosts | wc -l"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+  assert_equal "$output" "$((start_lines + 2))"
+}
+
+@test "(git) git:public-key" {
+  run /bin/bash -c "dokku git:public-key"
+  echo "output: $output"
+  echo "status: $status"
+  assert_failure
+
+  run /bin/bash -c "cp /root/.ssh/dokku_test_rsa.pub /home/dokku/.ssh/id_rsa.pub"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "dokku git:public-key"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+}