Skip to content

Commit 8d93699

Browse files
author
yangce
committed
issue=baidu#1271 Add mock zk adapter for test
1 parent 4e62477 commit 8d93699

File tree

3 files changed

+11
-9
lines changed

3 files changed

+11
-9
lines changed

src/master/master_impl.cc

+3-2
Original file line numberDiff line numberDiff line change
@@ -168,11 +168,12 @@ MasterImpl::~MasterImpl() {
168168
bool MasterImpl::Init() {
169169
if (FLAGS_tera_zk_enabled) {
170170
zk_adapter_.reset(new MasterZkAdapter(this, local_addr_));
171-
} else if (FLAGS_tera_mock_zk_enabled) {
172-
zk_adapter_.reset(new MockMasterZkAdapter(this, local_addr_));
173171
} else if (FLAGS_tera_ins_enabled) {
174172
LOG(INFO) << "ins mode" ;
175173
zk_adapter_.reset(new InsMasterZkAdapter(this, local_addr_));
174+
} else if (FLAGS_tera_mock_zk_enabled) {
175+
LOG(INFO) << "mock zk mode" ;
176+
zk_adapter_.reset(new MockMasterZkAdapter(this, local_addr_));
176177
} else if (FLAGS_tera_mock_ins_enabled) {
177178
LOG(INFO) << "mock ins mode" ;
178179
zk_adapter_.reset(new MockInsMasterZkAdapter(this, local_addr_));

src/sdk/sdk_zk.cc

+5-5
Original file line numberDiff line numberDiff line change
@@ -123,14 +123,14 @@ bool FakeZkClusterFinder::ReadNode(const std::string& name, std::string* value)
123123
}
124124

125125
ClusterFinder* NewClusterFinder() {
126-
if (FLAGS_tera_ins_enabled) {
126+
if (FLAGS_tera_zk_enabled) {
127+
return new sdk::FakeZkClusterFinder(FLAGS_tera_fake_zk_path_prefix);
128+
} else if (FLAGS_tera_ins_enabled) {
127129
return new sdk::InsClusterFinder(FLAGS_tera_ins_root_path, FLAGS_tera_ins_addr_list);
128-
} else if (FLAGS_tera_mock_ins_enabled) {
129-
return new sdk::MockInsClusterFinder(FLAGS_tera_ins_root_path, FLAGS_tera_ins_addr_list);
130130
} else if (FLAGS_tera_mock_zk_enabled) {
131131
return new sdk::MockZkClusterFinder(FLAGS_tera_zk_root_path, FLAGS_tera_zk_addr_list);
132-
} else if (!FLAGS_tera_zk_enabled) {
133-
return new sdk::FakeZkClusterFinder(FLAGS_tera_fake_zk_path_prefix);
132+
} else if (FLAGS_tera_mock_ins_enabled) {
133+
return new sdk::MockInsClusterFinder(FLAGS_tera_ins_root_path, FLAGS_tera_ins_addr_list);
134134
} else {
135135
return new sdk::ZkClusterFinder(FLAGS_tera_zk_root_path, FLAGS_tera_zk_addr_list);
136136
}

src/tabletnode/tabletnode_impl.cc

+3-2
Original file line numberDiff line numberDiff line change
@@ -159,11 +159,12 @@ TabletNodeImpl::~TabletNodeImpl() {
159159
bool TabletNodeImpl::Init() {
160160
if (FLAGS_tera_zk_enabled) {
161161
zk_adapter_.reset(new TabletNodeZkAdapter(this, local_addr_));
162-
} else if (FLAGS_tera_mock_zk_enabled) {
163-
zk_adapter_.reset(new MockTabletNodeZkAdapter(this, local_addr_));
164162
} else if(FLAGS_tera_ins_enabled) {
165163
LOG(INFO) << "ins mode!";
166164
zk_adapter_.reset(new InsTabletNodeZkAdapter(this, local_addr_));
165+
} else if (FLAGS_tera_mock_zk_enabled) {
166+
LOG(INFO) << "mock zk mode!";
167+
zk_adapter_.reset(new MockTabletNodeZkAdapter(this, local_addr_));
167168
} else if (FLAGS_tera_mock_ins_enabled) {
168169
LOG(INFO) << "mock ins mode!";
169170
zk_adapter_.reset(new MockInsTabletNodeZkAdapter(this, local_addr_));

0 commit comments

Comments
 (0)