THRIFT-932. hs: Haskell tests need to be run through 'make check' (and probably 'cabal check') too
Tests are now self-contained and correctly exit after running. There's a single run script which has improved error messages and defaults to the thrift binary compiled in the current source directory instead of those in PATH. And as a bonus hooks both cabal check and running the tests to make check.
Patch: Christian Lavoie
git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@1004861 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/test/hs/DebugProtoTest_Main.hs b/test/hs/DebugProtoTest_Main.hs
new file mode 100644
index 0000000..816eee3
--- /dev/null
+++ b/test/hs/DebugProtoTest_Main.hs
@@ -0,0 +1,164 @@
+--
+-- 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.
+--
+
+module Main where
+
+
+import qualified Control.Exception
+import qualified Data.ByteString.Lazy as DBL
+import qualified Maybe
+import qualified Network
+
+import Thrift.Protocol.Binary
+import Thrift.Server
+import Thrift.Transport.Handle
+
+import qualified ThriftTestUtils
+
+import qualified DebugProtoTest_Types as Types
+import qualified Inherited
+import qualified Inherited_Client as IClient
+import qualified Inherited_Iface as IIface
+import qualified Srv_Client as SClient
+import qualified Srv_Iface as SIface
+
+-- we don't actually need this import, but force it to check the code generator exports proper Haskell syntax
+import qualified Srv()
+
+
+data InheritedHandler = InheritedHandler
+instance SIface.Srv_Iface InheritedHandler where
+ janky _ arg = do
+ ThriftTestUtils.serverLog $ "Got janky method call: " ++ show arg
+ return $ 31
+
+ voidMethod _ = do
+ ThriftTestUtils.serverLog "Got voidMethod method call"
+ return ()
+
+ primitiveMethod _ = do
+ ThriftTestUtils.serverLog "Got primitiveMethod call"
+ return $ 42
+
+ structMethod _ = do
+ ThriftTestUtils.serverLog "Got structMethod call"
+ return $ Types.CompactProtoTestStruct {
+ Types.f_CompactProtoTestStruct_a_byte = Just 0x01,
+ Types.f_CompactProtoTestStruct_a_i16 = Just 0x02,
+ Types.f_CompactProtoTestStruct_a_i32 = Just 0x03,
+ Types.f_CompactProtoTestStruct_a_i64 = Just 0x04,
+ Types.f_CompactProtoTestStruct_a_double = Just 0.1,
+ Types.f_CompactProtoTestStruct_a_string = Just "abcdef",
+ Types.f_CompactProtoTestStruct_a_binary = Just DBL.empty,
+ Types.f_CompactProtoTestStruct_true_field = Just True,
+ Types.f_CompactProtoTestStruct_false_field = Just False,
+ Types.f_CompactProtoTestStruct_empty_struct_field = Just Types.Empty,
+
+ Types.f_CompactProtoTestStruct_byte_list = Nothing,
+ Types.f_CompactProtoTestStruct_i16_list = Nothing,
+ Types.f_CompactProtoTestStruct_i32_list = Nothing,
+ Types.f_CompactProtoTestStruct_i64_list = Nothing,
+ Types.f_CompactProtoTestStruct_double_list = Nothing,
+ Types.f_CompactProtoTestStruct_string_list = Nothing,
+ Types.f_CompactProtoTestStruct_binary_list = Nothing,
+ Types.f_CompactProtoTestStruct_boolean_list = Nothing,
+ Types.f_CompactProtoTestStruct_struct_list = Just [Types.Empty],
+
+ Types.f_CompactProtoTestStruct_byte_set = Nothing,
+ Types.f_CompactProtoTestStruct_i16_set = Nothing,
+ Types.f_CompactProtoTestStruct_i32_set = Nothing,
+ Types.f_CompactProtoTestStruct_i64_set = Nothing,
+ Types.f_CompactProtoTestStruct_double_set = Nothing,
+ Types.f_CompactProtoTestStruct_string_set = Nothing,
+ Types.f_CompactProtoTestStruct_binary_set = Nothing,
+ Types.f_CompactProtoTestStruct_boolean_set = Nothing,
+ Types.f_CompactProtoTestStruct_struct_set = Nothing,
+
+ Types.f_CompactProtoTestStruct_byte_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_i16_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_i32_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_i64_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_double_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_string_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_binary_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_boolean_byte_map = Nothing,
+
+ Types.f_CompactProtoTestStruct_byte_i16_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_i32_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_i64_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_double_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_string_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_binary_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_boolean_map = Nothing,
+
+ Types.f_CompactProtoTestStruct_list_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_set_byte_map = Nothing,
+ Types.f_CompactProtoTestStruct_map_byte_map = Nothing,
+
+ Types.f_CompactProtoTestStruct_byte_map_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_set_map = Nothing,
+ Types.f_CompactProtoTestStruct_byte_list_map = Nothing }
+
+ methodWithDefaultArgs _ arg = do
+ ThriftTestUtils.serverLog $ "Got methodWithDefaultArgs: " ++ show arg
+ return ()
+
+ onewayMethod _ = do
+ ThriftTestUtils.serverLog "Got onewayMethod"
+
+instance IIface.Inherited_Iface InheritedHandler where
+ identity _ arg = do
+ ThriftTestUtils.serverLog $ "Got identity method: " ++ show arg
+ return $ Maybe.fromJust arg
+
+client :: (String, Network.PortID) -> IO ()
+client addr = do
+ to <- hOpen addr
+ let p = BinaryProtocol to
+ let ps = (p,p)
+
+ v1 <- SClient.janky ps 42
+ ThriftTestUtils.clientLog $ show v1
+
+ SClient.voidMethod ps
+
+ v2 <- SClient.primitiveMethod ps
+ ThriftTestUtils.clientLog $ show v2
+
+ v3 <- SClient.structMethod ps
+ ThriftTestUtils.clientLog $ show v3
+
+ SClient.methodWithDefaultArgs ps 42
+
+ SClient.onewayMethod ps
+
+ v4 <- IClient.identity ps 42
+ ThriftTestUtils.clientLog $ show v4
+
+ return ()
+
+server :: Network.PortNumber -> IO ()
+server port = do
+ ThriftTestUtils.serverLog "Ready..."
+ (runBasicServer InheritedHandler Inherited.process port)
+ `Control.Exception.catch`
+ (\(TransportExn s _) -> error $ "FAILURE: " ++ show s)
+
+main :: IO ()
+main = ThriftTestUtils.runTest server client