oceanbase/unittest/storage/mock_ob_ss_store.h
wangzelin.wzl 93a1074b0c patch 4.0
2022-10-24 17:57:12 +08:00

99 lines
2.7 KiB
C++

/**
* Copyright (c) 2021 OceanBase
* OceanBase CE is licensed under Mulan PubL v2.
* You can use this software according to the terms and conditions of the Mulan PubL v2.
* You may obtain a copy of Mulan PubL v2 at:
* http://license.coscl.org.cn/MulanPubL-2.0
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PubL v2 for more details.
*/
#ifndef MOCK_OB_SS_STORE_H_
#define MOCK_OB_SS_STORE_H_
#include <gmock/gmock.h>
#include "storage/ob_ss_store.h"
namespace oceanbase
{
namespace storage
{
class MockObSSStore : public ObSSStore
{
public:
MockObSSStore() {}
virtual ~MockObSSStore() {}
MOCK_METHOD4(
get,
int(
const storage::ObTableAccessParam &param,
storage::ObTableAccessContext &context,
const common::ObStoreRowkey &rowkey,
ObStoreRowIterator *&row_iter));
MOCK_METHOD4(
scan,
int(
const ObTableAccessParam &param,
ObTableAccessContext &context,
const common::ObStoreRange &key_range,
ObStoreRowIterator *&row_iter));
MOCK_METHOD4(
multi_get,
int(
const ObTableAccessParam &param,
ObTableAccessContext &context,
const common::ObIArray<common::ObStoreRowkey> &rowkeys,
ObStoreRowIterator *&row_iter));
MOCK_METHOD4(
multi_scan,
int(
const ObTableAccessParam &param,
ObTableAccessContext &context,
const common::ObIArray<common::ObStoreRange> &ranges,
ObStoreRowIterator *&row_iter));
MOCK_METHOD5(
estimate_get_cost,
int (const common::ObQueryFlag query_flag,
const uint64_t table_id,
const common::ObIArray<common::ObStoreRowkey> &rowkeys,
const common::ObIArray<share::schema::ObColDesc> &columns,
ObPartitionEst &cost_metrics));
MOCK_METHOD5(
estimate_scan_cost,
int (const common::ObQueryFlag query_flag,
const uint64_t table_id,
const common::ObStoreRange &key_range,
const common::ObIArray<share::schema::ObColDesc> &columns,
ObPartitionEst &cost_metrics));
MOCK_METHOD5(
estimate_multi_scan_cost,
int (
const common::ObQueryFlag query_flag,
const uint64_t table_id,
const common::ObIArray<common::ObStoreRange> &ranges,
const common::ObIArray<share::schema::ObColDesc> &columns,
ObPartitionEst &cost_metrics));
//virtual enum storage::ObStoreType get_store_type() const { return MAJOR_SSTORE; }
MOCK_CONST_METHOD0(
get_store_type,
enum storage::ObStoreType());
};
}
}
#endif /* MOCK_OB_SS_STORE_H_ */