SWAMPy

✎ CARVE FLESH

EDITING: video_recorder_spec.rb

require 'headless'

require 'tempfile'

describe Headless::VideoRecorder do
  before do
    stub_environment
  end

describe "instantiation" do

it "throws an error if provider_binary_path is not installed" do
      allow(Headless::CliUtil).to receive(:application_exists?).and_return(false)
      expect { Headless::VideoRecorder.new(99, "1024x768x32") }.to raise_error(Headless::Exception)
    end

it "allows provider_binary_path to be specified" do
      Tempfile.open('some_provider') do |f|
        v = Headless::VideoRecorder.new(99, "1024x768x32", provider: :ffmpeg, provider_binary_path: f.path)
        expect(v.provider_binary_path).to eq(f.path)
      end
    end

context "provider_binary_path not specified" do
      it "assumes the provider binary is 'ffmpeg' if the provider is :ffmpeg" do
        v = Headless::VideoRecorder.new(99, "1024x768x32", provider: :ffmpeg)
        expect(v.provider_binary_path).to eq("ffmpeg")
      end

it "assumes the provider binary is 'avconv' if the provider is :libav" do
        v = Headless::VideoRecorder.new(99, "1024x768x32", provider: :libav)
        expect(v.provider_binary_path).to eq("avconv")
      end

end
  end

describe "#capture" do
    before do
      allow(Headless::CliUtil).to receive(:path_to).and_return('ffmpeg')
    end

it "starts ffmpeg" do
      expect(Headless::CliUtil).to receive(:fork_process).with(/^ffmpeg -y -r 30 -s 1024x768 -f x11grab -i :99 -g 600 -vcodec qtrle [^ ]+$/, "/tmp/.headless_ffmpeg_99.pid", '/dev/null')

recorder = Headless::VideoRecorder.new(99, "1024x768x32")
      recorder.start_capture
    end

it "starts ffmpeg with specified codec" do
      expect(Headless::CliUtil).to receive(:fork_process).with(/^ffmpeg -y -r 30 -s 1024x768 -f x11grab -i :99 -g 600 -vcodec libvpx [^ ]+$/, "/tmp/.headless_ffmpeg_99.pid", '/dev/null')

recorder = Headless::VideoRecorder.new(99, "1024x768x32", {:codec => 'libvpx'})
      recorder.start_capture
    end

it "starts ffmpeg from ffmpeg provider with correct parameters" do
      expect(Headless::CliUtil).to receive(:fork_process).with(/^ffmpeg -y -r 30 -s 1024x768 -f x11grab -i :99 -vcodec qtrle [^ ]+$/, "/tmp/.headless_ffmpeg_99.pid", '/dev/null')

recorder = Headless::VideoRecorder.new(99, "1024x768x32", {:provider => :ffmpeg})
      recorder.start_capture
    end

it "starts ffmpeg with specified extra device options" do
      expect(Headless::CliUtil).to receive(:fork_process).with(/^ffmpeg -y -r 30 -s 1024x768 -f x11grab -draw_mouse 0 -i :99 -g 600 -vcodec qtrle [^ ]+$/, "/tmp/.headless_ffmpeg_99.pid", '/dev/null')

recorder = Headless::VideoRecorder.new(99, "1024x768x32", {:devices => ["-draw_mouse 0"]})
      recorder.start_capture
    end
  end

context "stopping video recording" do
    let(:tmpfile) { '/tmp/ci.mov' }
    let(:filename) { '/tmp/test.mov' }
    let(:pidfile) { '/tmp/pid' }

subject do
      recorder = Headless::VideoRecorder.new(99, "1024x768x32", :pid_file_path => pidfile, :tmp_file_path => tmpfile)
      recorder.start_capture
      recorder
    end

describe "using #stop_and_save" do
      it "stops video recording and saves file" do
        expect(Headless::CliUtil).to receive(:kill_process).with(pidfile, :wait => true)
        expect(File).to receive(:exists?).with(tmpfile).and_return(true)
        expect(FileUtils).to receive(:mv).with(tmpfile, filename)

subject.stop_and_save(filename)
      end
    end

describe "using #stop_and_discard" do
      it "stops video recording and deletes temporary file" do
        expect(Headless::CliUtil).to receive(:kill_process).with(pidfile, :wait => true)
        expect(FileUtils).to receive(:rm).with(tmpfile)

subject.stop_and_discard
      end
    end
  end

private

def stub_environment
    allow(Headless::CliUtil).to receive(:application_exists?).and_return(true)
    allow(Headless::CliUtil).to receive(:fork_process).and_return(true)
  end
end

CANCEL

Name	Type	Size	Modified	Actions
↩ ..	DIR	—	—
📄 headless_spec.rb	RB	8.9 KB	2023-09-24 03:47	EDIT
📄 integration_spec.rb	RB	1.1 KB	2023-09-24 03:47	EDIT
📄 video_recorder_spec.rb	RB	4.2 KB	2023-09-24 03:47	EDIT