From ea0b73f90a98dae688a9b249cb5af93d50741fbd Mon Sep 17 00:00:00 2001
From: Rob A'Court <robacourt@users.noreply.github.com>
Date: Wed, 14 Aug 2024 12:05:45 +0100
Subject: [PATCH] chore(sync-service): Fix append_to_log! spec (#1521)

---
 packages/sync-service/lib/electric/shape_cache.ex         | 6 ++++--
 packages/sync-service/lib/electric/shape_cache/storage.ex | 6 +++---
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/packages/sync-service/lib/electric/shape_cache.ex b/packages/sync-service/lib/electric/shape_cache.ex
index 05f0f39e2c..971d25eef4 100644
--- a/packages/sync-service/lib/electric/shape_cache.ex
+++ b/packages/sync-service/lib/electric/shape_cache.ex
@@ -2,6 +2,7 @@ defmodule Electric.ShapeCacheBehaviour do
   @moduledoc """
   Behaviour defining the ShapeCache functions to be used in mocks
   """
+  alias Electric.LogItems
   alias Electric.ShapeCache.Storage
   alias Electric.Shapes.Shape
   alias Electric.Replication.LogOffset
@@ -13,7 +14,7 @@ defmodule Electric.ShapeCacheBehaviour do
   @callback append_to_log!(
               shape_id(),
               LogOffset.t(),
-              [Storage.prepared_change()],
+              [LogItems.log_item()],
               keyword()
             ) :: :ok
 
@@ -28,6 +29,7 @@ end
 
 defmodule Electric.ShapeCache do
   require Logger
+  alias Electric.LogItems
   alias Electric.Utils
   alias Electric.ShapeCache.Storage
   alias Electric.Shapes.Querying
@@ -87,7 +89,7 @@ defmodule Electric.ShapeCache do
   @spec append_to_log!(
           shape_id(),
           LogOffset.t(),
-          [Storage.prepared_change()],
+          [LogItems.log_item()],
           keyword()
         ) :: :ok
   def append_to_log!(shape_id, latest_offset, relevant_changes, opts) do
diff --git a/packages/sync-service/lib/electric/shape_cache/storage.ex b/packages/sync-service/lib/electric/shape_cache/storage.ex
index 19c3c9131e..0238134ee1 100644
--- a/packages/sync-service/lib/electric/shape_cache/storage.ex
+++ b/packages/sync-service/lib/electric/shape_cache/storage.ex
@@ -1,5 +1,5 @@
 defmodule Electric.ShapeCache.Storage do
-  alias Electric.LogItem
+  alias Electric.LogItems
   alias Electric.Shapes.Shape
   alias Electric.Replication.LogOffset
 
@@ -60,7 +60,7 @@ defmodule Electric.ShapeCache.Storage do
   @doc "Append log items from one transaction to the log"
   @callback append_to_log!(
               shape_id(),
-              [LogItem.t()],
+              [LogItems.log_item()],
               compiled_opts()
             ) :: :ok
   @doc "Get stream of the log for a shape since a given offset"
@@ -117,7 +117,7 @@ defmodule Electric.ShapeCache.Storage do
   @doc """
   Append log items from one transaction to the log
   """
-  @spec append_to_log!(shape_id(), [LogItem.t()], storage()) :: :ok
+  @spec append_to_log!(shape_id(), [LogItems.log_item()], storage()) :: :ok
   def append_to_log!(shape_id, log_items, {mod, opts}),
     do: mod.append_to_log!(shape_id, log_items, opts)