spec/datamosh_spec.rb in aviglitch-0.1.6 vs spec/datamosh_spec.rb in aviglitch-0.2.0

@@ -1,55 +1,45 @@
 require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
 
 describe AviGlitch, 'datamosh cli' do
 
   before :all do
-    FileUtils.mkdir OUTPUT_DIR unless File.exist? OUTPUT_DIR
-    @in = FILES_DIR + 'sample.avi'
-    @out = OUTPUT_DIR + 'out.avi'
     here = File.dirname(__FILE__)
     lib = Pathname.new(File.join(here, '..', 'lib')).realpath
     datamosh = Pathname.new(File.join(here, '..', 'bin/datamosh')).realpath
     @cmd = "ruby -I%s %s -o %s " % [lib, datamosh, @out]
   end
 
-  after :each do
-    FileUtils.rm Dir.glob((OUTPUT_DIR + '*').to_s)
-  end
-
-  after :all do
-    FileUtils.rmdir OUTPUT_DIR
-  end
-
   it 'should correctly process files' do
     a = AviGlitch.open @in
     keys = a.frames.inject(0) do |c, f|
       c += 1 if f.is_keyframe?
       c
     end
     total = a.frames.size
+    first_keyframe = a.frames.index(a.frames.first_of(:keyframe))
     a.close
 
     system [@cmd, @in].join(' ')
     o = AviGlitch.open @out
     o.frames.size.should == total
-    o.frames.first.is_keyframe?.should be true
+    o.frames[first_keyframe].is_keyframe?.should be true
     o.has_keyframe?.should be true
     o.close
     AviGlitch::Base.surely_formatted?(@out, true).should be true
 
     system [@cmd, '-a', @in].join(' ')
     o = AviGlitch.open @out
     o.frames.size.should == total
-    o.frames.first.is_keyframe?.should be false
+    o.frames[first_keyframe].is_keyframe?.should be false
     o.has_keyframe?.should be false
     o.close
     AviGlitch::Base.surely_formatted?(@out, true).should be true
 
     system [@cmd, @in, @in, @in].join(' ')
     o = AviGlitch.open @out
     o.frames.size.should == total * 3
-    o.frames.first.is_keyframe?.should be true
+    o.frames[first_keyframe].is_keyframe?.should be true
     o.close
     AviGlitch::Base.surely_formatted?(@out, true).should be true
 
     system [@cmd, '--fake', @in].join(' ')
     o = AviGlitch.open @out