2 missing spec fixes.

ckho · Aug 17, 2016 · 03888f0 · 03888f0
1 parent 733608d
commit 03888f0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/test/lib/seedbank/dsl_test.rb b/test/lib/seedbank/dsl_test.rb
@@ -92,8 +92,9 @@
 
     it 'sets Rake Task description' do
       Seedbank::DSL.define_seed_task(seed_file, task_name => dependencies)
+      relative_file = Pathname.new(seed_file).relative_path_from(Rails.root)
 
-      Rake::Task[task_name].comment.must_equal "Load the seed data from #{seed_file}"
+      Rake::Task[task_name].comment.must_equal "Load the seed data from #{relative_file}"
     end
 
     it 'sets Rake Task action to the seed_file contents' do

diff --git a/test/lib/tasks/seed_rake_test.rb b/test/lib/tasks/seed_rake_test.rb
@@ -57,7 +57,7 @@ def self.glob_dummy_seeds
       def setup
         main = TOPLEVEL_BINDING.eval('class << self; self; end')
         orig = original_seeds_file
-        main.send(:undef_method, :original_seeds_file)
+        main.send(:undef_method, :original_seeds_file) if main.respond_to?(:original_seeds_file, true)
         main.send(:define_method, :original_seeds_file) { nil }
         super
         main.send(:undef_method, :original_seeds_file)