test: add end to end test

influxdb_iox/tests/end_to_end_cases/management_cli.rs

@@ -7,13 +7,13 @@ use crate::{
     },
 };
 use assert_cmd::Command;
-use data_types::chunk_metadata::ChunkStorage;
+use data_types::chunk_metadata::{ChunkStorage, ChunkSummary};
 use generated_types::{
     google::longrunning::IoxOperation,
     influxdata::iox::management::v1::{operation_metadata::Job, WipePreservedCatalog},
 };
 use predicates::prelude::*;
-use std::time::Duration;
+use std::{sync::Arc, time::Duration};
 use tempfile::TempDir;
 use test_helpers::make_temp_file;
 use uuid::Uuid;
@@ -1158,9 +1158,8 @@ fn load_lp(addr: &str, db_name: &str, lp_data: Vec<&str>) {
         .stdout(predicate::str::contains("Lines OK"));
 }
 
-#[tokio::test]
+async fn setup_load_unload_partition_chunk() -> (Arc<ServerFixture>, String, String) {
-async fn test_unload_partition_chunk() {
+    let fixture = Arc::from(ServerFixture::create_shared(ServerType::Database).await);
-    let fixture = ServerFixture::create_shared(ServerType::Database).await;
     let addr = fixture.grpc_base();
     let db_name = rand_name();
 
@@ -1174,6 +1173,70 @@ async fn test_unload_partition_chunk() {
     let lp_data = vec!["cpu,region=west user=23.2 10"];
     load_lp(addr, &db_name, lp_data);
 
+    (Arc::clone(&fixture), db_name, String::from(addr))
+}
+
+#[tokio::test]
+async fn test_load_partition_chunk() {
+    let (fixture, db_name, addr) = setup_load_unload_partition_chunk().await;
+    let mut chunks = wait_for_exact_chunk_states(
+        &fixture,
+        &db_name,
+        vec![ChunkStorage::ReadBufferAndObjectStore],
+        std::time::Duration::from_secs(5),
+    )
+    .await;
+    let chunk = chunks.pop().unwrap();
+
+    Command::cargo_bin("influxdb_iox")
+        .unwrap()
+        .arg("database")
+        .arg("partition")
+        .arg("unload-chunk")
+        .arg(&db_name)
+        .arg("cpu")
+        .arg("cpu")
+        .arg(chunk.id.get().to_string())
+        .arg("--host")
+        .arg(&addr)
+        .assert()
+        .success()
+        .stdout(predicate::str::contains("Ok"));
+
+    let mut chunks = wait_for_exact_chunk_states(
+        &fixture,
+        &db_name,
+        vec![ChunkStorage::ObjectStoreOnly],
+        std::time::Duration::from_secs(5),
+    )
+    .await;
+    let chunk = chunks.pop().unwrap();
+
+    Command::cargo_bin("influxdb_iox")
+        .unwrap()
+        .arg("database")
+        .arg("chunk")
+        .arg("load")
+        .arg(&db_name)
+        .arg(chunk.id.get().to_string())
+        .arg("--host")
+        .arg(&addr)
+        .assert()
+        .success()
+        .stdout(predicate::str::contains("loadReadBufferChunk"));
+
+    wait_for_exact_chunk_states(
+        &fixture,
+        &db_name,
+        vec![ChunkStorage::ReadBufferAndObjectStore],
+        std::time::Duration::from_secs(5),
+    )
+    .await;
+}
+
+#[tokio::test]
+async fn test_unload_partition_chunk() {
+    let (fixture, db_name, addr) = setup_load_unload_partition_chunk().await;
     let mut chunks = wait_for_exact_chunk_states(
         &fixture,
         &db_name,