Parameterize TCP timeout

lib/hulaaki/client.ex

@@ -57,6 +57,7 @@ defmodule Hulaaki.Client do
       def handle_call({:connect, opts, conn_pid}, _from, state) do
         host          = opts |> Keyword.fetch!(:host)
         port          = opts |> Keyword.fetch!(:port)
+        timeout       = opts |> Keyword.get(:timeout, 100)
 
         client_id     = opts |> Keyword.fetch!(:client_id)
         username      = opts |> Keyword.get(:username, "")
@@ -74,7 +75,7 @@ defmodule Hulaaki.Client do
 
         state = Map.merge(%{connection: conn_pid}, state)
 
-        connect_opts = [host: host, port: port]
+        connect_opts = [host: host, port: port, timeout: timeout]
         :ok = state.connection |> Connection.connect(message, connect_opts)
         {:reply, :ok, state}
       end

lib/hulaaki/connection.ex

@@ -138,7 +138,7 @@ defmodule Hulaaki.Connection do
   end
 
   defp open_tcp_socket(opts) do
-    timeout  = 100
+    timeout  = opts |> Keyword.fetch!(:timeout)
     host     = opts |> Keyword.fetch!(:host)
     host     = if is_binary(host), do: String.to_char_list(host), else: host
     port     = opts |> Keyword.fetch!(:port)

test/hulaaki/client_test.exs

@@ -76,7 +76,7 @@ defmodule Hulaaki.ClientTest do
   end
 
   defp pre_connect(pid) do
-    options = [client_id: "some-name", host: TestConfig.mqtt_host, port: TestConfig.mqtt_port]
+    options = [client_id: "some-name", host: TestConfig.mqtt_host, port: TestConfig.mqtt_port, timeout: 200]
     SampleClient.connect(pid, options)
   end
 

test/hulaaki/connection_test.exs

@@ -24,7 +24,7 @@ defmodule Hulaaki.ConnectionTest do
 
   defp pre_connect(pid) do
     message = Message.connect(client_name, "", "", "", "", 0, 0, 0, 100)
-    Connection.connect(pid, message, [host: TestConfig.mqtt_host, port: TestConfig.mqtt_port])
+    Connection.connect(pid, message, [host: TestConfig.mqtt_host, port: TestConfig.mqtt_port, timeout: TestConfig.mqtt_timeout])
   end
 
   defp post_disconnect(pid) do

test/test_helper.exs

@@ -10,4 +10,8 @@ defmodule TestConfig do
     {port, _} = ( System.get_env("MQTT_PORT") || "1883" ) |> Integer.parse
     port
   end
+
+  def mqtt_timeout do
+    500
+  end
 end