From 0aab3c2027d54fc1ba834ab73fd53328952b7a1b Mon Sep 17 00:00:00 2001
From: Simponic <loganhunt@simponic.xyz>
Date: Thu, 29 Dec 2022 17:49:42 -0700
Subject: Make tests pass, rename jail threshold

---
 test/ssh/ssh_auth_test-emacs-elixir-format.exs | 81 --------------------------
 test/ssh/ssh_auth_test.exs                     | 28 ++++-----
 2 files changed, 15 insertions(+), 94 deletions(-)
 delete mode 100644 test/ssh/ssh_auth_test-emacs-elixir-format.exs

(limited to 'test/ssh')

diff --git a/test/ssh/ssh_auth_test-emacs-elixir-format.exs b/test/ssh/ssh_auth_test-emacs-elixir-format.exs
deleted file mode 100644
index cb07259..0000000
--- a/test/ssh/ssh_auth_test-emacs-elixir-format.exs
+++ /dev/null
@@ -1,81 +0,0 @@
-defmodule Chessh.SSH.AuthTest do
-  use ExUnit.Case
-  alias Chessh.{Player, Repo, Key}
-
-  @localhost '127.0.0.1'
-  @key_name "The Gamer Machine"
-  @valid_user %{username: "logan", password: "password"}
-  @client_test_keys_dir Path.join(Application.compile_env!(:chessh, :key_dir), "client_keys")
-  @client_pub_key 'id_ed25519.pub'
-
-  setup_all do
-    case Ecto.Adapters.SQL.Sandbox.checkout(Repo) do
-      :ok -> nil
-      {:already, :owner} -> nil
-    end
-
-    Ecto.Adapters.SQL.Sandbox.mode(Repo, {:shared, self()})
-
-    {:ok, player} = Repo.insert(Player.registration_changeset(%Player{}, @valid_user))
-
-    {:ok, key_text} = File.read(Path.join(@client_test_keys_dir, @client_pub_key))
-
-    {:ok, _key} =
-      Repo.insert(
-        Key.changeset(%Key{}, %{key: key_text, name: @key_name})
-        |> Ecto.Changeset.put_assoc(:player, player)
-      )
-
-    :ok
-  end
-
-  test "Password attempts are rate limited" do
-    assert :disconnect ==
-             Enum.reduce(
-               1..Application.fetch_env!(:chessh, RateLimits, :jail_threshold),
-               fn _, _ ->
-                 Chessh.SSH.Daemon.pwd_authenticate(
-                        @valid_user.username,
-                        'wrong_password',
-                        @localhost
-                      ) do
-               end
-             )
-  end
-
-  test "INTEGRATION - Can ssh into daemon with password or public key" do
-    {:ok, sup} = Task.Supervisor.start_link()
-    test_pid = self()
-
-    Task.Supervisor.start_child(sup, fn ->
-      {:ok, _pid} =
-        :ssh.connect(@localhost, Application.fetch_env!(:chessh, :port),
-          user: String.to_charlist(@valid_user.username),
-          password: String.to_charlist(@valid_user.password),
-          auth_methods: 'password',
-          silently_accept_hosts: true
-        )
-
-      send(test_pid, :connected_via_password)
-    end)
-
-    Task.Supervisor.start_child(sup, fn ->
-      {:ok, _pid} =
-        :ssh.connect(@localhost, Application.fetch_env!(:chessh, :port),
-          user: String.to_charlist(@valid_user.username),
-          auth_methods: 'publickey',
-          silently_accept_hosts: true,
-          user_dir: String.to_charlist(@client_test_keys_dir)
-        )
-
-      send(test_pid, :connected_via_public_key)
-    end)
-
-    assert_receive(:connected_via_password, 500)
-    assert_receive(:connected_via_public_key, 500)
-  end
-
-  test "INTEGRATION - User cannot have more than specified concurrent sessions" do
-    :ok
-  end
-end
diff --git a/test/ssh/ssh_auth_test.exs b/test/ssh/ssh_auth_test.exs
index cb07259..27d5e4c 100644
--- a/test/ssh/ssh_auth_test.exs
+++ b/test/ssh/ssh_auth_test.exs
@@ -3,17 +3,14 @@ defmodule Chessh.SSH.AuthTest do
   alias Chessh.{Player, Repo, Key}
 
   @localhost '127.0.0.1'
+  @localhost_inet {{127, 0, 0, 1}, 1}
   @key_name "The Gamer Machine"
   @valid_user %{username: "logan", password: "password"}
   @client_test_keys_dir Path.join(Application.compile_env!(:chessh, :key_dir), "client_keys")
   @client_pub_key 'id_ed25519.pub'
 
   setup_all do
-    case Ecto.Adapters.SQL.Sandbox.checkout(Repo) do
-      :ok -> nil
-      {:already, :owner} -> nil
-    end
-
+    Ecto.Adapters.SQL.Sandbox.checkout(Repo)
     Ecto.Adapters.SQL.Sandbox.mode(Repo, {:shared, self()})
 
     {:ok, player} = Repo.insert(Player.registration_changeset(%Player{}, @valid_user))
@@ -30,15 +27,19 @@ defmodule Chessh.SSH.AuthTest do
   end
 
   test "Password attempts are rate limited" do
+    jail_attempt_threshold =
+      Application.get_env(:chessh, RateLimits)
+      |> Keyword.get(:jail_attempt_threshold)
+
     assert :disconnect ==
              Enum.reduce(
-               1..Application.fetch_env!(:chessh, RateLimits, :jail_threshold),
+               0..(jail_attempt_threshold + 1),
                fn _, _ ->
                  Chessh.SSH.Daemon.pwd_authenticate(
-                        @valid_user.username,
-                        'wrong_password',
-                        @localhost
-                      ) do
+                   @valid_user.username,
+                   "wrong_password",
+                   @localhost_inet
+                 )
                end
              )
   end
@@ -75,7 +76,8 @@ defmodule Chessh.SSH.AuthTest do
     assert_receive(:connected_via_public_key, 500)
   end
 
-  test "INTEGRATION - User cannot have more than specified concurrent sessions" do
-    :ok
-  end
+  # TODO
+  #  test "INTEGRATION - User cannot have more than specified concurrent sessions" do
+  #    :ok
+  #  end
 end
-- 
cgit v1.2.3-70-g09d2