Schema version -- do formatted version like activerecord

Author: mcyoung

lib/data_migrate/schema_dumper.rb

@@ -15,7 +15,7 @@ def dump(connection = ActiveRecord::Base.connection, stream = $stdout)
     end
 
     def dump(stream)
-      define_params = @version ? "version: #{@version}" : ""
+      define_params = @version ? "version: #{formatted_version}" : ""
 
       stream.puts "DataMigrate::Data.define(#{define_params})"
 
@@ -34,5 +34,12 @@ def initialize(connection)
                     0
                   end
     end
+
+    # turns 20170404131909 into "2017_04_04_131909"
+    def formatted_version
+      stringified = @version.to_s
+      return stringified unless stringified.length == 14
+      stringified.insert(4, "_").insert(7, "_").insert(10, "_")
+    end
   end
 end

spec/data_migrate/schema_dumper_spec.rb

@@ -28,7 +28,7 @@
       DataMigrate::SchemaDumper.dump(ActiveRecord::Base.connection, stream)
       stream.rewind
 
-      last_version = fixture_file_timestamps.last
+      last_version = fixture_file_timestamps.last.dup.insert(4, "_").insert(7, "_").insert(10, "_")
       expected = "DataMigrate::Data.define(version: #{last_version})"
       expect(stream.read).to include expected
     end