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