Add API Key reset API

rjsamson · Jun 24, 2016 · 09925e3 · 09925e3
1 parent 9fba71f
commit 09925e3
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/lib/stackfooter/api_key_registry.ex b/lib/stackfooter/api_key_registry.ex
@@ -25,8 +25,18 @@ defmodule Stackfooter.ApiKeyRegistry do
     end
   end
 
+  def reset_api_keys(pid) do
+    GenServer.call(pid, :reset)
+  end
+
+  def handle_call(:reset, _from, api_keys) do
+    :ets.delete_all_objects(api_keys)
+    {:reply, :ok, api_keys}
+  end
+
   def handle_call({:add_key, api_key, account}, _from, api_keys) do
     :ets.insert(api_keys, {api_key, account})
+
     {:reply, {:ok, {api_key, account}}, api_keys}
   end
 end
diff --git a/test/otp/api_key_registry_test.exs b/test/otp/api_key_registry_test.exs
@@ -3,6 +3,7 @@ defmodule Stackfooter.ApiKeyRegistryTest do
   alias Stackfooter.ApiKeyRegistry
 
   setup do
+    Stackfooter.ApiKeyRegistry.reset_api_keys(Stackfooter.ApiKeyRegistry)
     Stackfooter.ApiKeyRegistry.add_key(Stackfooter.ApiKeyRegistry, "4cy7uf63Lw2Sx6652YmLwBKy662weU4q", "admin")
     Stackfooter.ApiKeyRegistry.add_key(Stackfooter.ApiKeyRegistry, "7eWeGhc8n0va5bjT66C0Vl1fBw2618BJ", "rjsamson")
 
@@ -14,4 +15,11 @@ defmodule Stackfooter.ApiKeyRegistryTest do
 
     assert all_accounts == ["RJSAMSON", "ADMIN"]
   end
+
+  test "resets all API keys" do
+    Stackfooter.ApiKeyRegistry.reset_api_keys(Stackfooter.ApiKeyRegistry)
+    all_accounts = ApiKeyRegistry.all_account_names(ApiKeyRegistry)
+
+    assert all_accounts == []
+  end
 end