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
20-module(test_client).
21
22-export([start/0, start/1]).
23
24-include("gen-erl/thrift_test_types.hrl").
25
26-record(options, {port = 9090,
27                  client_opts = []}).
28
29parse_args(Args) -> parse_args(Args, #options{}).
30parse_args([], Opts) ->
31  Opts;
32parse_args([Head | Rest], Opts) ->
33    NewOpts =
34        case Head of
35            "--port=" ++ Port ->
36                case string:to_integer(Port) of
37                  {IntPort,_} when is_integer(IntPort) ->
38                    Opts#options{port = IntPort};
39                  _Else ->
40                    erlang:error({bad_arg, Head})
41                end;
42            "--transport=" ++ Trans ->
43                % TODO: Enable Buffered and HTTP transport
44                case Trans of
45                    "framed" ->
46                        Opts#options{client_opts = [{framed, true} | Opts#options.client_opts]};
47                    _Else ->
48                        Opts
49                end;
50            "--ssl" ->
51                ssl:start(),
52                SslOptions =
53                    {ssloptions, [
54                        {cacertfile, "../keys/CA.pem"},
55                        {certfile, "../keys/client.pem"},
56                        {keyfile, "../keys/client.key"}
57                    ]},
58                Opts#options{client_opts = [{ssltransport, true} | [SslOptions | Opts#options.client_opts]]};
59            "--protocol=" ++ Proto ->
60                Opts#options{client_opts = [{protocol, list_to_atom(Proto)}]};
61            _Else ->
62                erlang:error({bad_arg, Head})
63        end,
64    parse_args(Rest, NewOpts).
65
66start() -> start(init:get_plain_arguments()).
67start(Args) ->
68  #options{port = Port, client_opts = ClientOpts} = parse_args(Args),
69  {ok, Client0} = thrift_client_util:new(
70    "127.0.0.1", Port, thrift_test_thrift, ClientOpts),
71
72  DemoXtruct = #'thrift.test.Xtruct'{
73                  string_thing = <<"Zero">>,
74                  byte_thing = 1,
75                  i32_thing = 9128361,
76                  i64_thing = 9223372036854775807},
77
78  DemoNest = #'thrift.test.Xtruct2'{
79    byte_thing = 7,
80    struct_thing = DemoXtruct,
81    % Note that we don't set i32_thing, it will come back as undefined
82    % from the Python server, but 0 from the C++ server, since it is not
83    % optional
84    i32_thing = 2},
85
86  % Is it safe to match these things?
87  DemoDict = dict:from_list([ {Key, Key-10} || Key <- lists:seq(0,10) ]),
88  DemoSet = sets:from_list([ Key || Key <- lists:seq(-3,3) ]),
89
90  DemoInsane = #'thrift.test.Insanity'{
91    userMap = dict:from_list([{?THRIFT_TEST_NUMBERZ_FIVE, 5000}]),
92    xtructs = [#'thrift.test.Xtruct'{ string_thing = <<"Truck">>, byte_thing = 8, i32_thing = 8, i64_thing = 8}]},
93
94  error_logger:info_msg("testVoid"),
95  {Client01, {ok, ok}} = thrift_client:call(Client0, testVoid, []),
96
97  error_logger:info_msg("testString"),
98  {Client02, {ok, <<"Test">>}}      = thrift_client:call(Client01, testString, ["Test"]),
99  error_logger:info_msg("testString"),
100  {Client03, {ok, <<"Test">>}}      = thrift_client:call(Client02, testString, [<<"Test">>]),
101  error_logger:info_msg("testByte"),
102  {Client04, {ok, 63}}              = thrift_client:call(Client03, testByte, [63]),
103  error_logger:info_msg("testI32"),
104  {Client05, {ok, -1}}              = thrift_client:call(Client04, testI32, [-1]),
105  error_logger:info_msg("testI32"),
106  {Client06, {ok, 0}}               = thrift_client:call(Client05, testI32, [0]),
107  error_logger:info_msg("testI64"),
108  {Client07, {ok, -34359738368}}    = thrift_client:call(Client06, testI64, [-34359738368]),
109  error_logger:info_msg("testDouble"),
110  {Client08, {ok, -5.2098523}}      = thrift_client:call(Client07, testDouble, [-5.2098523]),
111  %% TODO: add testBinary() call
112  error_logger:info_msg("testStruct"),
113  {Client09, {ok, DemoXtruct}}      = thrift_client:call(Client08, testStruct, [DemoXtruct]),
114  error_logger:info_msg("testNest"),
115  {Client10, {ok, DemoNest}}        = thrift_client:call(Client09, testNest, [DemoNest]),
116  error_logger:info_msg("testMap"),
117  {Client11, {ok, DemoDict}}        = thrift_client:call(Client10, testMap, [DemoDict]),
118  error_logger:info_msg("testSet"),
119  {Client12, {ok, DemoSet}}         = thrift_client:call(Client11, testSet, [DemoSet]),
120  error_logger:info_msg("testList"),
121  {Client13, {ok, [-1,2,3]}}        = thrift_client:call(Client12, testList, [[-1,2,3]]),
122  error_logger:info_msg("testEnum"),
123  {Client14, {ok, 1}}               = thrift_client:call(Client13, testEnum, [?THRIFT_TEST_NUMBERZ_ONE]),
124  error_logger:info_msg("testTypedef"),
125  {Client15, {ok, 309858235082523}} = thrift_client:call(Client14, testTypedef, [309858235082523]),
126  error_logger:info_msg("testInsanity"),
127  {Client16, {ok, InsaneResult}}    = thrift_client:call(Client15, testInsanity, [DemoInsane]),
128  io:format("~p~n", [InsaneResult]),
129
130  {Client17, {ok, #'thrift.test.Xtruct'{string_thing = <<"Message">>}}} =
131    thrift_client:call(Client16, testMultiException, ["Safe", "Message"]),
132
133  Client18 =
134    try
135      {ClientS1, Result1} = thrift_client:call(Client17, testMultiException, ["Xception", "Message"]),
136      io:format("Unexpected return! ~p~n", [Result1]),
137      ClientS1
138    catch
139      throw:{ClientS2, {exception, ExnS1 = #'thrift.test.Xception'{}}} ->
140        #'thrift.test.Xception'{errorCode = 1001, message = <<"This is an Xception">>} = ExnS1,
141        ClientS2;
142      throw:{ClientS2, {exception, _ExnS1 = #'thrift.test.Xception2'{}}} ->
143        io:format("Wrong exception type!~n", []),
144        ClientS2
145    end,
146
147  Client19 =
148    try
149      {ClientS3, Result2} = thrift_client:call(Client18, testMultiException, ["Xception2", "Message"]),
150      io:format("Unexpected return! ~p~n", [Result2]),
151      ClientS3
152    catch
153      throw:{ClientS4, {exception, _ExnS2 = #'thrift.test.Xception'{}}} ->
154        io:format("Wrong exception type!~n", []),
155        ClientS4;
156      throw:{ClientS4, {exception, ExnS2 = #'thrift.test.Xception2'{}}} ->
157        #'thrift.test.Xception2'{errorCode = 2002,
158                   struct_thing = #'thrift.test.Xtruct'{
159                     string_thing = <<"This is an Xception2">>}} = ExnS2,
160        ClientS4
161    end,
162
163  %% Started = erlang:monotonic_time(milli_seconds),
164  {_, StartSec, StartUSec} = erlang:timestamp(),
165  error_logger:info_msg("testOneway"),
166  {Client20, {ok, ok}} = thrift_client:call(Client19, testOneway, [1]),
167  {_, EndSec, EndUSec} = erlang:timestamp(),
168  Elapsed = (EndSec - StartSec) * 1000 + (EndUSec - StartUSec) / 1000,
169  if
170    Elapsed > 1000 -> exit(1);
171    true -> true
172  end,
173
174  thrift_client:close(Client20).
175