aboutsummaryrefslogtreecommitdiff
path: root/src/test/test_pubsub.c
blob: 2f047d9f2c4c35f9f05841ed598c797ad93c52b9 (plain)
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
/* Copyright (c) 2016-2017, The Tor Project, Inc. */
/* See LICENSE for licensing information */

/**
 * \file test_pubsub.c
 * \brief Unit tests for publish-subscribe abstraction.
 **/

#include "or.h"
#include "test.h"
#include "pubsub.h"

DECLARE_PUBSUB_STRUCT_TYPES(foobar)
DECLARE_PUBSUB_TOPIC(foobar)
DECLARE_NOTIFY_PUBSUB_TOPIC(static, foobar)
IMPLEMENT_PUBSUB_TOPIC(static, foobar)

struct foobar_event_data_t {
  unsigned u;
  const char *s;
};

struct foobar_subscriber_data_t {
  const char *name;
  long l;
};

static int
foobar_sub1(foobar_event_data_t *ev, foobar_subscriber_data_t *mine)
{
  ev->u += 10;
  mine->l += 100;
  return 0;
}

static int
foobar_sub2(foobar_event_data_t *ev, foobar_subscriber_data_t *mine)
{
  ev->u += 5;
  mine->l += 50;
  return 0;
}

static void
test_pubsub_basic(void *arg)
{
  (void)arg;
  foobar_subscriber_data_t subdata1 = { "hi", 0 };
  foobar_subscriber_data_t subdata2 = { "wow", 0 };
  const foobar_subscriber_t *sub1;
  const foobar_subscriber_t *sub2;
  foobar_event_data_t ed = { 0, "x" };
  foobar_event_data_t ed2 = { 0, "y" };
  sub1 = foobar_subscribe(foobar_sub1, &subdata1, SUBSCRIBE_ATSTART, 100);
  tt_assert(sub1);

  foobar_notify(&ed, 0);
  tt_int_op(subdata1.l, OP_EQ, 100);
  tt_int_op(subdata2.l, OP_EQ, 0);
  tt_int_op(ed.u, OP_EQ, 10);

  sub2 = foobar_subscribe(foobar_sub2, &subdata2, 0, 5);
  tt_assert(sub2);

  foobar_notify(&ed2, 0);
  tt_int_op(subdata1.l, OP_EQ, 200);
  tt_int_op(subdata2.l, OP_EQ, 50);
  tt_int_op(ed2.u, OP_EQ, 15);

  foobar_unsubscribe(sub1);

  foobar_notify(&ed, 0);
  tt_int_op(subdata1.l, OP_EQ, 200);
  tt_int_op(subdata2.l, OP_EQ, 100);
  tt_int_op(ed.u, OP_EQ, 15);

 done:
  foobar_clear();
}

struct testcase_t pubsub_tests[] = {
  { "pubsub_basic", test_pubsub_basic, TT_FORK, NULL, NULL },
  END_OF_TESTCASES
};