Add user_id to events

Author: joshsmith

lib/code_corps/stripe_service/webhook_processing/event_handler.ex

@@ -5,9 +5,9 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandler do
     ConnectEventHandler, IgnoredEventHandler, PlatformEventHandler
   }
 
-  def handle(%Stripe.Event{type: type} = api_event, handler) do
+  def handle(%Stripe.Event{type: type} = api_event, handler, user_id \\ nil) do
     with {:ok, endpoint} <- infer_endpoint_from_handler(handler),
-         {:ok, %StripeEvent{} = local_event} <- find_or_create_event(api_event, endpoint)
+         {:ok, %StripeEvent{} = local_event} <- find_or_create_event(api_event, endpoint, user_id)
     do
       case IgnoredEventHandler.should_handle?(type, handler) do
         true -> call_ignored_handler(local_event, handler)
@@ -21,20 +21,21 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandler do
   defp infer_endpoint_from_handler(ConnectEventHandler), do: {:ok, "connect"}
   defp infer_endpoint_from_handler(PlatformEventHandler), do: {:ok, "platform"}
 
-  defp find_or_create_event(%Stripe.Event{} = api_event, endpoint) do
+  defp find_or_create_event(%Stripe.Event{} = api_event, endpoint, user_id) do
     case find_event(api_event.id) do
       %StripeEvent{status: "processing"} -> {:error, :already_processing}
       %StripeEvent{} = local_event -> {:ok, local_event}
-      nil -> create_event(api_event, endpoint)
+      nil -> create_event(api_event, endpoint, user_id)
     end
   end
 
   defp find_event(id_from_stripe) do
     Repo.get_by(StripeEvent, id_from_stripe: id_from_stripe)
   end
 
-  defp create_event(%Stripe.Event{} = api_event, endpoint) do
-    with {:ok, params} <- StripeEventAdapter.to_params(api_event, %{"endpoint" => endpoint}) do
+  defp create_event(%Stripe.Event{} = api_event, endpoint, user_id) do
+    event_with_user_id = Map.merge(api_event, %{user_id: user_id})
+    with {:ok, params} <- StripeEventAdapter.to_params(event_with_user_id, %{"endpoint" => endpoint}) do
       %StripeEvent{} |> StripeEvent.create_changeset(params) |> Repo.insert
     end
   end

test/lib/code_corps/stripe_service/webhook_processing/event_handler_test.exs

@@ -65,10 +65,11 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandlerTest do
       connect_account = insert(:stripe_connect_account)
       event = build_event("account.updated", "account", %Stripe.Account{id: connect_account.id_from_stripe})
 
-      {:ok, event} = EventHandler.handle(event, ConnectEventHandler)
+      {:ok, event} = EventHandler.handle(event, ConnectEventHandler, "acc_123")
       assert event.object_type == "account"
       assert event.object_id == connect_account.id_from_stripe
       assert event.status == "processed"
+      assert event.user_id == "acc_123"
     end
 
     test "handles customer.subscription.updated" do
@@ -93,10 +94,11 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandlerTest do
         }
       )
 
-      {:ok, event} = EventHandler.handle(event, ConnectEventHandler)
+      {:ok, event} = EventHandler.handle(event, ConnectEventHandler, "acc_123")
       assert event.object_type == "subscription"
       assert event.object_id == subscription.id_from_stripe
       assert event.status == "processed"
+      assert event.user_id == "acc_123"
     end
 
     test "handles customer.subscription.deleted" do
@@ -121,10 +123,11 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandlerTest do
         }
       )
 
-      {:ok, event} = EventHandler.handle(event, ConnectEventHandler)
+      {:ok, event} = EventHandler.handle(event, ConnectEventHandler, "acc_123")
       assert event.object_type == "subscription"
       assert event.object_id == subscription.id_from_stripe
       assert event.status == "processed"
+      assert event.user_id == "acc_123"
     end
 
     test "handles invoice.payment_succeeded" do
@@ -150,10 +153,11 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandlerTest do
         }
       )
 
-      {:ok, event} = EventHandler.handle(event, ConnectEventHandler)
+      {:ok, event} = EventHandler.handle(event, ConnectEventHandler, "acc_123")
       assert event.object_type == "invoice"
       assert event.object_id == "ivc_123"
       assert event.status == "processed"
+      assert event.user_id == "acc_123"
 
       assert Repo.get_by(StripeInvoice, id_from_stripe: "ivc_123")
     end
@@ -165,13 +169,15 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandlerTest do
 
       {:ok, event} = EventHandler.handle(event, PlatformEventHandler)
       assert event.endpoint == "platform"
+      assert event.user_id == nil
     end
 
     test "sets endpoint to 'connect' when using ConnectEventHandler" do
       event = build_event
 
-      {:ok, event} = EventHandler.handle(event, ConnectEventHandler)
+      {:ok, event} = EventHandler.handle(event, ConnectEventHandler, "acc_123")
       assert event.endpoint == "connect"
+      assert event.user_id == "acc_123"
     end
 
     test "creates event if id is new" do
@@ -182,6 +188,7 @@ defmodule CodeCorps.StripeService.WebhookProcessing.EventHandlerTest do
       assert event.object_id == "stub_id"
       assert event.object_type == "any_object"
       assert event.status == "unhandled"
+      assert event.user_id == nil
     end
 
     test "uses existing event if id exists" do