test: add test for ash_sql datetime time zone issues

Author: zachdaniel

test/calculation_test.exs

@@ -1018,7 +1018,7 @@ defmodule AshPostgres.CalculationTest do
   end
 
   test "calculation references use the appropriate schema" do
-    record = Record |> Ash.Changeset.for_create(:create, %{full_name: "name"}) |> Ash.create!()
+    Record |> Ash.Changeset.for_create(:create, %{full_name: "name"}) |> Ash.create!()
 
     TempEntity |> Ash.Changeset.for_create(:create, %{full_name: "name"}) |> Ash.create!()
 

test/complex_calculations_test.exs

@@ -346,4 +346,31 @@ defmodule AshPostgres.Test.ComplexCalculationsTest do
     |> Ash.Query.for_read(:failing_many_reference)
     |> Ash.read!(page: [count: true])
   end
+
+  test "lazy datetime with timezone is respected in calculations" do
+    documentation_before =
+      AshPostgres.Test.ComplexCalculations.Documentation
+      |> Ash.Changeset.for_create(:create, %{
+        status: :demonstrated,
+        # 2 hours before Seoul time in UTC
+        inserted_at: ~U[2024-05-01 10:00:00Z]
+      })
+      |> Ash.create!()
+
+    documentation_after =
+      AshPostgres.Test.ComplexCalculations.Documentation
+      |> Ash.Changeset.for_create(:create, %{
+        status: :demonstrated,
+        # 2 hours after Seoul time in UTC
+        inserted_at: ~U[2024-05-01 14:00:00Z]
+      })
+      |> Ash.create!()
+
+    [doc_before, doc_after] =
+      [documentation_before, documentation_after]
+      |> Ash.load!([:is_active_with_timezone])
+
+    assert doc_before.is_active_with_timezone == false
+    assert doc_after.is_active_with_timezone == true
+  end
 end

test/support/complex_calculations/resources/documentation.ex

@@ -42,6 +42,10 @@ defmodule AshPostgres.Test.ComplexCalculations.Documentation do
         end
       )
     )
+
+    calculate :is_active_with_timezone, :boolean do
+      calculation(expr(inserted_at > lazy({AshPostgres.Test.TimezoneHelper, :seoul_time, []})))
+    end
   end
 
   postgres do
@@ -55,3 +59,10 @@ defmodule AshPostgres.Test.ComplexCalculations.Documentation do
     end
   end
 end
+
+defmodule AshPostgres.Test.TimezoneHelper do
+  def seoul_time do
+    # Fixed datetime for testing - equivalent to 2024-05-01 21:00:00 in Seoul (UTC+9)
+    ~U[2024-05-01 12:00:00Z] |> DateTime.shift_zone!("Asia/Seoul")
+  end
+end