refactor: current response

lib/safira_web/controllers/spotlight/spotlight_json.ex

@@ -8,8 +8,13 @@ defmodule SafiraWeb.SpotlightJSON do
   end
 
   def current(%{company: company, spotlight: spotlight}) do
+    data =
+      data(company, spotlight)
+      |> Map.drop([:remaining])
+      |> Map.put(:badge_id, company.badge_id)
+
     %{
-      data: data(company, spotlight)
+      data: data
     }
   end
 

test/safira_web/controllers/spotlight_controller_test.exs

@@ -201,7 +201,7 @@ defmodule SafiraWeb.SpotlightControllerTest do
                "data" => %{
                  "id" => company.id,
                  "name" => company.name,
-                 "remaining" => company.remaining_spotlights,
+                 "badge_id" => company.badge_id,
                  "end" => DateTime.to_iso8601(spotlight.end)
                }
              }
@@ -260,6 +260,31 @@ defmodule SafiraWeb.SpotlightControllerTest do
       assert json_response(conn, 404) == %{"error" => "Spotlight not found"}
     end
 
+    test "when user is an attendee and a spotlight is running", %{user: user} do
+      _staff = insert(:staff, user: user, is_admin: true)
+      company = insert(:company, remaining_spotlights: 1)
+
+      %{conn: conn, user: _user} = api_authenticate(user)
+
+      # Start a spotlight for the company
+      Interaction.start_spotlight(company)
+      spotlight = Interaction.get_spotlight()
+      assert spotlight != nil
+
+      conn =
+        conn
+        |> get(Routes.spotlight_path(conn, :current))
+
+      assert json_response(conn, 200) == %{
+               "data" => %{
+                 "id" => company.id,
+                 "name" => company.name,
+                 "badge_id" => company.badge_id,
+                 "end" => DateTime.to_iso8601(spotlight.end)
+               }
+             }
+    end
+
     test "when user is a company", %{user: user} do
       _company = insert(:company, user: user)