Merge pull request #168 from jpmcgrath/165-generator

Add spec for ShortenerGenerator
jpmcgrath · Feb 19, 2024 · 0a7d557 · 0a7d557
2 parents 95d06a5 + a589fee
commit 0a7d557
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lib/generators/shortener/shortener_generator.rb b/lib/generators/shortener/shortener_generator.rb
@@ -9,7 +9,7 @@ def self.source_root
   end
 
   def self.next_migration_number(dirname)
-    if ActiveRecord::Base.timestamped_migrations
+    if ActiveRecord.respond_to?(:timestamped_migrations) ? ActiveRecord.timestamped_migrations : ActiveRecord::Base.timestamped_migrations
       Time.new.utc.strftime("%Y%m%d%H%M%S")
     else
       "%.3d" % (current_migration_number(dirname) + 1)

diff --git a/spec/generators/shortener_generator_spec.rb b/spec/generators/shortener_generator_spec.rb
@@ -0,0 +1,10 @@
+require "spec_helper"
+require File.expand_path("../../../lib/generators/shortener/shortener_generator.rb", __FILE__)
+
+RSpec.describe ShortenerGenerator do
+  describe ".next_migration_number" do
+    it "returns the next migration number" do
+      expect { described_class.next_migration_number(".") }.not_to raise_error
+    end
+  end
+end