test: fix test case relying on hard buffer limit

pull/24376/head
Marco Neumann 2021-08-17 16:53:40 +02:00
parent 60ea637c38
commit ce35119633
2 changed files with 22 additions and 2 deletions

View File

@ -327,6 +327,16 @@ impl DatabaseBuilder {
} }
} }
pub fn buffer_size_hard(mut self, buffer_size_hard: u64) -> Self {
self.lifecycle_rules.buffer_size_hard = buffer_size_hard;
self
}
pub fn buffer_size_soft(mut self, buffer_size_soft: u64) -> Self {
self.lifecycle_rules.buffer_size_soft = buffer_size_soft;
self
}
pub fn persist(mut self, persist: bool) -> Self { pub fn persist(mut self, persist: bool) -> Self {
self.lifecycle_rules.persist = persist; self.lifecycle_rules.persist = persist;
self self

View File

@ -1,7 +1,7 @@
use influxdb_iox_client::write::WriteError; use influxdb_iox_client::write::WriteError;
use test_helpers::assert_contains; use test_helpers::assert_contains;
use crate::common::server_fixture::ServerFixture; use crate::{common::server_fixture::ServerFixture, end_to_end_cases::scenario::DatabaseBuilder};
use super::scenario::{create_readable_database, rand_name}; use super::scenario::{create_readable_database, rand_name};
use arrow_util::assert_batches_sorted_eq; use arrow_util::assert_batches_sorted_eq;
@ -22,8 +22,18 @@ async fn test_write() {
let fixture = ServerFixture::create_shared().await; let fixture = ServerFixture::create_shared().await;
let mut write_client = fixture.write_client(); let mut write_client = fixture.write_client();
// need a database that is easy to get into the hard buffer limit:
// 1. turn persist on so we cannot drop unpersisted data
// 2. set mutable buffer threshold high so IOx will keep the MUB growing
// 3. use small buffer limits to speed up the test
let db_name = rand_name(); let db_name = rand_name();
create_readable_database(&db_name, fixture.grpc_channel()).await; DatabaseBuilder::new(db_name.clone())
.persist(true)
.mub_row_threshold(1_000_000)
.buffer_size_soft(100_000)
.buffer_size_hard(200_000)
.build(fixture.grpc_channel())
.await;
// ---- test successful writes ---- // ---- test successful writes ----
let lp_lines = vec![ let lp_lines = vec![