use crate::util::test_client::TestClient;

#[tokio::test]
async fn workspace_usage_put_blob_test() {
  let client = TestClient::new_user_without_ws_conn().await;
  let mime = mime::TEXT_PLAIN_UTF_8;
  let file_1 = client.upload_blob("123", &mime).await;
  let file_2 = client.upload_blob("456", &mime).await;

  let usage = client.get_workspace_usage().await;
  assert_eq!(usage.consumed_capacity, 6);

  // after the test, delete the files
  client.delete_file(&file_1).await;
  client.delete_file(&file_2).await;
}

#[tokio::test]
async fn workspace_usage_put_and_then_delete_blob_test() {
  let client = TestClient::new_user_without_ws_conn().await;
  let mime = mime::TEXT_PLAIN_UTF_8;
  let file_1 = client.upload_blob("123", &mime).await;
  let file_2 = client.upload_blob("456", &mime).await;

  client.delete_file(&file_1).await;
  let usage = client.get_workspace_usage().await;
  assert_eq!(usage.consumed_capacity, 3);

  client.delete_file(&file_2).await;
  let usage = client.get_workspace_usage().await;
  assert_eq!(usage.consumed_capacity, 0);
}