require File.expand_path(File.dirname(__FILE__) + '/../../spec_helper')

describe Win32::Screenshot::Image do

  before :all do
    IO.popen("notepad")
    @notepad = RAutomation::Window.new(:title => /notepad/i).pid
    @image = Win32::Screenshot::Take.of(:window, :pid => @notepad)
  end

  describe "stores raw bitmap data" do
    it "#bitmap" do
      expect(@image.bitmap[0..1]).to eq('BM')
    end
  end

  describe "saving the image to the disk" do
    context "saving to the bmp format" do
      it "#write" do
        file_path = @temp_dir + '/image.bmp'
        expect {@image.write(file_path)}.to_not raise_exception
        expect(File.open(file_path, "rb") {|io| io.read}[0..1]).to eq("BM")
      end
    end

    context "saving to the png format" do
      it "#write" do
        file_path = @temp_dir + '/image.png'
        expect {@image.write(file_path)}.to_not raise_exception
        expect(File.open(file_path, "rb") {|io| io.read}[0..3]).to eq("\211PNG".force_encoding(Encoding::ASCII_8BIT))
      end
    end

    context "trying to save to the unknown format" do
      it "#write" do
        file_path = @temp_dir + '/image.notknown'
        expect {@image.write(file_path)}.
                to raise_exception("File '#{file_path}' has to have one of the following extensions: #{Win32::Screenshot::Image::FORMATS.join(", ")}")
        expect(File.exist?(file_path)).to eq(false)
      end
    end

    context "trying to save with the filename without an extension" do
      it "#write" do
        file_path = @temp_dir + '/image'
        expect {@image.write(file_path)}.
                to raise_exception("File '#{file_path}' has to have one of the following extensions: #{Win32::Screenshot::Image::FORMATS.join(", ")}")
        expect(File.exist?(file_path)).to eq(false)
      end
    end

    context "not allowing to overwrite existing files" do
      it "#write" do
        file_path = @temp_dir + '/invalid-image.png'
        content = "invalid content"
        File.open(file_path, "w") {|io| io.write content}
        expect {@image.write(file_path)}.to raise_exception("File already exists: #{file_path}!")
        expect(File.size(file_path)).to eq(content.size)
      end
    end

    context "allowing to overwrite existing files when desired" do
      it "#write!" do
        file_path = @temp_dir + '/overwritten-image.png'
        content = "content"
        File.open(file_path, "w") {|io| io.write content}
        @image.write!(file_path)
        expect(File.size(file_path)).to_not eq(content.size)
      end
    end    
  end

  after :all do
    # kill in a jruby friendly way
    system("taskkill /PID #{@notepad} > nul")
  end

end