Ash Authentication - How To Test Liveview with Authentication

I am getting error (ArgumentError) session not fetched, call fetch_session/2 while testing liveview that require authentication. I am using Ash Authentication.

How do I test liveviews with Ash Authentication?

Here is the helper function I am using to login while testing

def login(conn, email\\ "test@example.com", hashed_password \\  "$2b$12$mZjy/lPACxT/WcwL8V9og.MtGzgOZc9VEuDdbnRZ2mmHbRV/9xvdK") do
    {:ok, user} = Accounts.create_user(email, hashed_password)
  
    conn
    |> assign(:current_user, user)
    |> put_session(:user, user)
  end

Here is a test I am trying to run.

    @tag run: true
    test "User can list stock categories", %{conn: conn} do
     {:ok, view, html} = 
              conn
              |> login()
              |> live(~p"/stock/categories")

      # Click on each one of them to be edited
      for category <- Category.all!() do
        assert html =~ category.name
        assert html =~ category.description
      end
    end