Add PointZ handling

felt · Feb 7, 2019 · 53b6edb · 53b6edb
1 parent ac47c38
commit 53b6edb
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/lib/geo_postgis/geometry.ex b/lib/geo_postgis/geometry.ex
@@ -19,6 +19,7 @@ if Code.ensure_loaded?(Ecto.Type) do
 
     @types [
       "Point",
+      "PointZ",
       "LineString",
       "Polygon",
       "MultiPoint",

diff --git a/test/geo_postgis_test.exs b/test/geo_postgis_test.exs
@@ -48,6 +48,18 @@ defmodule Geo.PostGIS.Test do
     assert(result.rows == [[42, "test", geo]])
   end
 
+  test "insert pointz", context do
+    pid = context[:pid]
+    geo = %Geo.PointZ{coordinates: {30, -90, 70}, srid: 4326}
+
+    {:ok, _} =
+      Postgrex.query(pid, "CREATE TABLE point_test (id int, geom geometry(PointZ, 4326))", [])
+
+    {:ok, _} = Postgrex.query(pid, "INSERT INTO point_test VALUES ($1, $2)", [42, geo])
+    {:ok, result} = Postgrex.query(pid, "SELECT * FROM point_test", [])
+    assert(result.rows == [[42, geo]])
+  end
+
   test "insert linestring", context do
     pid = context[:pid]
     geo = %Geo.LineString{srid: 4326, coordinates: [{30, 10}, {10, 30}, {40, 40}]}