Associate the keywords with user

Author: rosle

lib/google_crawler/accounts/user.ex

@@ -10,6 +10,8 @@ defmodule GoogleCrawler.Accounts.User do
     field :password, :string, virtual: true
     field :password_confirmation, :string, virtual: true
 
+    has_many :keywords, GoogleCrawler.Search.Keyword
+
     timestamps()
   end
 

lib/google_crawler/search.ex

@@ -43,16 +43,17 @@ defmodule GoogleCrawler.Search do
 
   ## Examples
 
-      iex> create_keyword(%{field: value})
+      iex> create_keyword(%User{}, %{field: value})
       {:ok, %Keyword{}}
 
-      iex> create_keyword(%{field: bad_value})
+      iex> create_keyword(%User{}, %{field: bad_value})
       {:error, %Ecto.Changeset{}}
 
   """
-  def create_keyword(attrs \\ %{}) do
+  def create_keyword(attrs \\ %{}, user) do
     %Keyword{}
     |> Keyword.changeset(attrs)
+    |> Ecto.Changeset.put_assoc(:user, user)
     |> Repo.insert()
   end
 

lib/google_crawler/search/keyword.ex

@@ -5,6 +5,8 @@ defmodule GoogleCrawler.Search.Keyword do
   schema "keywords" do
     field :keyword, :string
 
+    belongs_to :user, GoogleCrawler.Accounts.User
+
     timestamps()
   end
 

lib/google_crawler_web/controllers/upload_controller.ex

@@ -5,14 +5,15 @@ defmodule GoogleCrawlerWeb.UploadController do
   alias GoogleCrawler.Search
   alias GoogleCrawler.Search.KeywordFile
 
+  @spec create(atom | %{__struct__: atom}, map) :: Plug.Conn.t()
   def create(conn, %{"keyword_file" => keyword_file}) do
     changeset = KeywordFile.changeset(%KeywordFile{}, keyword_file)
 
     if changeset.valid? do
       file = get_change(changeset, :file, nil)
 
       Search.parse_keywords_from_file!(file.path, file.content_type)
-      |> create_and_trigger_google_search
+      |> create_and_trigger_google_search(conn)
       |> put_error_flash_for_failed_keywords(conn)
       |> redirect(to: Routes.dashboard_path(conn, :index))
     else
@@ -23,11 +24,11 @@ defmodule GoogleCrawlerWeb.UploadController do
   end
 
   # TODO: Trigger the scrapper background worker
-  defp create_and_trigger_google_search(csv_result) do
+  defp create_and_trigger_google_search(csv_result, conn) do
     csv_result
     |> Stream.map(fn keyword_row -> List.first(keyword_row) end)
     |> Stream.map(fn keyword -> %{keyword: keyword} end)
-    |> Enum.map(&Search.create_keyword/1)
+    |> Enum.map(&Search.create_keyword(&1, conn.assigns.current_user))
   end
 
   defp put_error_flash_for_failed_keywords(create_result, conn) do

priv/repo/migrations/20200330130512_add_user_id_to_keywords.exs

@@ -0,0 +1,9 @@
+defmodule GoogleCrawler.Repo.Migrations.AddUserIdToKeywords do
+  use Ecto.Migration
+
+  def change do
+    alter table(:keywords) do
+      add :user_id, references(:users)
+    end
+  end
+end

test/factories/keyword_factory.ex

@@ -1,5 +1,6 @@
 defmodule GoogleCrawler.KeywordFactory do
   alias GoogleCrawler.Search
+  alias GoogleCrawler.UserFactory
 
   def default_attrs do
     %{
@@ -11,10 +12,10 @@ defmodule GoogleCrawler.KeywordFactory do
     Enum.into(attrs, default_attrs())
   end
 
-  def create(attrs \\ %{}) do
+  def create(attrs \\ %{}, user \\ UserFactory.create()) do
     keyword_attrs = build_attrs(attrs)
 
-    {:ok, keyword} = Search.create_keyword(keyword_attrs)
+    {:ok, keyword} = Search.create_keyword(keyword_attrs, user)
 
     keyword
   end

test/google_crawler/search_test.exs

@@ -4,6 +4,7 @@ defmodule GoogleCrawler.SearchTest do
   alias GoogleCrawler.Search
   alias GoogleCrawler.Search.Keyword
   alias GoogleCrawler.KeywordFactory
+  alias GoogleCrawler.UserFactory
 
   describe "keywords" do
     test "list_keywords/0 returns all keywords" do
@@ -19,16 +20,18 @@ defmodule GoogleCrawler.SearchTest do
     end
 
     test "create_keyword/1 with valid data creates a keyword" do
+      user = UserFactory.create()
       keyword_attrs = KeywordFactory.build_attrs(%{keyword: "elixir"})
 
-      assert {:ok, %Keyword{} = keyword} = Search.create_keyword(keyword_attrs)
+      assert {:ok, %Keyword{} = keyword} = Search.create_keyword(keyword_attrs, user)
       assert keyword.keyword == "elixir"
     end
 
     test "create_keyword/1 with invalid data returns error changeset" do
+      user = UserFactory.create()
       keyword_attrs = KeywordFactory.build_attrs(%{keyword: ""})
 
-      assert {:error, %Ecto.Changeset{}} = Search.create_keyword(keyword_attrs)
+      assert {:error, %Ecto.Changeset{}} = Search.create_keyword(keyword_attrs, user)
     end
   end