Add new count functions for multi tenancy in test support models

izelnakri · Aug 1, 2017 · 8a2dfb5 · 8a2dfb5
1 parent 95b3312
commit 8a2dfb5
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/test/support/simple_models.exs b/test/support/simple_models.exs
@@ -1,6 +1,8 @@
 defmodule SimpleCompany do
   use Ecto.Schema
 
+  alias PaperTrailTest.MultiTenantHelper, as: MultiTenant
+
   import Ecto.Changeset
   import Ecto.Query
 
@@ -31,11 +33,18 @@ defmodule SimpleCompany do
   def count do
     from(record in __MODULE__, select: count(record.id)) |> PaperTrail.RepoClient.repo.one
   end
+  def count(:multitenant) do
+    from(record in __MODULE__, select: count(record.id))
+    |> MultiTenant.add_prefix_to_query()
+    |> PaperTrail.RepoClient.repo.one
+  end
 end
 
 defmodule SimplePerson do
   use Ecto.Schema
 
+  alias PaperTrailTest.MultiTenantHelper, as: MultiTenant
+
   import Ecto.Changeset
   import Ecto.Query
 
@@ -62,4 +71,9 @@ defmodule SimplePerson do
   def count do
     from(record in __MODULE__, select: count(record.id)) |> PaperTrail.RepoClient.repo.one
   end
+  def count(:multitenant) do
+    from(record in __MODULE__, select: count(record.id))
+    |> MultiTenant.add_prefix_to_query()
+    |> PaperTrail.RepoClient.repo.one
+  end
 end
diff --git a/test/support/strict_models.exs b/test/support/strict_models.exs
@@ -1,6 +1,8 @@
 defmodule StrictCompany do
   use Ecto.Schema
 
+  alias PaperTrailTest.MultiTenantHelper, as: MultiTenant
+
   import Ecto.Changeset
   import Ecto.Query
 
@@ -34,11 +36,18 @@ defmodule StrictCompany do
   def count do
     from(record in __MODULE__, select: count(record.id)) |> PaperTrail.RepoClient.repo.one
   end
+  def count(:multitenant) do
+    from(record in __MODULE__, select: count(record.id))
+    |> MultiTenant.add_prefix_to_query()
+    |> PaperTrail.RepoClient.repo.one
+  end
 end
 
 defmodule StrictPerson do
   use Ecto.Schema
 
+  alias PaperTrailTest.MultiTenantHelper, as: MultiTenant
+
   import Ecto.Changeset
   import Ecto.Query
 
@@ -67,4 +76,9 @@ defmodule StrictPerson do
   def count do
     from(record in __MODULE__, select: count(record.id)) |> PaperTrail.RepoClient.repo.one
   end
+  def count(:multitenant) do
+    from(record in __MODULE__, select: count(record.id))
+    |> MultiTenant.add_prefix_to_query()
+    |> PaperTrail.RepoClient.repo.one
+  end
 end