Add case for unknown token in Token.get/1

lib/hostas/token.ex

@@ -48,15 +48,20 @@ defmodule Hostas.Token do
   end
 
   @doc """
-  Returns {:ok, struct} if the token hasn't expired. Otherwise, returns :expired
+  Returns:
+   * `{:ok, struct}` if the token exists and is not expired
+   * `{:error, :unknown}` if the token was not found
+   * `{:error, :expired}` if the token exists, but is expired
   """
   def get(key) do
-    struct = Repo.one!(from t in Token, where: t.token == ^key)
+    case Repo.one(from t in Token, where: t.token == ^key) do
-
+      nil -> {:error, :unknown}
-    unless expired?(struct) do
+      struct ->
-      {:ok, struct}
+        unless expired?(struct) do
-    else
+          {:ok, struct}
-      :expired
+        else
+          {:error, :expired}
+        end
     end
   end
 

test/hostas/token_test.exs

@@ -33,12 +33,16 @@ defmodule Hostas.TokenTest do
 
     test "indicates the token is expired" do
       {:ok, struct} = Token.new(create_denizen().id, -30)
-      assert Token.get(struct.token) == :expired
+      assert Token.get(struct.token) == {:error, :expired}
     end
 
     test "by key" do
       {:ok, struct} = Token.new(create_denizen().id, -30)
-      assert Token.expired?(struct.token)
+      assert Token.get(struct.token) == {:error, :expired}
+    end
+
+    test "doesn't exist" do
+      assert Token.get("unknown token") == {:error, :unknown}
     end
   end
 

test/support/data_case.ex

@@ -16,6 +16,9 @@ defmodule Hostas.DataCase do
 
   use ExUnit.CaseTemplate
 
+  alias Hostas.Repo
+  alias Hostas.Denizen
+
   using do
     quote do
       alias Hostas.Repo
@@ -55,4 +58,11 @@ defmodule Hostas.DataCase do
       end)
     end)
   end
+
+  def create_denizen(handle \\ "denizen") do
+    Repo.insert!(
+      Denizen.changeset(%Denizen{}, %{handle: handle, name: "Test Denizen", password: "password"})
+    )
+  end
+
 end