require "helper"

describe HTTP2::Header::Huffman do
  huffman_examples = [# plain, encoded
    ["www.example.com", "f1e3c2e5f23a6ba0ab90f4ff"],
    ["no-cache",        "a8eb10649cbf"],
    ["Mon, 21 Oct 2013 20:13:21 GMT", "d07abe941054d444a8200595040b8166e082a62d1bff"],
  ]
  context "encode" do
    before(:all) { @encoder = HTTP2::Header::Huffman.new }
    huffman_examples.each do |plain, encoded|
      it "should encode #{plain} into #{encoded}" do
        @encoder.encode(plain).unpack("H*").first.should eq encoded
      end
    end
  end
  context "decode" do
    before(:all) { @encoder = HTTP2::Header::Huffman.new }
    huffman_examples.each do |plain, encoded|
      it "should decode #{encoded} into #{plain}" do
        @encoder.decode(HTTP2::Buffer.new([encoded].pack("H*"))).should eq plain
      end
    end

    [
      "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.8; rv:16.0) Gecko/20100101 Firefox/16.0",
      "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
      "http://www.craigslist.org/about/sites/",
      "cl_b=AB2BKbsl4hGM7M4nH5PYWghTM5A; cl_def_lang=en; cl_def_hp=shoals",
      "image/png,image/*;q=0.8,*/*;q=0.5",
      "BX=c99r6jp89a7no&b=3&s=q4; localization=en-us%3Bus%3Bus",
      "UTF-8でエンコードした日本語文字列",
    ].each do |string|
      it "should encode then decode '#{string}' into the same" do
        s = string.dup.force_encoding('binary')
        encoded = @encoder.encode(s)
        @encoder.decode(HTTP2::Buffer.new(encoded)).should eq s
      end
    end

    it "should encode/decode all_possible 2-byte sequences" do
      (2**16).times do |n|
        str = [n].pack("V")[0,2].force_encoding('binary')
        @encoder.decode(HTTP2::Buffer.new(@encoder.encode(str))).should eq str
      end
    end

    it "should raise when input is shorter than expected" do
      plain, encoded = huffman_examples[0]
      encoded = [encoded].pack("H*")
      expect { @encoder.decode(HTTP2::Buffer.new(encoded[0...-1])) }.to raise_error(/EOS invalid/)
    end
    it "should raise when input is not padded by 1s" do
      plain, encoded = ["www.example.com", "f1e3c2e5f23a6ba0ab90f4fe"] # note the fe at end
      encoded = [encoded].pack("H*")
      expect { @encoder.decode(HTTP2::Buffer.new(encoded)) }.to raise_error(/EOS invalid/)
    end
    it "should raise when exceedingly padded" do
      plain, encoded = ["www.example.com", "e7cf9bebe89b6fb16fa9b6ffff"] # note the extra ff
      encoded = [encoded].pack("H*")
      expect { @encoder.decode(HTTP2::Buffer.new(encoded)) }.to raise_error(/EOS invalid/)
    end
    it "should raise when EOS is explicitly encoded" do
      encoded = ["1c7fffffffff"].pack("H*") # a b EOS
      expect { @encoder.decode(HTTP2::Buffer.new(encoded)) }.to raise_error(/EOS found/)
    end
  end
end