chore: add test

Author: zachdaniel

test/calculation_test.exs

@@ -1,6 +1,6 @@
 defmodule AshPostgres.CalculationTest do
   use AshPostgres.RepoCase, async: false
-  alias AshPostgres.Test.{Account, Author, Comedian, Comment, Post, User}
+  alias AshPostgres.Test.{Account, Author, Comedian, Comment, Post, Record, TempEntity, User}
 
   require Ash.Query
   import Ash.Expr
@@ -1016,4 +1016,18 @@ defmodule AshPostgres.CalculationTest do
   def fred do
     "fred"
   end
+
+  test "calculation references use the appropriate schema" do
+    record = Record |> Ash.Changeset.for_create(:create, %{full_name: "name"}) |> Ash.create!()
+
+    TempEntity |> Ash.Changeset.for_create(:create, %{full_name: "name"}) |> Ash.create!()
+
+    full_name =
+      Record
+      |> Ash.Query.load(:temp_entity_full_name)
+      |> Ash.read_first!()
+      |> Map.get(:temp_entity_full_name)
+
+    assert full_name == "name"
+  end
 end

test/support/resources/record.ex

@@ -14,23 +14,31 @@ defmodule AshPostgres.Test.Record do
   end
 
   relationships do
-    alias AshPostgres.Test.Entity
-
-    has_one :entity, Entity do
+    has_one :entity, AshPostgres.Test.Entity do
       public?(true)
       no_attributes?(true)
 
       read_action(:read_from_temp)
 
       filter(expr(full_name == parent(full_name)))
     end
+
+    has_one :temp_entity, AshPostgres.Test.TempEntity do
+      public?(true)
+      source_attribute(:full_name)
+      destination_attribute(:full_name)
+    end
   end
 
   postgres do
     table "records"
     repo AshPostgres.TestRepo
   end
 
+  calculations do
+    calculate(:temp_entity_full_name, :string, expr(temp_entity.full_name))
+  end
+
   actions do
     default_accept(:*)