Merge pull request ManageIQ#605 from NickLaMuro/add_timestamps_to_bin…

…ary_blobs

Adds timestamps to binary blobs

db/migrate/20210824231354_add_timestamps_to_binary_blobs.rb

@@ -0,0 +1,5 @@
+class AddTimestampsToBinaryBlobs < ActiveRecord::Migration[6.0]
+  def change
+    add_timestamps :binary_blobs, :null => false, :default => -> { 'NOW()' }
+  end
+end

spec/migrations/20210824231354_add_timestamps_to_binary_blobs_spec.rb

@@ -0,0 +1,38 @@
+require_migration
+
+describe AddTimestampsToBinaryBlobs do
+  let(:binary_blob_stub) { new_migration_stub(:binary_blobs) }
+
+  migration_context :up do
+    it "adds timestamp columns and auto-populates the value" do
+      record = binary_blob_stub.create!
+      expect(record.respond_to?(:created_at)).to be false
+      expect(record.respond_to?(:updated_at)).to be false
+
+      before_migration_time = 5.seconds.ago.utc
+
+      migrate
+
+      record.reload
+      expect(record.created_at).to be > before_migration_time
+      expect(record.updated_at).to be > before_migration_time
+    end
+  end
+
+  migration_context :down do
+    it "removes timestamp columns" do
+      before_migration_time = 5.seconds.ago.utc
+
+      record = binary_blob_stub.create!
+
+      expect(record.created_at).to be > before_migration_time
+      expect(record.updated_at).to be > before_migration_time
+
+      migrate
+
+      record.reload
+      expect(record.respond_to?(:created_at)).to be false
+      expect(record.respond_to?(:updated_at)).to be false
+    end
+  end
+end

spec/support/migration_helper.rb

@@ -59,6 +59,10 @@ def migration_stub(klass)
         stub
       end
 
+      def new_migration_stub(table_name)
+        Class.new(ActiveRecord::Base) { self.table_name = table_name.to_s }
+      end
+
       private
 
       # Clears any cached column information on stubs, since the migrations