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 20package common 21 22import ( 23 "context" 24 "errors" 25 "gen/thrifttest" 26 "reflect" 27 "sync" 28 "testing" 29 "thrift" 30 31 "github.com/golang/mock/gomock" 32) 33 34type test_unit struct { 35 host string 36 port int64 37 domain_socket string 38 transport string 39 protocol string 40 ssl bool 41} 42 43var units = []test_unit{ 44 {"127.0.0.1", 9095, "", "", "binary", false}, 45 {"127.0.0.1", 9091, "", "", "compact", false}, 46 {"127.0.0.1", 9092, "", "", "binary", true}, 47 {"127.0.0.1", 9093, "", "", "compact", true}, 48} 49 50func TestAllConnection(t *testing.T) { 51 certPath = "../../../keys" 52 wg := &sync.WaitGroup{} 53 wg.Add(len(units)) 54 for _, unit := range units { 55 go func(u test_unit) { 56 defer wg.Done() 57 doUnit(t, &u) 58 }(unit) 59 } 60 wg.Wait() 61} 62 63func doUnit(t *testing.T, unit *test_unit) { 64 ctrl := gomock.NewController(t) 65 defer ctrl.Finish() 66 handler := NewMockThriftTest(ctrl) 67 68 processor, serverTransport, transportFactory, protocolFactory, err := GetServerParams(unit.host, unit.port, unit.domain_socket, unit.transport, unit.protocol, unit.ssl, "../../../keys", handler) 69 70 server := thrift.NewTSimpleServer4(processor, serverTransport, transportFactory, protocolFactory) 71 if err = server.Listen(); err != nil { 72 t.Errorf("Unable to start server: %v", err) 73 return 74 } 75 go server.AcceptLoop() 76 defer server.Stop() 77 client, trans, err := StartClient(unit.host, unit.port, unit.domain_socket, unit.transport, unit.protocol, unit.ssl) 78 if err != nil { 79 t.Errorf("Unable to start client: %v", err) 80 return 81 } 82 defer trans.Close() 83 callEverythingWithMock(t, client, handler) 84} 85 86var rmapmap = map[int32]map[int32]int32{ 87 -4: map[int32]int32{-4: -4, -3: -3, -2: -2, -1: -1}, 88 4: map[int32]int32{4: 4, 3: 3, 2: 2, 1: 1}, 89} 90 91var xxs = &thrifttest.Xtruct{ 92 StringThing: "Hello2", 93 ByteThing: 42, 94 I32Thing: 4242, 95 I64Thing: 424242, 96} 97 98var xcept = &thrifttest.Xception{ErrorCode: 1001, Message: "some"} 99var defaultCtx = context.Background() 100 101func callEverythingWithMock(t *testing.T, client *thrifttest.ThriftTestClient, handler *MockThriftTest) { 102 gomock.InOrder( 103 handler.EXPECT().TestVoid(gomock.Any()), 104 handler.EXPECT().TestString(gomock.Any(), "thing").Return("thing", nil), 105 handler.EXPECT().TestBool(gomock.Any(), true).Return(true, nil), 106 handler.EXPECT().TestBool(gomock.Any(), false).Return(false, nil), 107 handler.EXPECT().TestByte(gomock.Any(), int8(42)).Return(int8(42), nil), 108 handler.EXPECT().TestI32(gomock.Any(), int32(4242)).Return(int32(4242), nil), 109 handler.EXPECT().TestI64(gomock.Any(), int64(424242)).Return(int64(424242), nil), 110 // TODO: add TestBinary() 111 handler.EXPECT().TestDouble(gomock.Any(), float64(42.42)).Return(float64(42.42), nil), 112 handler.EXPECT().TestStruct(gomock.Any(), &thrifttest.Xtruct{StringThing: "thing", ByteThing: 42, I32Thing: 4242, I64Thing: 424242}).Return(&thrifttest.Xtruct{StringThing: "thing", ByteThing: 42, I32Thing: 4242, I64Thing: 424242}, nil), 113 handler.EXPECT().TestNest(gomock.Any(), &thrifttest.Xtruct2{StructThing: &thrifttest.Xtruct{StringThing: "thing", ByteThing: 42, I32Thing: 4242, I64Thing: 424242}}).Return(&thrifttest.Xtruct2{StructThing: &thrifttest.Xtruct{StringThing: "thing", ByteThing: 42, I32Thing: 4242, I64Thing: 424242}}, nil), 114 handler.EXPECT().TestMap(gomock.Any(), map[int32]int32{1: 2, 3: 4, 5: 42}).Return(map[int32]int32{1: 2, 3: 4, 5: 42}, nil), 115 handler.EXPECT().TestStringMap(gomock.Any(), map[string]string{"a": "2", "b": "blah", "some": "thing"}).Return(map[string]string{"a": "2", "b": "blah", "some": "thing"}, nil), 116 handler.EXPECT().TestSet(gomock.Any(), []int32{1, 2, 42}).Return([]int32{1, 2, 42}, nil), 117 handler.EXPECT().TestList(gomock.Any(), []int32{1, 2, 42}).Return([]int32{1, 2, 42}, nil), 118 handler.EXPECT().TestEnum(gomock.Any(), thrifttest.Numberz_TWO).Return(thrifttest.Numberz_TWO, nil), 119 handler.EXPECT().TestTypedef(gomock.Any(), thrifttest.UserId(42)).Return(thrifttest.UserId(42), nil), 120 handler.EXPECT().TestMapMap(gomock.Any(), int32(42)).Return(rmapmap, nil), 121 // TODO: not testing insanity 122 handler.EXPECT().TestMulti(gomock.Any(), int8(42), int32(4242), int64(424242), map[int16]string{1: "blah", 2: "thing"}, thrifttest.Numberz_EIGHT, thrifttest.UserId(24)).Return(xxs, nil), 123 handler.EXPECT().TestException(gomock.Any(), "some").Return(xcept), 124 handler.EXPECT().TestException(gomock.Any(), "TException").Return(errors.New("Just random exception")), 125 handler.EXPECT().TestMultiException(gomock.Any(), "Xception", "ignoreme").Return(nil, &thrifttest.Xception{ErrorCode: 1001, Message: "This is an Xception"}), 126 handler.EXPECT().TestMultiException(gomock.Any(), "Xception2", "ignoreme").Return(nil, &thrifttest.Xception2{ErrorCode: 2002, StructThing: &thrifttest.Xtruct{StringThing: "This is an Xception2"}}), 127 handler.EXPECT().TestOneway(gomock.Any(), int32(2)).Return(nil), 128 handler.EXPECT().TestVoid(gomock.Any()), 129 ) 130 var err error 131 if err = client.TestVoid(defaultCtx); err != nil { 132 t.Errorf("Unexpected error in TestVoid() call: %s", err) 133 } 134 135 thing, err := client.TestString(defaultCtx, "thing") 136 if err != nil { 137 t.Errorf("Unexpected error in TestString() call: %s", err) 138 } 139 if thing != "thing" { 140 t.Errorf("Unexpected TestString() result, expected 'thing' got '%s' ", thing) 141 } 142 143 bl, err := client.TestBool(defaultCtx, true) 144 if err != nil { 145 t.Errorf("Unexpected error in TestBool() call: %s", err) 146 } 147 if !bl { 148 t.Errorf("Unexpected TestBool() result expected true, got %v ", bl) 149 } 150 bl, err = client.TestBool(defaultCtx, false) 151 if err != nil { 152 t.Errorf("Unexpected error in TestBool() call: %s", err) 153 } 154 if bl { 155 t.Errorf("Unexpected TestBool() result expected false, got %v ", bl) 156 } 157 158 b, err := client.TestByte(defaultCtx, 42) 159 if err != nil { 160 t.Errorf("Unexpected error in TestByte() call: %s", err) 161 } 162 if b != 42 { 163 t.Errorf("Unexpected TestByte() result expected 42, got %d ", b) 164 } 165 166 i32, err := client.TestI32(defaultCtx, 4242) 167 if err != nil { 168 t.Errorf("Unexpected error in TestI32() call: %s", err) 169 } 170 if i32 != 4242 { 171 t.Errorf("Unexpected TestI32() result expected 4242, got %d ", i32) 172 } 173 174 i64, err := client.TestI64(defaultCtx, 424242) 175 if err != nil { 176 t.Errorf("Unexpected error in TestI64() call: %s", err) 177 } 178 if i64 != 424242 { 179 t.Errorf("Unexpected TestI64() result expected 424242, got %d ", i64) 180 } 181 182 d, err := client.TestDouble(defaultCtx, 42.42) 183 if err != nil { 184 t.Errorf("Unexpected error in TestDouble() call: %s", err) 185 } 186 if d != 42.42 { 187 t.Errorf("Unexpected TestDouble() result expected 42.42, got %f ", d) 188 } 189 190 // TODO: add TestBinary() call 191 192 xs := thrifttest.NewXtruct() 193 xs.StringThing = "thing" 194 xs.ByteThing = 42 195 xs.I32Thing = 4242 196 xs.I64Thing = 424242 197 xsret, err := client.TestStruct(defaultCtx, xs) 198 if err != nil { 199 t.Errorf("Unexpected error in TestStruct() call: %s", err) 200 } 201 if *xs != *xsret { 202 t.Errorf("Unexpected TestStruct() result expected %#v, got %#v ", xs, xsret) 203 } 204 205 x2 := thrifttest.NewXtruct2() 206 x2.StructThing = xs 207 x2ret, err := client.TestNest(defaultCtx, x2) 208 if err != nil { 209 t.Errorf("Unexpected error in TestNest() call: %s", err) 210 } 211 if !reflect.DeepEqual(x2, x2ret) { 212 t.Errorf("Unexpected TestNest() result expected %#v, got %#v ", x2, x2ret) 213 } 214 215 m := map[int32]int32{1: 2, 3: 4, 5: 42} 216 mret, err := client.TestMap(defaultCtx, m) 217 if err != nil { 218 t.Errorf("Unexpected error in TestMap() call: %s", err) 219 } 220 if !reflect.DeepEqual(m, mret) { 221 t.Errorf("Unexpected TestMap() result expected %#v, got %#v ", m, mret) 222 } 223 224 sm := map[string]string{"a": "2", "b": "blah", "some": "thing"} 225 smret, err := client.TestStringMap(defaultCtx, sm) 226 if err != nil { 227 t.Errorf("Unexpected error in TestStringMap() call: %s", err) 228 } 229 if !reflect.DeepEqual(sm, smret) { 230 t.Errorf("Unexpected TestStringMap() result expected %#v, got %#v ", sm, smret) 231 } 232 233 s := []int32{1, 2, 42} 234 sret, err := client.TestSet(defaultCtx, s) 235 if err != nil { 236 t.Errorf("Unexpected error in TestSet() call: %s", err) 237 } 238 // Sets can be in any order, but Go slices are ordered, so reflect.DeepEqual won't work. 239 stemp := map[int32]struct{}{} 240 for _, val := range s { 241 stemp[val] = struct{}{} 242 } 243 for _, val := range sret { 244 if _, ok := stemp[val]; !ok { 245 t.Fatalf("Unexpected TestSet() result expected %#v, got %#v ", s, sret) 246 } 247 } 248 249 l := []int32{1, 2, 42} 250 lret, err := client.TestList(defaultCtx, l) 251 if err != nil { 252 t.Errorf("Unexpected error in TestList() call: %s", err) 253 } 254 if !reflect.DeepEqual(l, lret) { 255 t.Errorf("Unexpected TestList() result expected %#v, got %#v ", l, lret) 256 } 257 258 eret, err := client.TestEnum(defaultCtx, thrifttest.Numberz_TWO) 259 if err != nil { 260 t.Errorf("Unexpected error in TestEnum() call: %s", err) 261 } 262 if eret != thrifttest.Numberz_TWO { 263 t.Errorf("Unexpected TestEnum() result expected %#v, got %#v ", thrifttest.Numberz_TWO, eret) 264 } 265 266 tret, err := client.TestTypedef(defaultCtx, thrifttest.UserId(42)) 267 if err != nil { 268 t.Errorf("Unexpected error in TestTypedef() call: %s", err) 269 } 270 if tret != thrifttest.UserId(42) { 271 t.Errorf("Unexpected TestTypedef() result expected %#v, got %#v ", thrifttest.UserId(42), tret) 272 } 273 274 mapmap, err := client.TestMapMap(defaultCtx, 42) 275 if err != nil { 276 t.Errorf("Unexpected error in TestMapmap() call: %s", err) 277 } 278 if !reflect.DeepEqual(mapmap, rmapmap) { 279 t.Errorf("Unexpected TestMapmap() result expected %#v, got %#v ", rmapmap, mapmap) 280 } 281 282 xxsret, err := client.TestMulti(defaultCtx, 42, 4242, 424242, map[int16]string{1: "blah", 2: "thing"}, thrifttest.Numberz_EIGHT, thrifttest.UserId(24)) 283 if err != nil { 284 t.Errorf("Unexpected error in TestMulti() call: %s", err) 285 } 286 if !reflect.DeepEqual(xxs, xxsret) { 287 t.Errorf("Unexpected TestMulti() result expected %#v, got %#v ", xxs, xxsret) 288 } 289 290 err = client.TestException(defaultCtx, "some") 291 if err == nil { 292 t.Errorf("Expecting exception in TestException() call") 293 } 294 if !reflect.DeepEqual(err, xcept) { 295 t.Errorf("Unexpected TestException() result expected %#v, got %#v ", xcept, err) 296 } 297 298 // TODO: connection is being closed on this 299 err = client.TestException(defaultCtx, "TException") 300 if err == nil { 301 t.Error("expected exception got nil") 302 } else if tex, ok := err.(thrift.TApplicationException); !ok { 303 t.Errorf("Unexpected TestException() result expected ApplicationError, got %T ", err) 304 } else if tex.TypeId() != thrift.INTERNAL_ERROR { 305 t.Errorf("expected internal_error got %v", tex.TypeId()) 306 } 307 308 ign, err := client.TestMultiException(defaultCtx, "Xception", "ignoreme") 309 if ign != nil || err == nil { 310 t.Errorf("Expecting exception in TestMultiException() call") 311 } 312 if !reflect.DeepEqual(err, &thrifttest.Xception{ErrorCode: 1001, Message: "This is an Xception"}) { 313 t.Errorf("Unexpected TestMultiException() %#v ", err) 314 } 315 316 ign, err = client.TestMultiException(defaultCtx, "Xception2", "ignoreme") 317 if ign != nil || err == nil { 318 t.Errorf("Expecting exception in TestMultiException() call") 319 } 320 expecting := &thrifttest.Xception2{ErrorCode: 2002, StructThing: &thrifttest.Xtruct{StringThing: "This is an Xception2"}} 321 322 if !reflect.DeepEqual(err, expecting) { 323 t.Errorf("Unexpected TestMultiException() %#v ", err) 324 } 325 326 err = client.TestOneway(defaultCtx, 2) 327 if err != nil { 328 t.Errorf("Unexpected error in TestOneway() call: %s", err) 329 } 330 331 //Make sure the connection still alive 332 if err = client.TestVoid(defaultCtx); err != nil { 333 t.Errorf("Unexpected error in TestVoid() call: %s", err) 334 } 335} 336