From 034882672ac158ee6564537ebfbb12201ab6e61d Mon Sep 17 00:00:00 2001
From: KKlochko <KKlochko@protonmail.com>
Date: Wed, 9 Apr 2025 21:54:46 +0300
Subject: [PATCH] Add the helpers for tests.

---
 test/support/data_case.ex |  1 +
 test/support/helpers.ex   | 20 ++++++++++++++++++++
 2 files changed, 21 insertions(+)
 create mode 100644 test/support/helpers.ex

diff --git a/test/support/data_case.ex b/test/support/data_case.ex
index d7ca407..3b93551 100644
--- a/test/support/data_case.ex
+++ b/test/support/data_case.ex
@@ -25,6 +25,7 @@ defmodule DecentralisedBookIndex.DataCase do
       import Ecto.Query
       import DecentralisedBookIndex.DataCase
       import DecentralisedBookIndex.Generators
+      import DecentralisedBookIndex.Helpers
     end
   end
 
diff --git a/test/support/helpers.ex b/test/support/helpers.ex
new file mode 100644
index 0000000..0cfae90
--- /dev/null
+++ b/test/support/helpers.ex
@@ -0,0 +1,20 @@
+defmodule DecentralisedBookIndex.Helpers do
+  @moduledoc "Helpers for tests"
+
+  @doc """
+  Rerurn a new map with same keys as in map_with_keys.
+  """
+  def get_submap(map, map_with_keys) when is_map(map_with_keys) do
+    Map.take(map, Map.keys(map_with_keys))
+  end
+
+  def get_submap(map, keys) when is_list(keys) do
+    Map.take(map, keys)
+  end
+
+  def get_submaps(map_list, keys) do
+    Enum.map(map_list, fn map ->
+      get_submap(map, keys)
+    end)
+  end
+end