mirror of
https://github.com/unclechu/gRPC-haskell.git
synced 2024-11-26 21:19:43 +01:00
Expose GRPC_ARG_MAX_RECEIVE_MESSAGE_LENGTH
channel arg (#35)
* Expose the max receive message length channel argument + unittest * Remove unused `MultiWayIf` ext
This commit is contained in:
parent
192bea70e8
commit
3aa835a6f2
4 changed files with 49 additions and 4 deletions
|
@ -438,6 +438,8 @@ char* translate_arg_key(enum supported_arg_key key){
|
|||
return GRPC_ARG_PRIMARY_USER_AGENT_STRING;
|
||||
case user_agent_suffix_key:
|
||||
return GRPC_ARG_SECONDARY_USER_AGENT_STRING;
|
||||
case max_receive_message_length_key:
|
||||
return GRPC_ARG_MAX_RECEIVE_MESSAGE_LENGTH;
|
||||
default:
|
||||
return "unknown_arg_key";
|
||||
}
|
||||
|
|
|
@ -154,7 +154,8 @@ enum supported_arg_key {
|
|||
compression_algorithm_key = 0,
|
||||
compression_level_key,
|
||||
user_agent_prefix_key,
|
||||
user_agent_suffix_key
|
||||
user_agent_suffix_key,
|
||||
max_receive_message_length_key,
|
||||
};
|
||||
|
||||
grpc_arg* create_arg_array(size_t n);
|
||||
|
|
|
@ -6,6 +6,7 @@ import Control.Exception
|
|||
import Control.Monad
|
||||
import Foreign.Marshal.Alloc (malloc, free)
|
||||
import Foreign.Storable
|
||||
import Numeric.Natural
|
||||
|
||||
#include <grpc/grpc.h>
|
||||
#include <grpc/status.h>
|
||||
|
@ -50,6 +51,7 @@ data Arg = CompressionAlgArg CompressionAlgorithm
|
|||
| CompressionLevelArg CompressionLevel
|
||||
| UserAgentPrefix String
|
||||
| UserAgentSuffix String
|
||||
| MaxReceiveMessageLength Natural
|
||||
deriving (Show, Eq)
|
||||
|
||||
{#fun create_string_arg as ^ {`GrpcArg', `Int', `ArgKey', `String'} -> `()'#}
|
||||
|
@ -67,6 +69,9 @@ createArg array (UserAgentPrefix prefix) i =
|
|||
createStringArg array i UserAgentPrefixKey prefix
|
||||
createArg array (UserAgentSuffix suffix) i =
|
||||
createStringArg array i UserAgentSuffixKey suffix
|
||||
createArg array (MaxReceiveMessageLength n) i =
|
||||
createIntArg array i MaxReceiveMessageLengthKey $
|
||||
fromIntegral (min n (fromIntegral (maxBound :: Int)))
|
||||
|
||||
createChannelArgs :: [Arg] -> IO GrpcChannelArgs
|
||||
createChannelArgs args = do
|
||||
|
|
|
@ -57,6 +57,7 @@ lowLevelTests = testGroup "Unit tests of low-level Haskell library"
|
|||
, testCustomUserAgent
|
||||
, testClientCompression
|
||||
, testClientServerCompression
|
||||
, testClientMaxReceiveMessageLengthChannelArg
|
||||
, testClientStreaming
|
||||
, testClientStreamingUnregistered
|
||||
, testServerStreaming
|
||||
|
@ -766,6 +767,42 @@ testClientServerCompressionLvl =
|
|||
return ("hello", dummyMeta, StatusOk, StatusDetails "")
|
||||
return ()
|
||||
|
||||
testClientMaxReceiveMessageLengthChannelArg :: TestTree
|
||||
testClientMaxReceiveMessageLengthChannelArg = do
|
||||
testGroup "max receive message length channel arg (client channel)"
|
||||
[ csTest' "payload size < small bound succeeds" shouldSucceed server
|
||||
, csTest' "payload size > small bound fails" shouldFail server
|
||||
]
|
||||
where
|
||||
-- The server always sends a 4-byte payload
|
||||
pay = "four"
|
||||
server = TestServer (ServerConfig "localhost" 50051 ["/foo"] [] [] [] [] Nothing) $ \s -> do
|
||||
let rm = head (normalMethods s)
|
||||
void $ serverHandleNormalCall s rm mempty $ \sc -> do
|
||||
payload sc @?= pay
|
||||
pure (pay, mempty, StatusOk, StatusDetails "")
|
||||
|
||||
clientMax n k = TestClient conf $ \c -> do
|
||||
rm <- clientRegisterMethodNormal c "/foo"
|
||||
clientRequest c rm 1 pay mempty >>= k
|
||||
where
|
||||
conf = ClientConfig "localhost" 50051 [MaxReceiveMessageLength n] Nothing
|
||||
|
||||
-- Expect success when the max recv payload size is set to 4 bytes, and we
|
||||
-- are sent 4.
|
||||
shouldSucceed = clientMax 4 $ checkReqRslt $ \NormalRequestResult{..} -> do
|
||||
rspCode @?= StatusOk
|
||||
rspBody @?= pay
|
||||
details @?= ""
|
||||
|
||||
-- Expect failure when the max recv payload size is set to 3 bytes, and we
|
||||
-- are sent 4.
|
||||
shouldFail = clientMax 3 $ \case
|
||||
Left (GRPCIOBadStatusCode StatusInvalidArgument _)
|
||||
-> pure ()
|
||||
rsp
|
||||
-> clientFail ("Expected failure response, but got: " ++ show rsp)
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
-- Utilities and helpers
|
||||
|
||||
|
|
Loading…
Reference in a new issue