Improve test coverage

lib/nimble_ownership.ex

@@ -444,7 +444,7 @@ defmodule NimbleOwnership do
         state =
           if not Map.has_key?(state.owner_cleanup, owner_pid) do
             _ref = Process.monitor(owner_pid)
-            state
+            put_in(state.owner_cleanup[owner_pid], :auto)
           else
             state
           end
@@ -514,11 +514,11 @@ defmodule NimbleOwnership do
   # An owner went down, so we need to clean up all of its allowances as well as all its keys.
   def handle_info({:DOWN, _ref, _, down_pid, _}, state)
       when is_map_key(state.owners, down_pid) do
-    case state.owner_cleanup[down_pid] do
+    case state.owner_cleanup[down_pid] || :auto do
       :manual ->
         {:noreply, state}
 
-      nil ->
+      :auto ->
         state = pop_owner_and_clean_up_allowances(state, down_pid)
         {:noreply, state}
     end

test/nimble_ownership_test.exs

@@ -314,6 +314,24 @@ defmodule NimbleOwnershipTest do
 
       assert :error = NimbleOwnership.fetch_owner(@server, [child_pid, owner_pid], key)
     end
+
+    test "if a child shuts down, the deps of that child are cleaned up but not the whole allowance",
+         %{key: key} do
+      {owner_pid, _owner_monitor_ref} = spawn_monitor(fn -> Process.sleep(:infinity) end)
+      {child_pid1, child_monitor_ref1} = spawn_monitor(fn -> Process.sleep(:infinity) end)
+      {child_pid2, _child_monitor_ref2} = spawn_monitor(fn -> Process.sleep(:infinity) end)
+
+      init_key(owner_pid, key, %{counter: 1})
+
+      assert :ok = NimbleOwnership.allow(@server, owner_pid, child_pid1, key)
+      assert :ok = NimbleOwnership.allow(@server, child_pid1, child_pid2, key)
+
+      Process.exit(child_pid1, :kill)
+      assert_receive {:DOWN, ^child_monitor_ref1, _, _, _}
+
+      assert {:ok, ^owner_pid} =
+               NimbleOwnership.fetch_owner(@server, [child_pid1, owner_pid], key)
+    end
   end
 
   describe "set_owner_to_manual_cleanup/2" do