1# 2# Licensed to the Apache Software Foundation (ASF) under one 3# or more contributor license agreements. See the NOTICE file 4# distributed with this work for additional information 5# regarding copyright ownership. The ASF licenses this file 6# to you under the Apache License, Version 2.0 (the 7# "License"); you may not use this file except in compliance 8# with the License. You may obtain a copy of the License at 9# 10# http://www.apache.org/licenses/LICENSE-2.0 11# 12# Unless required by applicable law or agreed to in writing, 13# software distributed under the License is distributed on an 14# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 15# KIND, either express or implied. See the License for the 16# specific language governing permissions and limitations 17# under the License. 18# 19 20require 'spec_helper' 21 22describe 'Exception' do 23 24 describe Thrift::Exception do 25 it "should have an accessible message" do 26 e = Thrift::Exception.new("test message") 27 expect(e.message).to eq("test message") 28 end 29 end 30 31 describe Thrift::ApplicationException do 32 it "should inherit from Thrift::Exception" do 33 expect(Thrift::ApplicationException.superclass).to eq(Thrift::Exception) 34 end 35 36 it "should have an accessible type and message" do 37 e = Thrift::ApplicationException.new 38 expect(e.type).to eq(Thrift::ApplicationException::UNKNOWN) 39 expect(e.message).to be_nil 40 e = Thrift::ApplicationException.new(Thrift::ApplicationException::UNKNOWN_METHOD, "test message") 41 expect(e.type).to eq(Thrift::ApplicationException::UNKNOWN_METHOD) 42 expect(e.message).to eq("test message") 43 end 44 45 it "should read a struct off of a protocol" do 46 prot = double("MockProtocol") 47 expect(prot).to receive(:read_struct_begin).ordered 48 expect(prot).to receive(:read_field_begin).exactly(3).times.and_return( 49 ["message", Thrift::Types::STRING, 1], 50 ["type", Thrift::Types::I32, 2], 51 [nil, Thrift::Types::STOP, 0] 52 ) 53 expect(prot).to receive(:read_string).ordered.and_return "test message" 54 expect(prot).to receive(:read_i32).ordered.and_return Thrift::ApplicationException::BAD_SEQUENCE_ID 55 expect(prot).to receive(:read_field_end).exactly(2).times 56 expect(prot).to receive(:read_struct_end).ordered 57 58 e = Thrift::ApplicationException.new 59 e.read(prot) 60 expect(e.message).to eq("test message") 61 expect(e.type).to eq(Thrift::ApplicationException::BAD_SEQUENCE_ID) 62 end 63 64 it "should skip bad fields when reading a struct" do 65 prot = double("MockProtocol") 66 expect(prot).to receive(:read_struct_begin).ordered 67 expect(prot).to receive(:read_field_begin).exactly(5).times.and_return( 68 ["type", Thrift::Types::I32, 2], 69 ["type", Thrift::Types::STRING, 2], 70 ["message", Thrift::Types::MAP, 1], 71 ["message", Thrift::Types::STRING, 3], 72 [nil, Thrift::Types::STOP, 0] 73 ) 74 expect(prot).to receive(:read_i32).and_return Thrift::ApplicationException::INVALID_MESSAGE_TYPE 75 expect(prot).to receive(:skip).with(Thrift::Types::STRING).twice 76 expect(prot).to receive(:skip).with(Thrift::Types::MAP) 77 expect(prot).to receive(:read_field_end).exactly(4).times 78 expect(prot).to receive(:read_struct_end).ordered 79 80 e = Thrift::ApplicationException.new 81 e.read(prot) 82 expect(e.message).to be_nil 83 expect(e.type).to eq(Thrift::ApplicationException::INVALID_MESSAGE_TYPE) 84 end 85 86 it "should write a Thrift::ApplicationException struct to the oprot" do 87 prot = double("MockProtocol") 88 expect(prot).to receive(:write_struct_begin).with("Thrift::ApplicationException").ordered 89 expect(prot).to receive(:write_field_begin).with("message", Thrift::Types::STRING, 1).ordered 90 expect(prot).to receive(:write_string).with("test message").ordered 91 expect(prot).to receive(:write_field_begin).with("type", Thrift::Types::I32, 2).ordered 92 expect(prot).to receive(:write_i32).with(Thrift::ApplicationException::UNKNOWN_METHOD).ordered 93 expect(prot).to receive(:write_field_end).twice 94 expect(prot).to receive(:write_field_stop).ordered 95 expect(prot).to receive(:write_struct_end).ordered 96 97 e = Thrift::ApplicationException.new(Thrift::ApplicationException::UNKNOWN_METHOD, "test message") 98 e.write(prot) 99 end 100 101 it "should skip nil fields when writing to the oprot" do 102 prot = double("MockProtocol") 103 expect(prot).to receive(:write_struct_begin).with("Thrift::ApplicationException").ordered 104 expect(prot).to receive(:write_field_begin).with("message", Thrift::Types::STRING, 1).ordered 105 expect(prot).to receive(:write_string).with("test message").ordered 106 expect(prot).to receive(:write_field_end).ordered 107 expect(prot).to receive(:write_field_stop).ordered 108 expect(prot).to receive(:write_struct_end).ordered 109 110 e = Thrift::ApplicationException.new(nil, "test message") 111 e.write(prot) 112 113 prot = double("MockProtocol") 114 expect(prot).to receive(:write_struct_begin).with("Thrift::ApplicationException").ordered 115 expect(prot).to receive(:write_field_begin).with("type", Thrift::Types::I32, 2).ordered 116 expect(prot).to receive(:write_i32).with(Thrift::ApplicationException::BAD_SEQUENCE_ID).ordered 117 expect(prot).to receive(:write_field_end).ordered 118 expect(prot).to receive(:write_field_stop).ordered 119 expect(prot).to receive(:write_struct_end).ordered 120 121 e = Thrift::ApplicationException.new(Thrift::ApplicationException::BAD_SEQUENCE_ID) 122 e.write(prot) 123 124 prot = double("MockProtocol") 125 expect(prot).to receive(:write_struct_begin).with("Thrift::ApplicationException").ordered 126 expect(prot).to receive(:write_field_stop).ordered 127 expect(prot).to receive(:write_struct_end).ordered 128 129 e = Thrift::ApplicationException.new(nil) 130 e.write(prot) 131 end 132 end 133 134 describe Thrift::ProtocolException do 135 it "should have an accessible type" do 136 prot = Thrift::ProtocolException.new(Thrift::ProtocolException::SIZE_LIMIT, "message") 137 expect(prot.type).to eq(Thrift::ProtocolException::SIZE_LIMIT) 138 expect(prot.message).to eq("message") 139 end 140 end 141end 142