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
119
120
121
122
|
//
// Copyright 2016 Garrett D'Amore <garrett@damore.org>
//
// 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.
//
#include "convey.h"
#include "nng.h"
#include <string.h>
Main({
int rv;
const char *addr = "inproc://test";
Test("REQ/REP pattern", {
Convey("We can create a REQ socket", {
nng_socket *req;
rv = nng_open(&req, NNG_PROTO_REQ);
So(rv == 0);
So(req != NULL);
Reset({
nng_close(req);
})
Convey("Protocols match", {
So(nng_protocol(req) == NNG_PROTO_REQ);
So(nng_peer(req) == NNG_PROTO_REP);
})
Convey("Recv with no send fails", {
nng_msg *msg;
rv = nng_recvmsg(req, &msg, 0);
So(rv == NNG_ESTATE);
})
})
Convey("We can create a REP socket", {
nng_socket *rep;
rv = nng_open(&rep, NNG_PROTO_REP);
So(rv == 0);
So(rep != NULL);
Reset({
nng_close(rep);
})
Convey("Protocols match", {
So(nng_protocol(rep) == NNG_PROTO_REP);
So(nng_peer(rep) == NNG_PROTO_REQ);
})
Convey("Send with no recv fails", {
nng_msg *msg;
rv = nng_msg_alloc(&msg, 0);
So(rv == 0);
rv = nng_sendmsg(rep, msg, 0);
So(rv == NNG_ESTATE);
nng_msg_free(msg);
})
})
Convey("We can create a linked REQ/REP pair", {
nng_socket *req;
nng_socket *rep;
rv = nng_open(&rep, NNG_PROTO_REP);
So(rv == 0);
So(rep != NULL);
rv = nng_open(&req, NNG_PROTO_REQ);
So(rv == 0);
So(req != NULL);
Reset({
nng_close(rep);
nng_close(req);
})
rv = nng_listen(rep, addr, NULL, NNG_FLAG_SYNCH);
So(rv == 0);
rv = nng_dial(req, addr, NULL, NNG_FLAG_SYNCH);
So(rv == 0);
Convey("They can REQ/REP exchange", {
nng_msg *ping;
nng_msg *pong;
char *body;
size_t len;
So(nng_msg_alloc(&ping, 0) == 0);
So(nng_msg_append(ping, "ping", 5) == 0);
body = nng_msg_body(ping, &len);
So(len == 5);
So(memcmp(body, "ping", 5) == 0);
So(nng_sendmsg(req, ping, 0) == 0);
pong = NULL;
So(nng_recvmsg(rep, &pong, 0) == 0);
So(pong != NULL);
body = nng_msg_body(pong, &len);
So(len == 5);
So(memcmp(body, "ping", 5) == 0);
nng_msg_trim(pong, 5);
So(nng_msg_append(pong, "pong", 5) == 0);
So(nng_sendmsg(rep, pong, 0) == 0);
ping = 0;
So(nng_recvmsg(req, &ping, 0) == 0);
So(ping != NULL);
body = nng_msg_body(ping, &len);
So(len == 5);
So(memcmp(body, "pong", 5) == 0);
nng_msg_free(ping);
})
})
})
})
|
