1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
//
// Copyright 2020 Staysail Systems, Inc. <info@staysail.tech>
//
// This software is supplied under the terms of the MIT License, a
// copy of which should be located in the distribution where this
// file was obtained (LICENSE.txt). A copy of the license may also be
// found online at https://opensource.org/licenses/MIT.
//
#ifndef TESTUTIL_H
#define TESTUTIL_H
#include <stdbool.h>
#include <stdint.h>
// The following headers are provided for test code convenience.
#include <nng/nng.h>
#ifdef __cplusplus
extern "C" {
#endif
// testutil_clock returns the current time in milliseconds.
// The reference clock may be any point in the past (typically since
// the program started running.)
extern uint64_t testutil_clock(void);
// testutil_pollfd tests if the given file descriptor polls as readable.
extern bool testutil_pollfd(int);
// testutil_htons is just htons portably.
extern uint16_t testutil_htons(uint16_t);
// testutil_htonl is just htonl portably.
extern uint32_t testutil_htonl(uint32_t);
// testutil_sleep sleeps the specified number of msec
extern void testutil_sleep(int);
// testutil_next_port returns a new port number (presumably unique)
extern uint16_t testutil_next_port(void);
// testutil_scratch_addr makes a scratch address for the given scheme.
// The address buffer must be supplied, and the size should be at least
// 64 bytes to ensure no truncation occurs.
extern void testutil_scratch_addr(const char *, size_t, char *);
// testutil_marry connects two sockets using inproc. It uses socket
// pipe hooks to ensure that it does not return before both sockets
// are fully connected.
extern int testutil_marry(nng_socket, nng_socket);
// testutil_marry_ex is like testutil_marry, but returns the pipes that
// were connected, and includes an optional URL. The pipe pointers and the
// URL may be NULL if not needed.
extern int testutil_marry_ex(
nng_socket, nng_socket, const char *, nng_pipe *, nng_pipe *);
// testutil_stream_send_start and testutil_stream_recv_start are used
// to initiate transfers asynchronously. They return a token which can
// be used with testutil_stream_send_wait and testutil_stream_recv_wait.
// Those wait functions will return the result of operation (0 on
// success, an NNG error number otherwise.)
extern void *testutil_stream_send_start(nng_stream *, void *, size_t);
extern void *testutil_stream_recv_start(nng_stream *, void *, size_t);
extern int testutil_stream_send_wait(void *);
extern int testutil_stream_recv_wait(void *);
// These are TLS certificates. The client and server are signed with the
// root. The server uses CN 127.0.0.1. Other details are bogus, but
// designed to prevent accidental use elsewhere.
extern const char *testutil_server_key;
extern const char *testutil_server_crt;
extern const char *testutil_client_key;
extern const char *testutil_client_crt;
// TEST_NNG_PASS tests for NNG success. It reports the failure if it
// did not.
#define TEST_NNG_PASS(cond) \
do { \
int result_ = (cond); \
TEST_CHECK_(result_ == 0, "%s succeeds", #cond); \
TEST_MSG("%s: expected success, got %s (%d)", #cond, \
nng_strerror(result_), result_); \
} while (0)
#define TEST_NNG_FAIL(cond, expect) \
do { \
int result_ = (cond); \
TEST_CHECK_(result_ == expect, "%s fails with %s", #cond, \
nng_strerror(expect)); \
TEST_MSG("%s: expected %s (%d), got %s (%d)", #cond, \
nng_strerror(expect), expect, nng_strerror(result_), \
result_); \
} while (0)
#define TEST_NNG_SEND_STR(sock, string) \
TEST_NNG_PASS(nng_send(sock, string, strlen(string) + 1, 0))
#define TEST_NNG_RECV_STR(sock, string) \
do { \
char buf_[64]; \
size_t sz_ = sizeof(buf_); \
int rv_ = nng_recv(sock, &buf_, &sz_, 0); \
TEST_CHECK_( \
rv_ == 0, "nng_recv (%d %s)", rv_, nng_strerror(rv_)); \
TEST_CHECK_(sz_ == strlen(string) + 1, "length %d want %d", \
sz_, strlen(string) + 1); \
buf_[sizeof(buf_) - 1] = '\0'; \
TEST_CHECK_( \
strcmp(string, buf_) == 0, "%s == %s", string, buf_); \
} while (0)
#ifdef __cplusplus
};
#endif
#endif // TESTUTIL_H
|
