# -*- coding:binary -*-
require 'spec_helper'

require 'rex/java'
require 'stringio'

RSpec.describe Rex::Java::Serialization::Model::BlockData do
  subject(:block) do
    described_class.new
  end

  let(:sample_block) { "\x10\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10" }
  let(:sample_block_io) { StringIO.new(sample_block) }
  let(:empty_block) { "\x00" }
  let(:empty_block_io) { StringIO.new(empty_block) }
  let(:incomplete_block) { "\x10\x01\x02\x03\x04\x05" }
  let(:incomplete_block_io) { StringIO.new(incomplete_block) }
  let(:empty_io) { StringIO.new('') }

  describe ".new" do
    it "Rex::Java::Serialization::Model::BlockData" do
      expect(block).to be_a(Rex::Java::Serialization::Model::BlockData)
    end

    it "initializes length to 0" do
      expect(block.length).to eq(0)
    end

    it "initializes contents with empty string" do
      expect(block.contents).to be_empty
    end
  end

  describe "#encode" do
    context "when empty block" do
      it { expect(block.encode).to eq(empty_block) }
    end

    context "when filled block" do
      it do
        block.length = 16
        block.contents = "\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10"
        expect(block.encode).to eq(sample_block)
      end
    end
  end

  describe "#decode" do
    context "when stream contains empty string" do
      it "returns nil" do
        expect { block.decode(empty_io) }.to raise_error(Rex::Java::Serialization::DecodeError)
      end
    end

    context "when stream contains empty block" do
      it "returns a Rex::Java::Serialization::Model::BlockData" do
        expect(block.decode(empty_block_io)).to be_a(Rex::Java::Serialization::Model::BlockData)
      end

      it "sets length to 0" do
        block.decode(empty_block_io)
        expect(block.length).to eq(0)
      end

      it "sets contents to empty string" do
        block.decode(empty_block_io)
        expect(block.contents).to be_empty
      end
    end

    context "when stream contains incomplete block" do
      it "returns nil" do
        expect { block.decode(incomplete_block_io) }.to raise_error(Rex::Java::Serialization::DecodeError)
      end
    end

    context "when stream contains correct block" do

      it "returns a Rex::Java::Serialization::Model::BlockData" do
        expect(block.decode(sample_block_io)).to be_a(Rex::Java::Serialization::Model::BlockData)
      end

      it "sets length to 0" do
        block.decode(sample_block_io)
        expect(block.length).to eq(16)
      end

      it "sets contents to sample string" do
        block.decode(sample_block_io)
        expect(block.contents).to eq("\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f\x10")
      end
    end
  end

  describe "#to_s" do
    it "prints a block with contents" do
      block.decode(sample_block_io)
      expect(block.to_s).to eq('[ 0x1, 0x2, 0x3, 0x4, 0x5, 0x6, 0x7, 0x8, 0x9, 0xa, 0xb, 0xc, 0xd, 0xe, 0xf, 0x10 ]')
    end

    it "prints an empty string for an empty block" do
      block.decode(empty_block_io)
      expect(block.to_s).to eq('[  ]')
    end
  end
end