Monitor owner of the connection

The connection process should monitor the process that started it
and die with the same reason that the owner process did.
This should solve the issue of zombie connections laying around
after the processes that started them die.

1 files changed, 22 insertions, 3 deletions

diff --git a/test/client_test.exs b/test/client_test.exs
index 6cff0b7..b7277b6 100644
--- a/test/client_test.exs
+++ b/test/client_test.exs
@@ -1,11 +1,30 @@
 defmodule ExIrc.ClientTest do
   use ExUnit.Case
 
-
   test "start multiple clients" do
-    {:ok, pid} = ExIrc.start_client!
-    {:ok, pid2} = ExIrc.start_client!
+    assert {:ok, pid} = ExIrc.start_client!
+    assert {:ok, pid2} = ExIrc.start_client!
     assert pid != pid2
   end
 
+  test "client dies if owner process dies" do
+    test_pid = self()
+
+    pid = spawn_link(fn ->
+      assert {:ok, pid} = ExIrc.start_client!
+      send(test_pid, {:client, pid})
+      receive do
+        :stop -> :ok
+      end
+    end)
+
+    client_pid = receive do
+      {:client, pid} -> pid
+    end
+
+    assert Process.alive?(client_pid)
+    send(pid, :stop)
+    :timer.sleep(1)
+    refute Process.alive?(client_pid)
+  end
 end