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
|
// SPDX-License-Identifier: BSD-3-Clause
/* Copyright (c) 2022, Fujitsu Limited */
/*
* utils-ibv_context_is_atomic_write_capable.c -- a unit test
* for rpma_utils_ibv_context_is_atomic_write_capable()
*/
#include "cmocka_headers.h"
#include "mocks-ibverbs.h"
#include "librpma.h"
#include "test-common.h"
#include "utils.h"
/*
* ibvc_atomic_write__cap_no -- atomic write attribute is not set
* in attr.device_cap_flags_ex
*/
static void
ibvc_atomic_write__cap_no(void **unused)
{
#ifdef NATIVE_ATOMIC_WRITE_SUPPORTED
/* configure mocks */
struct ibv_device_attr_ex attr = {
.device_cap_flags_ex = 0, /* atomic write attribute is not set */
};
will_return(ibv_query_device_ex_mock, &attr);
#endif
/* run test */
int is_atomic_write_capable;
int ret = rpma_utils_ibv_context_is_atomic_write_capable(MOCK_VERBS,
&is_atomic_write_capable);
/* verify the results */
assert_int_equal(ret, 0);
assert_int_equal(is_atomic_write_capable, 0);
}
#ifdef NATIVE_ATOMIC_WRITE_SUPPORTED
/*
* ibvc_atomic_write__query_fail -- ibv_query_device_ex() failed
*/
static void
ibvc_atomic_write__query_fail(void **unused)
{
/* configure mocks */
will_return(ibv_query_device_ex_mock, NULL);
will_return(ibv_query_device_ex_mock, MOCK_ERRNO);
/* run test */
int is_atomic_write_capable;
int ret = rpma_utils_ibv_context_is_atomic_write_capable(MOCK_VERBS,
&is_atomic_write_capable);
/* verify the results */
assert_int_equal(ret, RPMA_E_PROVIDER);
}
/*
* ibvc_atomic_write__cap_yes -- atomic write attribute is set
* in attr.device_cap_flags_ex
*/
static void
ibvc_atomic_write__cap_yes(void **unused)
{
/* configure mocks */
struct ibv_device_attr_ex attr = {
/* atomic write attribute is set */
.device_cap_flags_ex = IB_UVERBS_DEVICE_ATOMIC_WRITE,
};
will_return(ibv_query_device_ex_mock, &attr);
/* run test */
int is_atomic_write_capable;
int ret = rpma_utils_ibv_context_is_atomic_write_capable(MOCK_VERBS,
&is_atomic_write_capable);
/* verify the results */
assert_int_equal(ret, 0);
assert_int_equal(is_atomic_write_capable, 1);
}
#endif
int
main(int argc, char *argv[])
{
MOCK_VERBS->abi_compat = __VERBS_ABI_IS_EXTENDED;
#ifdef NATIVE_ATOMIC_WRITE_SUPPORTED
Verbs_context.query_device_ex = ibv_query_device_ex_mock;
#endif
Verbs_context.sz = sizeof(struct verbs_context);
const struct CMUnitTest tests[] = {
/* rpma_utils_ibv_context_is_atomic_write_capable() unit tests */
cmocka_unit_test(ibvc_atomic_write__cap_no),
#ifdef NATIVE_ATOMIC_WRITE_SUPPORTED
cmocka_unit_test(ibvc_atomic_write__query_fail),
cmocka_unit_test(ibvc_atomic_write__cap_yes),
#endif
};
return cmocka_run_group_tests(tests, NULL, NULL);
}
|
