From 9e22e1bfde31a027e416a3165fa3e95696de3db3 Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Fri, 6 Sep 2024 12:54:20 +0300
Subject: [PATCH] Add the configuration for wallaby.

---
 config/test.exs                    | 9 ++++++++-
 lib/link_shortener_web/endpoint.ex | 2 +-
 test/test_helper.exs               | 3 +++
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/config/test.exs b/config/test.exs
index a776e98..fdc194a 100644
--- a/config/test.exs
+++ b/config/test.exs
@@ -21,7 +21,7 @@ config :link_shortener, LinkShortener.Repo,
 config :link_shortener, LinkShortenerWeb.Endpoint,
   http: [ip: {127, 0, 0, 1}, port: 4002],
   secret_key_base: System.get_env("SECRET_KEY_BASE"),
-  server: false
+  server: true
 
 # In test we don't send emails
 config :link_shortener, LinkShortener.Mailer, adapter: Swoosh.Adapters.Test
@@ -38,3 +38,10 @@ config :phoenix, :plug_init_mode, :runtime
 # Enable helpful, but potentially expensive runtime checks
 config :phoenix_live_view,
   enable_expensive_runtime_checks: true
+
+config :wallaby,
+  otp_app: :link_shortener,
+  screenshot_on_failure: true,
+  screenshot_dir: "screenshots",
+  driver: Wallaby.Selenium
+
diff --git a/lib/link_shortener_web/endpoint.ex b/lib/link_shortener_web/endpoint.ex
index 7324f90..f9e4e28 100644
--- a/lib/link_shortener_web/endpoint.ex
+++ b/lib/link_shortener_web/endpoint.ex
@@ -12,7 +12,7 @@ defmodule LinkShortenerWeb.Endpoint do
   ]
 
   socket "/live", Phoenix.LiveView.Socket,
-    websocket: [connect_info: [session: @session_options]],
+    websocket: [connect_info: [:user_agent, session: @session_options]],
     longpoll: [connect_info: [session: @session_options]]
 
   # Serve at "/" the static files from "priv/static" directory.
diff --git a/test/test_helper.exs b/test/test_helper.exs
index fbe0342..adb882d 100644
--- a/test/test_helper.exs
+++ b/test/test_helper.exs
@@ -1,2 +1,5 @@
 ExUnit.start()
 Ecto.Adapters.SQL.Sandbox.mode(LinkShortener.Repo, :manual)
+
+{:ok, _} = Application.ensure_all_started(:wallaby)
+Application.put_env(:wallaby, :base_url, LinkShortenerWeb.Endpoint.url)