record invites

test/lokal/accounts/invites_test.exs

@@ -5,17 +5,13 @@ defmodule Lokal.InvitesTest do
 
   use Lokal.DataCase
   alias Ecto.Changeset
+  alias Lokal.Accounts
   alias Lokal.Accounts.{Invite, Invites}
 
   @moduletag :invites_test
 
   @valid_attrs %{
-    "name" => "some name",
-    "uses_left" => 10
-  }
-  @update_attrs %{
-    "name" => "some updated name",
-    "uses_left" => 5
+    "name" => "some name"
   }
   @invalid_attrs %{
     "name" => nil,
@@ -55,6 +51,27 @@ defmodule Lokal.InvitesTest do
       refute Invites.valid_invite_token?(token)
     end
 
+    test "get_use_count/2 returns the correct invite usage",
+         %{invite: %{token: token} = invite, current_user: current_user} do
+      assert 0 == Invites.get_use_count(invite, current_user)
+
+      assert {:ok, _user} =
+               Accounts.register_user(
+                 %{"email" => unique_user_email(), "password" => valid_user_password()},
+                 token
+               )
+
+      assert 1 == Invites.get_use_count(invite, current_user)
+
+      assert {:ok, _user} =
+               Accounts.register_user(
+                 %{"email" => unique_user_email(), "password" => valid_user_password()},
+                 token
+               )
+
+      assert 2 == Invites.get_use_count(invite, current_user)
+    end
+
     test "use_invite/1 successfully uses an unlimited invite",
          %{invite: %{token: token} = invite, current_user: current_user} do
       {:ok, invite} = Invites.update_invite(invite, %{uses_left: nil}, current_user)
@@ -63,7 +80,9 @@ defmodule Lokal.InvitesTest do
       assert {:ok, ^invite} = Invites.use_invite(token)
     end
 
-    test "use_invite/1 successfully decrements an invite", %{invite: %{token: token}} do
+    test "use_invite/1 successfully decrements an invite",
+         %{invite: %{token: token} = invite, current_user: current_user} do
+      {:ok, _invite} = Invites.update_invite(invite, %{uses_left: 10}, current_user)
       assert {:ok, %{uses_left: 9}} = Invites.use_invite(token)
       assert {:ok, %{uses_left: 8}} = Invites.use_invite(token)
       assert {:ok, %{uses_left: 7}} = Invites.use_invite(token)
@@ -83,26 +102,61 @@ defmodule Lokal.InvitesTest do
       assert {:error, :invalid_token} = Invites.use_invite(token)
     end
 
-    test "create_invite/1 with valid data creates a invite",
+    test "create_invite/1 with valid data creates an unlimited invite",
          %{current_user: current_user} do
-      assert {:ok, %Invite{} = invite} = Invites.create_invite(current_user, @valid_attrs)
+      assert {:ok, %Invite{} = invite} =
+               Invites.create_invite(current_user, %{
+                 "name" => "some name"
+               })
+
       assert invite.name == "some name"
     end
 
+    test "create_invite/1 with valid data creates a limited invite",
+         %{current_user: current_user} do
+      assert {:ok, %Invite{} = invite} =
+               Invites.create_invite(current_user, %{
+                 "name" => "some name",
+                 "uses_left" => 10
+               })
+
+      assert invite.name == "some name"
+      assert invite.uses_left == 10
+    end
+
     test "create_invite/1 with invalid data returns error changeset",
          %{current_user: current_user} do
       assert {:error, %Changeset{}} = Invites.create_invite(current_user, @invalid_attrs)
     end
 
-    test "update_invite/2 with valid data updates the invite",
+    test "update_invite/2 can set an invite to be limited",
          %{invite: invite, current_user: current_user} do
       assert {:ok, %Invite{} = new_invite} =
-               Invites.update_invite(invite, @update_attrs, current_user)
+               Invites.update_invite(
+                 invite,
+                 %{"name" => "some updated name", "uses_left" => 5},
+                 current_user
+               )
 
       assert new_invite.name == "some updated name"
       assert new_invite.uses_left == 5
     end
 
+    test "update_invite/2 can set an invite to be unlimited",
+         %{invite: invite, current_user: current_user} do
+      {:ok, invite} = Invites.update_invite(invite, %{"uses_left" => 5}, current_user)
+
+      assert {:ok, %Invite{} = new_invite} =
+               Invites.update_invite(
+                 invite,
+                 %{"name" => "some updated name", "uses_left" => nil},
+                 current_user
+               )
+
+      assert new_invite.name == "some updated name"
+      assert new_invite.uses_left |> is_nil()
+    end
+
     test "update_invite/2 with invalid data returns error changeset",
          %{invite: invite, current_user: current_user} do
       assert {:error, %Changeset{}} = Invites.update_invite(invite, @invalid_attrs, current_user)

test/lokal/accounts_test.exs

@@ -6,7 +6,7 @@ defmodule Lokal.AccountsTest do
   use Lokal.DataCase
   alias Ecto.Changeset
   alias Lokal.Accounts
-  alias Lokal.Accounts.{User, UserToken}
+  alias Lokal.Accounts.{Invites, User, UserToken}
 
   @moduletag :accounts_test
 
@@ -102,6 +102,17 @@ defmodule Lokal.AccountsTest do
       assert is_nil(user.confirmed_at)
       assert is_nil(user.password)
     end
+
+    test "records used invite during registration" do
+      {:ok, %{id: invite_id, token: token}} =
+        admin_fixture() |> Invites.create_invite(%{"name" => "my invite"})
+
+      assert {:ok, %{invite_id: ^invite_id}} =
+               Accounts.register_user(
+                 %{"email" => unique_user_email(), "password" => valid_user_password()},
+                 token
+               )
+    end
   end
 
   describe "change_user_registration/2" do