Fix rmr_call() parameter checking bug

[ric-plt/lib/rmr.git] / test / test_msg_support.c

diff --git a/test/test_msg_support.c b/test/test_msg_support.c
new file mode 100644 (file)
index 0000000..1e52c83
--- /dev/null
+++ b/test/test_msg_support.c
@@ -0,0 +1,94 @@
+// vi: ts=4 sw=4 noet :
+/*
+==================================================================================
+           Copyright (c) 2019-2020 Nokia
+           Copyright (c) 2018-2020 AT&T Intellectual Property.
+
+   Licensed 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.
+==================================================================================
+*/
+
+/*
+       Mnemonic:       test_msg_support.c
+       Abstract:       Support for testing with messages; requires RMR defs, so not
+                               to be included for things like ring tests etc.
+       Author:         E. Scott Daniels
+       Date:           14 April 2020   (AKD)
+*/
+
+#ifndef _test_msg_support_c
+#define _test_msg_support_c
+
+#include <signal.h>
+#include <string.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <fcntl.h>
+#include <unistd.h>
+
+int test_support_xact_count = 0;
+/*
+       Send a 'burst' of messages to drive some send retry failures to increase RMr coverage
+       by handling the retry caee.
+*/
+static void send_n_msgs( void* ctx, int n ) {
+       rmr_mbuf_t*     msg;                    // message buffers
+       int i;
+
+       msg = rmr_alloc_msg( ctx,  1024 );
+       if( ! msg ) {
+               fprintf( stderr, "<FAIL> mass send of %d messages couldn't allocate message!\n", n );
+               return;
+       }
+
+       fprintf( stderr, "<INFO> mass send of %d messages\n", n );
+       for( i = 0; i < n; i++ ) {
+               msg->len = 100;
+               msg->mtype = 1;
+               msg->state = 999;
+
+               snprintf( msg->xaction, 32, "%015d", test_support_xact_count++ );               // simple transaction id so we can test receive specific and ring stuff
+
+               errno = 999;
+               msg = rmr_send_msg( ctx, msg );
+               if( msg && msg->state != 0 ) {
+                       fprintf( stderr, "<WARN> mass send failed: state=%d type=%d\n", msg->state, msg->mtype );
+               }
+       }
+}
+
+
+/*
+       Allow test to reset the transaction id counter.
+*/
+static void reset_xact_count() {
+       test_support_xact_count = 0;
+}
+
+/*
+       Refresh or allocate a message with some default values
+*/
+static rmr_mbuf_t* fresh_msg( void* ctx, rmr_mbuf_t* msg ) {
+       if( ! msg )  {
+               msg = rmr_alloc_msg( ctx, 2048 );
+       }
+
+       msg->mtype = 0;
+       msg->sub_id = -1;
+       msg->state = 0;
+       msg->len = 100;
+
+       return msg;
+}
+
+#endif