require 'rdf/spec'
require 'fileutils'
require 'tmpdir'

module RDF_Writer
  extend RSpec::SharedContext
  include RDF::Spec::Matchers

  before(:each) do
    raise '+@writer+ must be defined in a before(:each) block' unless instance_variable_get('@writer')
    @writer_class = @writer.class
  end

  describe RDF::Writer do
    describe ".each" do
      it "yields each writer" do
        @writer_class.each do |r|
          expect(r).not_to be_nil
        end
      end
    end
  
    describe ".buffer" do
      it "calls .new with buffer and other arguments" do
        expect(@writer_class).to receive(:new)
        @writer_class.buffer do |r|
          expect(r).to be_a(@writer_class)
        end
      end
    end

    describe ".open" do
      before(:each) do
        RDF::Util::File.stub(:open_file).and_yield(StringIO.new("foo"))
        @dir = Dir.mktmpdir
        @basename = File.join(@dir, "foo")
      end
    
      after(:each) do
        FileUtils.rm_rf(@dir)
      end

      it "yields writer given file_name" do
        @writer_class.format.each do |f|
          f.file_extensions.each_pair do |sym, content_type|
            writer_mock = double("writer")
            expect(writer_mock).to receive(:got_here)
            expect(@writer_class).to receive(:for).with(:file_name => "#{@basename}.#{sym}").and_return(@writer_class)
            @writer_class.open("#{@basename}.#{sym}") do |r|
              expect(r).to be_a(RDF::Writer)
              writer_mock.got_here
            end
          end
        end
      end

      it "yields writer given symbol" do
        @writer_class.format.each do |f|
          sym = f.to_sym  # Like RDF::NTriples::Format => :ntriples
          writer_mock = double("writer")
          expect(writer_mock).to receive(:got_here)
          expect(@writer_class).to receive(:for).with(sym).and_return(@writer_class)
          @writer_class.open("#{@basename}.#{sym}", :format => sym) do |r|
            expect(r).to be_a(RDF::Writer)
            writer_mock.got_here
          end
        end
      end

      it "yields writer given {:file_name => file_name}" do
        @writer_class.format.each do |f|
          f.file_extensions.each_pair do |sym, content_type|
            writer_mock = double("writer")
            expect(writer_mock).to receive(:got_here)
            expect(@writer_class).to receive(:for).with(:file_name => "#{@basename}.#{sym}").and_return(@writer_class)
            @writer_class.open("#{@basename}.#{sym}", :file_name => "#{@basename}.#{sym}") do |r|
              expect(r).to be_a(RDF::Writer)
              writer_mock.got_here
            end
          end
        end
      end

      it "yields writer given {:content_type => 'a/b'}" do
        @writer_class.format.each do |f|
          f.content_types.each_pair do |content_type, formats|
            writer_mock = double("writer")
            expect(writer_mock).to receive(:got_here)
            expect(@writer_class).to receive(:for).with(:content_type => content_type, :file_name => @basename).and_return(@writer_class)
            @writer_class.open(@basename, :content_type => content_type) do |r|
              expect(r).to be_a(RDF::Writer)
              writer_mock.got_here
            end
          end
        end
      end
    end

    describe ".format" do
      it "returns itself even if given explicit format" do
        other_format = @writer_class == RDF::NTriples::Writer ? :nquads : :ntriples
        expect(@writer_class.for(other_format)).to eq @writer_class
      end
    end

    describe ".new" do
      it "sets @output to $stdout by default" do
        writer_mock = double("writer")
        expect(writer_mock).to receive(:got_here)
        save, $stdout = $stdout, StringIO.new

        @writer_class.new do |r|
          writer_mock.got_here
          expect(r.instance_variable_get(:@output)).to eq $stdout
        end
        $stdout = save
      end
    
      it "sets @output to file given something other than a string" do
        writer_mock = double("writer")
        expect(writer_mock).to receive(:got_here)
        file = StringIO.new
        @writer_class.new(file) do |r|
          writer_mock.got_here
          expect(r.instance_variable_get(:@output)).to eq file
        end
      end
    
      it "sets prefixes given :prefixes => {}" do
        writer_mock = double("writer")
        expect(writer_mock).to receive(:got_here)
        @writer_class.new(StringIO.new, :prefixes => {:a => "b"}) do |r|
          writer_mock.got_here
          expect(r.prefixes).to eq({:a => "b"})
        end
      end
    
      #it "calls #write_prologue" do
      #  writer_mock = double("writer")
      #  writer_mock.should_receive(:got_here)
      #  @writer_class.any_instance.should_receive(:write_prologue)
      #  @writer_class.new(StringIO.new) do |r|
      #    writer_mock.got_here
      #  end
      #end
      #
      #it "calls #write_epilogue" do
      #  writer_mock = double("writer")
      #  writer_mock.should_receive(:got_here)
      #  @writer_class.any_instance.should_receive(:write_epilogue)
      #  @writer_class.new(StringIO.new) do |r|
      #    writer_mock.got_here
      #  end
      #end
    end
  
    describe "#prefixes=" do
      it "sets prefixes from hash" do
        @writer.prefixes = {:a => "b"}
        expect(@writer.prefixes).to eq({:a => "b"})
      end
    end
  
    describe "#prefix" do
      {
        nil     => "nil",
        :a      => "b",
        "foo"   => "bar",
      }.each_pair do |pfx, uri|
        it "sets prefix(#{pfx}) to #{uri}" do
          expect(@writer.prefix(pfx, uri)).to eq uri
          expect(@writer.prefix(pfx)).to eq uri
        end
      end
    end
  end
end