add scalar RowBinary integration tests

Author: ruslandoga

lib/ch/http.ex

@@ -177,9 +177,9 @@ defmodule Ch.HTTP do
   defp encode_array_param(nil), do: "null"
 
   # DateTime64 array literals parse unquoted numbers as scaled ticks, while
-  # scalar query params parse the same text as seconds. Quote DateTime values so
+  # scalar query params parse the same text as seconds. Quote date/time values so
   # ClickHouse uses timestamp parsing inside arrays, tuples, and maps too.
-  defp encode_array_param(%s{} = param) when s in [Date, NaiveDateTime, DateTime] do
+  defp encode_array_param(%s{} = param) when s in [Date, Time, NaiveDateTime, DateTime] do
     [?', encode_param(param), ?']
   end
 

test/ch/query_string_test.exs

@@ -402,7 +402,7 @@ defmodule Ch.QueryStringTest do
 
   defp expected_array_param(nil), do: "null"
 
-  defp expected_array_param(%value{} = param) when value in [Date, NaiveDateTime, DateTime],
+  defp expected_array_param(%value{} = param) when value in [Date, Time, NaiveDateTime, DateTime],
     do: "'" <> expected_param(param) <> "'"
 
   defp expected_array_param(value), do: expected_param(value)

test/ch/row_binary_bool_test.exs

@@ -0,0 +1,110 @@
+defmodule Ch.RowBinaryBoolTest do
+  use ExUnit.Case, async: true
+  use ExUnitProperties
+
+  alias Ch.RowBinary
+
+  setup do
+    {:ok, pool: start_supervised!(Ch)}
+  end
+
+  property "Bool params round-trip through ClickHouse", %{pool: pool} do
+    check all value <- boolean() do
+      assert Ch.query!(pool, "SELECT {value:Bool}", %{"value" => value}).rows == [[value]]
+    end
+  end
+
+  property "Bool arrays round-trip as query params through ClickHouse", %{pool: pool} do
+    check all values <- list_of(boolean(), max_length: 16) do
+      assert Ch.query!(pool, "SELECT {value:Array(Bool)}", %{"value" => values}).rows == [
+               [values]
+             ]
+    end
+  end
+
+  test "query params cover nullable, empty arrays, and invalid Bool values", %{pool: pool} do
+    assert Ch.query!(
+             pool,
+             "SELECT {t:Bool}, {f:Bool}, {n:Nullable(Bool)}, {empty:Array(Bool)}",
+             %{"t" => true, "f" => false, "n" => nil, "empty" => []}
+           ).rows == [[true, false, nil, []]]
+
+    assert {:error, %Ch.Error{message: message}} =
+             Ch.query(pool, "SELECT {value:Bool}", %{"value" => "not-a-bool"})
+
+    assert message =~ "Bool"
+  end
+
+  property "RowBinary Bool inserts round-trip through ClickHouse", %{pool: pool} do
+    Help.query!("""
+    CREATE TABLE row_binary_bool_property (
+      id UInt8,
+      value Bool
+    ) ENGINE Memory
+    """)
+
+    on_exit(fn -> Help.query!("DROP TABLE row_binary_bool_property") end)
+
+    check all rows <- rowbinary_bool_rows() do
+      Ch.query!(pool, "TRUNCATE TABLE row_binary_bool_property")
+
+      rowbinary = RowBinary.encode_rows(rows, ["UInt8", "Bool"])
+      Ch.query!(pool, ["INSERT INTO row_binary_bool_property FORMAT RowBinary\n" | rowbinary])
+
+      assert Ch.query!(pool, "SELECT * FROM row_binary_bool_property ORDER BY id").rows ==
+               Enum.sort_by(rows, &List.first/1)
+    end
+  end
+
+  test "RowBinary inserts cover nullable, arrays, tuples, maps, and defaults", %{pool: pool} do
+    Help.query!("""
+    CREATE TABLE row_binary_bool_representative (
+      id UInt8,
+      value Bool,
+      nullable Nullable(Bool),
+      bools Array(Bool),
+      pair Tuple(Bool, Bool),
+      mapped Map(String, Bool)
+    ) ENGINE Memory
+    """)
+
+    on_exit(fn -> Help.query!("DROP TABLE row_binary_bool_representative") end)
+
+    rows = [
+      [1, true, nil, [], {true, false}, %{}],
+      [2, false, true, [true, false, true], {false, true}, %{"a" => true, "b" => false}],
+      [3, nil, false, [false], {true, true}, %{"zero" => false}]
+    ]
+
+    types = [
+      "UInt8",
+      "Bool",
+      "Nullable(Bool)",
+      "Array(Bool)",
+      "Tuple(Bool, Bool)",
+      "Map(String, Bool)"
+    ]
+
+    rowbinary = RowBinary.encode_rows(rows, types)
+    Ch.query!(pool, ["INSERT INTO row_binary_bool_representative FORMAT RowBinary\n" | rowbinary])
+
+    assert Ch.query!(pool, "SELECT * FROM row_binary_bool_representative ORDER BY id").rows == [
+             [1, true, nil, [], {true, false}, %{}],
+             [2, false, true, [true, false, true], {false, true}, %{"a" => true, "b" => false}],
+             [3, false, false, [false], {true, true}, %{"zero" => false}]
+           ]
+  end
+
+  test "RowBinary rejects invalid Bool values" do
+    assert_raise FunctionClauseError, fn ->
+      RowBinary.encode_rows([["true"]], ["Bool"])
+    end
+  end
+
+  defp rowbinary_bool_rows do
+    gen all ids <- uniq_list_of(integer(0..255), max_length: 32),
+            values <- list_of(boolean(), length: length(ids)) do
+      Enum.zip_with(ids, values, fn id, value -> [id, value] end)
+    end
+  end
+end