Added amount_refunded to Charge

lib/stripe_mock/api/charge.ex

@@ -3,6 +3,7 @@ defmodule StripeMock.API.Charge do
 
   schema "charges" do
     field :amount, :integer
+    field :amount_refunded, :integer, default: 0
     field :capture, :boolean, virtual: true, default: false
     field :captured, :boolean, default: false
     field :currency, :string

lib/stripe_mock/api/refund.ex

@@ -21,6 +21,7 @@ defmodule StripeMock.API.Refund do
     |> validate_required(:amount)
     |> validate_amount()
     |> validate_reason()
+    |> update_charge()
     |> put_common_fields()
   end
 
@@ -75,4 +76,17 @@ defmodule StripeMock.API.Refund do
       _ -> nil
     end
   end
+
+  defp update_charge(changeset) do
+    prepare_changes(changeset, fn prepared_changeset ->
+      if charge_id = get_field(prepared_changeset, :charge_id) do
+        amount = get_field(prepared_changeset, :amount)
+
+        from(c in API.Charge, where: c.id == ^charge_id)
+        |> prepared_changeset.repo.update_all([inc: [amount_refunded: amount]], [])
+      end
+
+      prepared_changeset
+    end)
+  end
 end

lib/stripe_mock_web/views/charge_view.ex

@@ -14,7 +14,7 @@ defmodule StripeMockWeb.ChargeView do
     charge
     |> as_map("charge")
     |> Map.take(
-      ~w(id object amount currency capture description metadata statement_descriptor transfer_group)a
+      ~w(id object amount amount_refunded currency capture description metadata statement_descriptor transfer_group)a
     )
     |> Map.merge(%{
       customer: charge.customer_id,

priv/repo/migrations/20191015221147_create_tables.exs

@@ -77,6 +77,7 @@ defmodule StripeMock.Repo.Migrations.CreateTables do
 
     create table(:charges, primary_key: false) do
       add(:amount, :integer)
+      add(:amount_refunded, :integer, default: 0)
       add(:captured, :boolean, default: false)
       add(:currency, :string)
       add(:statement_descriptor, :string)

test/stripe_mock_web/controllers/charge_controller_test.exs

@@ -32,6 +32,7 @@ defmodule StripeMockWeb.ChargeControllerTest do
       assert %{
                "id" => "ch_" <> id,
                "amount" => 5000,
+               "amount_refunded" => 0,
                "currency" => "some currency",
                "customer" => _,
                "description" => "some description",

test/stripe_mock_web/controllers/refund_controller_test.exs

@@ -45,6 +45,11 @@ defmodule StripeMockWeb.RefundControllerTest do
                "charge" => ^charge_id,
                "metadata" => %{}
              } = json_response(conn, 200)
+
+      # Ensure charge.amount_refunded is updated
+
+      conn = get(conn, Routes.charge_path(conn, :show, charge.id))
+      assert %{"amount" => ^amount, "amount_refunded" => ^amount} = json_response(conn, 200)
     end
 
     test "refunds limited to charge amount", %{conn: conn, charge: charge} do