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
|
/* \brief ����ڵ��������Ԫ���Դ���
* \author Qiuwen Lu<[email protected]>
* \date 2016-09-02
*/
#include <gtest/gtest.h>
#include <mr_ctx.h>
#include <mr_common.h>
#include <mr_vman.h>
extern struct mr_module module_desc_vnodeman;
class TestCaseVNodeMan : public ::testing::Test
{
protected:
TestCaseVNodeMan()
{
}
~TestCaseVNodeMan()
{
}
static void SetUpTestCase()
{
mr_global_cfg_init();
mr_global_ctx_init();
struct mr_global_ctx * g_ctx = mr_global_ctx_get();
struct mr_config * g_cfg = mr_global_config_get();
int ret = module_desc_vnodeman.preinit(
&module_desc_vnodeman, g_cfg, g_ctx);
ASSERT_EQ(ret, 0);
}
};
TEST_F(TestCaseVNodeMan, TestRegisterAndAttach)
{
int ret = 0;
ret = mr_vnodeman_create("TN", 1024, 128, 0);
ASSERT_EQ(ret, 0);
ret = mr_vnodeman_register_consumer("TN", "TNS",4);
ASSERT_EQ(ret, 0);
ret = mr_vnodeman_register_producer("TN", "TNP", 4);
ASSERT_EQ(ret, 0);
struct vnodeman_attach_desc * cons_desc[4];
struct vnodeman_attach_desc * prod_desc[4];
for (int i = 0; i < 4; i++)
{
cons_desc[i] = mr_vnodeman_consumer_attach("TN", "TNS");
ASSERT_TRUE(cons_desc[i] != NULL);
EXPECT_EQ(cons_desc[i]->desc_type, VNODEMAN_ATTACH_CONSUMER);
EXPECT_EQ(cons_desc[i]->queue_id, i);
}
}
|
