real_data_tests 0.3.15 → 0.3.16

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9b6d2be28f1fa0fc3982e0d8bb4927ea8f25eb1701d0c77059ec6a4059cb33ee
-  data.tar.gz: 19f5865fbb86760845c079a5b46ba49231533a33a91596e2b7c73438ca3941eb
+  metadata.gz: fbf98b8c1c4b72e614a490731c40ff4f56c36eb14b4b13cf19fc13b7c3022fe3
+  data.tar.gz: '029a17e72f0a42f34eae153cbf4a212e185100027e438b8a9e48efb112b7e349'
 SHA512:
-  metadata.gz: 6d68fbb46440b6950ec57363bbd995ea64503c41c9c66955ad1ad28b9338f51798a7aad945238e0a61c7654651a365b80056b3108e4dff1d2486b5c77cfad91c
-  data.tar.gz: e82c24d4082fb0d5e4e56e6de6cfd0e7130029057f2fa993e4f9bffdc2c5583bde94ee7c224535890f7e74ae01d8dd08ad670e2daef20041d572bf9845955fe8
+  metadata.gz: 854233f093058fe510b0c37409a6a4c3445a2c65a70dc9c862d3a8eaec12a535ba435b354a5348b8e4f41bf15291caacabcb1d9b4274b506f02d1baeec97e2b0
+  data.tar.gz: ce700cb666714010f2622307da5321fac568eb32afbaf6795121d855e6e47409047192a0c38d3b0ed1f3b8e82d109988135c7c8b6a6cfb63c6e8e274c79a686e

data/lib/real_data_tests/rspec_helper.rb

@@ -90,36 +90,80 @@ module RealDataTests
       options.join(" ")
     end
 
+    class SqlBlock
+      attr_reader :type, :content, :table_name
+
+      def initialize(content)
+        @content = content.strip
+        @type = determine_block_type
+        @table_name = extract_table_name if @type == :insert
+      end
+
+      private
+
+      def determine_block_type
+        if @content.match?(/\AINSERT INTO/i)
+          :insert
+        elsif @content.match?(/\ACOPY.*FROM stdin/i)
+          :copy
+        elsif @content.match?(/\AALTER TABLE/i)
+          :alter
+        elsif @content.match?(/\ASET/i)
+          :set
+        else
+          :other
+        end
+      end
+
+      def extract_table_name
+        if @content =~ /INSERT INTO\s+"?([^\s"(]+)"?\s/i
+          $1
+        end
+      end
+    end
+
     def parse_sql_blocks(content)
       blocks = []
       current_block = []
       in_copy_block = false
 
       content.each_line do |line|
-        line = line.strip
-        next if line.empty? || line.start_with?('--')
+        line = line.chomp
+
+        # Skip empty lines and comments unless in COPY block
+        next if !in_copy_block && (line.empty? || line.start_with?('--'))
 
-        if line.upcase.start_with?('COPY') && line.upcase.include?('FROM stdin')
+        # Handle start of COPY block
+        if !in_copy_block && line.upcase.match?(/\ACOPY.*FROM stdin/i)
+          current_block = [line]
           in_copy_block = true
+          next
+        end
+
+        # Handle end of COPY block
+        if in_copy_block && line == '\\.'
           current_block << line
-        elsif line == '\.' && in_copy_block
+          blocks << SqlBlock.new(current_block.join("\n"))
+          current_block = []
           in_copy_block = false
+          next
+        end
+
+        # Accumulate lines in COPY block
+        if in_copy_block
           current_block << line
+          next
+        end
+
+        # Handle regular SQL statements
+        current_block << line
+        if line.end_with?(';')
           blocks << SqlBlock.new(current_block.join("\n"))
           current_block = []
-        elsif in_copy_block
-          current_block << line
-        else
-          # Handle regular SQL statements
-          current_block << line
-          if line.end_with?(';') && !in_copy_block
-            blocks << SqlBlock.new(current_block.join("\n"))
-            current_block = []
-          end
         end
       end
 
-      # Add any remaining block
+      # Handle any remaining block
       blocks << SqlBlock.new(current_block.join("\n")) unless current_block.empty?
       blocks
     end
@@ -137,8 +181,24 @@ module RealDataTests
 
     def execute_insert_block(block, index, total)
       puts "Executing INSERT block #{index}/#{total} for table: #{block.table_name}"
-      statement = normalize_insert_statement(block.content)
-      ActiveRecord::Base.connection.execute(statement)
+      # Don't modify statements that already end with semicolon
+      statement = if block.content.strip.end_with?(';')
+        block.content
+      else
+        "#{block.content};"
+      end
+
+      begin
+        ActiveRecord::Base.connection.execute(statement)
+      rescue ActiveRecord::StatementInvalid => e
+        if e.message.include?('syntax error at or near "ON"')
+          # Try alternative formatting for ON CONFLICT
+          modified_statement = statement.gsub(/\)\s+ON\s+CONFLICT/, ') ON CONFLICT')
+          ActiveRecord::Base.connection.execute(modified_statement)
+        else
+          raise
+        end
+      end
     end
 
     def execute_copy_block(block, index, total)
@@ -152,13 +212,22 @@ module RealDataTests
     end
 
     def normalize_insert_statement(statement)
-      # Handle ON CONFLICT clauses properly
-      if statement =~ /(.*?)\s+(ON\s+CONFLICT.*?)\s*;\s*\z/i
-        main_insert = $1
-        conflict_clause = $2
-        "#{main_insert} #{conflict_clause};"
+      # First clean up any excess whitespace around parentheses
+      statement = statement.gsub(/\(\s+/, '(')
+                          .gsub(/\s+\)/, ')')
+                          .gsub(/\)\s+ON\s+CONFLICT/, ') ON CONFLICT')
+
+      # Ensure proper spacing around ON CONFLICT
+      if statement =~ /(.*?)\s*ON\s+CONFLICT\s+(.*?)\s*(?:DO\s+.*?)?\s*;\s*\z/i
+        base = $1.strip
+        conflict_part = $2.strip
+        action_part = $3&.strip || 'DO NOTHING'
+
+        # Rebuild the statement with consistent formatting
+        "#{base} ON CONFLICT #{conflict_part} #{action_part};"
       else
-        statement
+        # If no ON CONFLICT clause, just clean up the spacing
+        statement.strip.sub(/;?\s*$/, ';')
       end
     end
 

data/lib/real_data_tests/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RealDataTests
-  VERSION = "0.3.15"
+  VERSION = "0.3.16"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: real_data_tests
 version: !ruby/object:Gem::Version
-  version: 0.3.15
+  version: 0.3.16
 platform: ruby
 authors:
 - Kevin Dias