|
| 1 | +# SPDX-License-Identifier: Apache-2.0 |
| 2 | +# SPDX-FileCopyrightText: 2025 DBVisor |
| 3 | + |
| 4 | +defmodule Mix.Tasks.Sql.Gen.Test do |
| 5 | + use Mix.Task |
| 6 | + import Mix.Generator |
| 7 | + |
| 8 | + @shortdoc "Generates test from the BNF rules" |
| 9 | + @shortdoc since: "0.2.0" |
| 10 | + |
| 11 | + def run([base]) do |
| 12 | + create_file("test/conformance/e_test.exs", test_template([mod: SQL.Conformance.ETest, dir: Path.join(base, "E")])) |
| 13 | + create_file("test/conformance/f_test.exs", test_template([mod: SQL.Conformance.FTest, dir: Path.join(base, "F")])) |
| 14 | + create_file("test/conformance/s_test.exs", test_template([mod: SQL.Conformance.STest, dir: Path.join(base, "S")])) |
| 15 | + create_file("test/conformance/t_test.exs", test_template([mod: SQL.Conformance.TTest, dir: Path.join(base, "T")])) |
| 16 | + end |
| 17 | + |
| 18 | + def generate_test(dir) do |
| 19 | + for path <- File.ls!(dir), path =~ ".tests.yml", [{~c"feature", feature}, {~c"id", id}, {~c"sql", sql}] <- :yamerl.decode_file(to_charlist(Path.join(dir, path))) do |
| 20 | + statements = if is_list(hd(sql)), do: sql, else: [sql] |
| 21 | + statements = Enum.map(statements, &String.replace(to_string(&1), ~r{(VARING)}, "VARYING")) |
| 22 | + {"#{feature} #{id}", Enum.map(statements, &{trim(&1), &1})} |
| 23 | + end |
| 24 | + end |
| 25 | + |
| 26 | + def trim(value) do |
| 27 | + value |
| 28 | + |> String.replace(~r{\(\s+\b}, &String.replace(&1, " ", "")) |
| 29 | + |> String.replace(~r{\(\s+'}, &String.replace(&1, " ", "")) |
| 30 | + |> String.replace(~r{\(\s+"}, &String.replace(&1, " ", "")) |
| 31 | + |> String.replace(~r{\(\s+\*}, &String.replace(&1, " ", "")) |
| 32 | + |> String.replace(~r{[[:alpha:]]+\s+\(}, &String.replace(&1, " ", "")) |
| 33 | + |> String.replace(~r{\b\s+\,}, &String.replace(&1, " ", "")) |
| 34 | + |> String.replace(~r{\)\s+\,}, &String.replace(&1, " ", "")) |
| 35 | + |> String.replace(~r{\'\s+\,}, &String.replace(&1, " ", "")) |
| 36 | + |> String.replace(~r{\b\s+\)}, &String.replace(&1, " ", "")) |
| 37 | + |> String.replace(~r{'\s+\)}, &String.replace(&1, " ", "")) |
| 38 | + |> String.replace(~r{\*\s+\)}, &String.replace(&1, " ", "")) |
| 39 | + |> String.replace(~r{\)\s+\)}, &String.replace(&1, " ", "")) |
| 40 | + |> String.replace(~r{\W(SELECT|REFERENCES|INSERT|UPDATE|IN|MYTEMP)\(}, &Enum.join(Regex.split(~r{\(}, &1, include_captures: true, trim: true), " ")) |
| 41 | + |> String.replace(~r{^(SELECT)\(}, &Enum.join(Regex.split(~r{\(}, &1, include_captures: true, trim: true), " ")) |
| 42 | + |> String.replace(~r{\s+\.\s+}, &String.replace(&1, " ", "")) |
| 43 | + |> String.replace(~r{\d\s(\+|\-)\d}, &Enum.join(Enum.map(Regex.split(~r{\+|\-}, &1, include_captures: true, trim: true), fn x -> String.trim(x) end), " ")) |
| 44 | + |> String.trim() |
| 45 | + end |
| 46 | + |
| 47 | + embed_template(:test, """ |
| 48 | + # SPDX-License-Identifier: Apache-2.0 |
| 49 | + # SPDX-FileCopyrightText: 2025 DBVisor |
| 50 | +
|
| 51 | + defmodule <%= inspect @mod %>.Adapter do |
| 52 | + use SQL.Token |
| 53 | +
|
| 54 | + def token_to_string(value, mod \\\\ __MODULE__) |
| 55 | + def token_to_string(value, _mod) when is_atom(value), do: String.upcase(Atom.to_string(value)) |
| 56 | + def token_to_string(token, mod), do: SQL.Adapters.ANSI.token_to_string(token, mod) |
| 57 | + end |
| 58 | + defmodule <%= inspect @mod %> do |
| 59 | + use ExUnit.Case, async: true |
| 60 | + use SQL, adapter: <%= inspect @mod %>.Adapter |
| 61 | +
|
| 62 | + <%= for {name, statements} <- generate_test(@dir) do %> |
| 63 | + test <%= inspect name %> do |
| 64 | + <%= for {left, right} <- statements do %> |
| 65 | + assert ~s{<%= left %>} == to_string(~SQL[<%= right %>]) |
| 66 | + <% end %> |
| 67 | + end |
| 68 | + <% end %> |
| 69 | + end |
| 70 | + """) |
| 71 | +end |
0 commit comments