init push

unittest/sql/optimizer/ob_mock_partition_storage.h

@@ -0,0 +1,53 @@
+/**
+ * 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 _OB_MOCK_PARTITION_STORAGE_H_
+#define _OB_MOCK_PARTITION_STORAGE_H_
+
+#include "storage/ob_partition_storage.h"
+#include "storage/ob_range_iterator.h"
+
+namespace test {
+class MockPartitionStorage : public oceanbase::storage::ObPartitionStorage {
+  public:
+  MockPartitionStorage() : is_default_stat_(false)
+  {}
+  virtual ~MockPartitionStorage()
+  {}
+  virtual int get_batch_rows(const oceanbase::storage::ObTableScanParam& param,
+      const oceanbase::storage::ObBatch& batch, int64_t& rows, int64_t& rows_unreliable,
+      oceanbase::common::ObIArray<oceanbase::common::ObEstRowCountRecord>& est_records) override
+  {
+    UNUSED(param);
+    UNUSED(batch);
+    UNUSED(est_records);
+    if (is_default_stat_) {
+      rows = 0;
+    } else if (batch.type_ == oceanbase::storage::ObBatch::T_MULTI_SCAN) {
+      rows = 100 * batch.ranges_->count();
+    } else {
+      rows = 100;
+    }
+    rows_unreliable = rows;
+    return oceanbase::common::OB_SUCCESS;
+  }
+  void set_default_stat(const bool is_default_stat)
+  {
+    is_default_stat_ = is_default_stat;
+  }
+
+  private:
+  bool is_default_stat_;
+};
+}  // namespace test
+
+#endif