summaryrefslogtreecommitdiff
path: root/service/test/test_node_mocks.c
blob: 564cd64baa9b49705767dd0c6b7472cb450f00dd (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
#include <stdint.h>
#include <rte_graph.h>
#include <rte_graph_worker.h>
#include <stdarg.h>
#include <stddef.h>
#include <stdint.h>
#include <setjmp.h>
#include <cmocka.h>

uint16_t mock_pkt_input_process(struct rte_graph * graph, struct rte_node * node, void ** objs, uint16_t cnt)
{
    unsigned int nr_source_pkts = mock_type(unsigned int);
    struct rte_mbuf ** source_mbufs = mock_ptr_type(struct rte_mbuf **);

    rte_node_enqueue(graph, node, 0, (void **)source_mbufs, nr_source_pkts);
    return nr_source_pkts;
}

uint16_t mock_pkt_output_process(struct rte_graph * graph, struct rte_node * node, void ** objs, uint16_t cnt)
{
    struct rte_mbuf ** output_mbufs = mock_ptr_type(struct rte_mbuf **);
    unsigned int * nr_output_pkts = mock_ptr_type(unsigned int *);

    for (int i = 0; i < cnt; i++)
    {
        output_mbufs[i] = objs[i];
    }

    *nr_output_pkts = (unsigned int)cnt;
    return cnt;
}

uint16_t mock_pkt_drop_process(struct rte_graph * graph, struct rte_node * node, void ** objs, uint16_t cnt)
{
    struct rte_mbuf ** output_mbufs = mock_ptr_type(struct rte_mbuf **);
    unsigned int * nr_output_pkts = mock_ptr_type(unsigned int *);

    for (int i = 0; i < cnt; i++)
    {
        output_mbufs[i] = objs[i];
    }

    *nr_output_pkts = (unsigned int)cnt;
    return cnt;
}

struct rte_node_register mock_pkt_output_node = {
    .process = mock_pkt_output_process,
    .name = "mock_pkt_output",
};

struct rte_node_register mock_pkt_drop_node = {
    .process = mock_pkt_drop_process,
    .name = "mock_pkt_drop",
};

struct rte_node_register mock_pkt_input_node = {
    .process = mock_pkt_input_process,
    .name = "mock_pkt_input",
    .nb_edges = 1,
    .next_nodes =
        {
            "vwire_ingress",
        },
    .flags = RTE_NODE_SOURCE_F,
};

RTE_NODE_REGISTER(mock_pkt_input_node);
RTE_NODE_REGISTER(mock_pkt_output_node);
RTE_NODE_REGISTER(mock_pkt_drop_node);