Subversion Repositories SmartDukaan

#

# Licensed to the Apache Software Foundation (ASF) under one

# or more contributor license agreements. See the NOTICE file

# distributed with this work for additional information

# regarding copyright ownership. The ASF licenses this file

# to you under the Apache License, Version 2.0 (the

# "License"); you may not use this file except in compliance

# with the License. You may obtain a copy of the License at

#

#   http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing,

# software distributed under the License is distributed on an

# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

# KIND, either express or implied. See the License for the

# specific language governing permissions and limitations

# under the License.

#

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

class ThriftBaseProtocolSpec < Spec::ExampleGroup

  include Thrift

  before(:each) do

    @trans = mock("MockTransport")

    @prot = BaseProtocol.new(@trans)

  end

  describe BaseProtocol do

    # most of the methods are stubs, so we can ignore them

    it "should make trans accessible" do

      @prot.trans.should eql(@trans)

    end

    it "should write out a field nicely" do

      @prot.should_receive(:write_field_begin).with('field', 'type', 'fid').ordered

      @prot.should_receive(:write_type).with('type', 'value').ordered

      @prot.should_receive(:write_field_end).ordered

      @prot.write_field('field', 'type', 'fid', 'value')

    end

    it "should write out the different types" do

      @prot.should_receive(:write_bool).with('bool').ordered

      @prot.should_receive(:write_byte).with('byte').ordered

      @prot.should_receive(:write_double).with('double').ordered

      @prot.should_receive(:write_i16).with('i16').ordered

      @prot.should_receive(:write_i32).with('i32').ordered

      @prot.should_receive(:write_i64).with('i64').ordered

      @prot.should_receive(:write_string).with('string').ordered

      struct = mock('Struct')

      struct.should_receive(:write).with(@prot).ordered

      @prot.write_type(Types::BOOL, 'bool')

      @prot.write_type(Types::BYTE, 'byte')

      @prot.write_type(Types::DOUBLE, 'double')

      @prot.write_type(Types::I16, 'i16')

      @prot.write_type(Types::I32, 'i32')

      @prot.write_type(Types::I64, 'i64')

      @prot.write_type(Types::STRING, 'string')

      @prot.write_type(Types::STRUCT, struct)

      # all other types are not implemented

      [Types::STOP, Types::VOID, Types::MAP, Types::SET, Types::LIST].each do |type|

        lambda { @prot.write_type(type, type.to_s) }.should raise_error(NotImplementedError)

      end

    end

    it "should read the different types" do

      @prot.should_receive(:read_bool).ordered

      @prot.should_receive(:read_byte).ordered

      @prot.should_receive(:read_i16).ordered

      @prot.should_receive(:read_i32).ordered

      @prot.should_receive(:read_i64).ordered

      @prot.should_receive(:read_double).ordered

      @prot.should_receive(:read_string).ordered

      @prot.read_type(Types::BOOL)

      @prot.read_type(Types::BYTE)

      @prot.read_type(Types::I16)

      @prot.read_type(Types::I32)

      @prot.read_type(Types::I64)

      @prot.read_type(Types::DOUBLE)

      @prot.read_type(Types::STRING)

      # all other types are not implemented

      [Types::STOP, Types::VOID, Types::MAP, Types::SET, Types::LIST].each do |type|

        lambda { @prot.read_type(type) }.should raise_error(NotImplementedError)

      end

    end

    it "should skip the basic types" do

      @prot.should_receive(:read_bool).ordered

      @prot.should_receive(:read_byte).ordered

      @prot.should_receive(:read_i16).ordered

      @prot.should_receive(:read_i32).ordered

      @prot.should_receive(:read_i64).ordered

      @prot.should_receive(:read_double).ordered

      @prot.should_receive(:read_string).ordered

      @prot.skip(Types::BOOL)

      @prot.skip(Types::BYTE)

      @prot.skip(Types::I16)

      @prot.skip(Types::I32)

      @prot.skip(Types::I64)

      @prot.skip(Types::DOUBLE)

      @prot.skip(Types::STRING)

      @prot.skip(Types::STOP) # should do absolutely nothing

    end

    it "should skip structs" do

      real_skip = @prot.method(:skip)

      @prot.should_receive(:read_struct_begin).ordered

      @prot.should_receive(:read_field_begin).exactly(4).times.and_return(

        ['field 1', Types::STRING, 1],

        ['field 2', Types::I32, 2],

        ['field 3', Types::MAP, 3],

        [nil, Types::STOP, 0]

      )

      @prot.should_receive(:read_field_end).exactly(3).times

      @prot.should_receive(:read_string).exactly(3).times

      @prot.should_receive(:read_i32).ordered

      @prot.should_receive(:read_map_begin).ordered.and_return([Types::STRING, Types::STRING, 1])

      # @prot.should_receive(:read_string).exactly(2).times

      @prot.should_receive(:read_map_end).ordered

      @prot.should_receive(:read_struct_end).ordered

      real_skip.call(Types::STRUCT)

    end

    it "should skip maps" do

      real_skip = @prot.method(:skip)

      @prot.should_receive(:read_map_begin).ordered.and_return([Types::STRING, Types::STRUCT, 1])

      @prot.should_receive(:read_string).ordered

      @prot.should_receive(:read_struct_begin).ordered.and_return(["some_struct"])

      @prot.should_receive(:read_field_begin).ordered.and_return([nil, Types::STOP, nil]);

      @prot.should_receive(:read_struct_end).ordered

      @prot.should_receive(:read_map_end).ordered

      real_skip.call(Types::MAP)

    end

    it "should skip sets" do

      real_skip = @prot.method(:skip)

      @prot.should_receive(:read_set_begin).ordered.and_return([Types::I64, 9])

      @prot.should_receive(:read_i64).ordered.exactly(9).times

      @prot.should_receive(:read_set_end)

      real_skip.call(Types::SET)

    end

    it "should skip lists" do

      real_skip = @prot.method(:skip)

      @prot.should_receive(:read_list_begin).ordered.and_return([Types::DOUBLE, 11])

      @prot.should_receive(:read_double).ordered.exactly(11).times

      @prot.should_receive(:read_list_end)

      real_skip.call(Types::LIST)

    end

  end

  describe BaseProtocolFactory do

    it "should raise NotImplementedError" do

      # returning nil since Protocol is just an abstract class

      lambda {BaseProtocolFactory.new.get_protocol(mock("MockTransport"))}.should raise_error(NotImplementedError)

    end

  end

end