Use seeded deletion strategy

Allow audit events and sessions changes, do pre-count

Author: jrafanie

cypress/e2e/app_commands/clean.rb

@@ -1,10 +1,12 @@
 if defined?(DatabaseCleaner)
   # cleaning the database using database_cleaner
-  DatabaseCleaner.strategy = :truncation
+  require 'extensions/database_cleaner-activerecord-seeded_deletion'
+  DatabaseCleaner[:active_record].strategy = DatabaseCleaner::ActiveRecord::SeededDeletion.new(:pre_count => true, :except => %w[audit_events sessions])
   DatabaseCleaner.clean
 else
-  logger.warn "add database_cleaner or update cypress/app_commands/clean.rb"
-  Post.delete_all if defined?(Post)
+  msg = "add database_cleaner or update cypress/app_commands/clean.rb"
+  logger.warn(msg)
+  raise msg
 end
 
 CypressOnRails::SmartFactoryWrapper.reload

lib/extensions/database_cleaner-activerecord-seeded_deletion.rb

@@ -0,0 +1,71 @@
+require 'database_cleaner-active_record'
+
+module DatabaseCleaner
+  module ActiveRecord
+    # SeededDeletion is a strategy that deletes all records from tables except those that existed before it was instantiated
+    # This is useful for tests that need the seeded data to be present.
+    class SeededDeletion < Deletion
+      def initialize(opts = {})
+        super
+        cache_max_seeded_ids
+      end
+
+      def self.table_max_id_cache
+        @table_max_id_cache ||= {}
+      end
+
+      # Memoize the maximum ID for each class table with non-zero number of rows
+      def self.table_max_id_cache=(table_id_hash)
+        @table_max_id_cache ||= table_id_hash
+      end
+
+      def cache_max_seeded_ids
+        self.class.table_max_id_cache = table_max_id_hash
+      end
+
+      delegate :table_max_id_cache, to: :class
+
+      private
+      def table_max_id_hash
+        h = {}
+        tables_to_clean(connection).each do |table_name|
+          begin
+            # Check if the table has an id column using SQL
+            column_exists_query = "SELECT EXISTS (
+              SELECT 1
+              FROM information_schema.columns
+              WHERE table_name = '#{table_name}'
+              AND column_name = 'id'
+            )"
+
+            has_id_column = connection.select_value(column_exists_query)
+
+            if has_id_column
+              # Get the maximum id using SQL
+              max_id = connection.select_value("SELECT MAX(id) FROM #{connection.quote_table_name(table_name)}")
+
+              # store the table and max id if there is non-zero number of rows
+              h[table_name] = max_id.to_i if max_id.to_i > 0
+            end
+          rescue => e
+            # Skip tables that don't exist or have other issues
+            puts "Warning: Could not cache max ID for table #{table_name}: #{e.message}"
+          end
+        end
+        h
+      end
+
+      # Override the delete_table to delete records with IDs greater than the cached max ID (new rows)
+      def delete_table(connection, table_name)
+        if table_max_id_cache.key?(table_name)
+          # For seedable tables, only delete records with IDs greater than the cached max ID
+          max_id = table_max_id_cache[table_name]
+          connection.execute("DELETE FROM #{connection.quote_table_name(table_name)} WHERE id > '#{max_id}'") if max_id.kind_of?(Numeric) && max_id > 0
+        else
+          # For all other tables, use the standard deletion from the parent class
+          super
+        end
+      end
+    end
+  end
+end