#!/usr/bin/env bats

# Copyright 2024 Versity Software
# This file is licensed under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http:#www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

load ./bats-support/load
load ./bats-assert/load

source ./tests/setup.sh
source ./tests/commands/get_object.sh
source ./tests/commands/put_object.sh
source ./tests/drivers/create_bucket/create_bucket_rest.sh
source ./tests/drivers/list_buckets/list_buckets_rest.sh
source ./tests/util/util_rest.sh

test_file="test_file"

@test "REST - check, enable, suspend versioning" {
  if [ "$RECREATE_BUCKETS" == "false" ]; then
    skip "cannot change versioning status for static buckets"
  fi
  run get_bucket_name "$BUCKET_ONE_NAME"
  assert_success
  bucket_name="$output"

  run setup_bucket_v2 "$bucket_name"
  assert_success

  run check_versioning_status_rest "$bucket_name" ""
  assert_success

  run put_bucket_versioning_rest "$bucket_name" "Enabled"
  assert_success

  run check_versioning_status_rest "$bucket_name" "Enabled"
  assert_success

  run put_bucket_versioning_rest "$bucket_name" "Suspended"
  assert_success

  run check_versioning_status_rest "$bucket_name" "Suspended"
  assert_success
}

@test "test_rest_versioning" {
  if [ "$RECREATE_BUCKETS" == "false" ]; then
    skip "cannot change versioning status for static buckets"
  fi
  run get_bucket_name "$BUCKET_ONE_NAME"
  assert_success
  bucket_name="$output"

  run setup_bucket_and_file_v2 "$bucket_name" "$test_file"
  assert_success

  run put_object "rest" "$TEST_FILE_FOLDER/$test_file" "$bucket_name" "$test_file"
  assert_success

  run get_and_check_versions_rest "$bucket_name" "$test_file" "1" "true" "true"
  assert_success

  run put_bucket_versioning_rest "$bucket_name" "Enabled"
  assert_success

  run get_and_check_versions_rest "$bucket_name" "$test_file" "1" "true" "true"
  assert_success

  run put_object "rest" "$TEST_FILE_FOLDER/$test_file" "$bucket_name" "$test_file"
  assert_success

  run get_and_check_versions_rest "$bucket_name" "$test_file" "2" "true" "false" "false" "true"
  assert_success
}

@test "versioning - add version, then delete and check for marker" {
  if [ "$RECREATE_BUCKETS" == "false" ]; then
    skip "cannot change versioning status for static buckets"
  fi
  run get_bucket_name "$BUCKET_ONE_NAME"
  assert_success
  bucket_name="$output"

  run setup_bucket_and_file_v2 "$bucket_name" "$test_file"
  assert_success

  run put_object "rest" "$TEST_FILE_FOLDER/$test_file" "$bucket_name" "$test_file"
  assert_success

  run put_bucket_versioning_rest "$bucket_name" "Enabled"
  assert_success

  run delete_object_rest "$bucket_name" "$test_file"
  assert_success

  run check_versions_after_file_deletion "$bucket_name" "$test_file"
  assert_success
}

@test "versioning - retrieve after delete" {
  run get_bucket_name "$BUCKET_ONE_NAME"
  assert_success
  bucket_name="$output"

  run setup_bucket_and_file_v2 "$bucket_name" "$test_file"
  assert_success

  run put_object "rest" "$TEST_FILE_FOLDER/$test_file" "$bucket_name" "$test_file"
  assert_success

  run put_bucket_versioning_rest "$bucket_name" "Enabled"
  assert_success

  run delete_object "s3api" "$bucket_name" "$test_file"
  assert_success

  run get_object "s3api" "$bucket_name" "$test_file" "$TEST_FILE_FOLDER/$test_file-copy"
  assert_failure
}

@test "REST - HeadObject does not return 405 with versioning, after file deleted" {
  if [ "$RECREATE_BUCKETS" == "false" ] || [[ ( -z "$VERSIONING_DIR" ) && ( "$DIRECT" != "true" ) ]]; then
    skip "test isn't valid for this configuration"
  fi
  run get_bucket_name "$BUCKET_ONE_NAME"
  assert_success
  bucket_name="$output"

  run setup_bucket_object_lock_enabled_v2 "$bucket_name"
  assert_success

  run create_test_files "$test_file"
  assert_success

  run put_object "rest" "$TEST_FILE_FOLDER/$test_file" "$bucket_name" "$test_file"
  assert_success

  run delete_object "s3api" "$bucket_name" "$test_file"
  assert_success

  run verify_object_not_found "$bucket_name" "$test_file"
  assert_success
}

@test "REST - HeadObject returns 405 when querying DeleteMarker" {
  if [ "$RECREATE_BUCKETS" == "false" ] || [[ ( -z "$VERSIONING_DIR" ) && ( "$DIRECT" != "true" ) ]]; then
    skip "test isn't valid for this configuration"
  fi
  run get_bucket_name "$BUCKET_ONE_NAME"
  assert_success
  bucket_name="$output"

  run setup_bucket_object_lock_enabled_v2 "$bucket_name"
  assert_success

  run create_test_files "$test_file"
  assert_success

  run put_object "rest" "$TEST_FILE_FOLDER/$test_file" "$bucket_name" "$test_file"
  assert_success

  run delete_object "s3api" "$bucket_name" "$test_file"
  assert_success

  run get_delete_marker_and_verify_405 "$bucket_name" "$test_file"
  assert_success
}