replacer_bench_test.mbt

// Copyright 2026 International Digital Economy Academy
//
// 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.

///|
fn make_replacer_bench_data(fields : Int) -> (Json, Array[StringView]) {
  let object = Map([])
  let keys : Array[StringView] = Array::makei(2500, i => {
    "key-" + (i * 2).to_string()
  })
  for i in 0..<fields {
    object["key-" + i.to_string()] = Json::number(i.to_double())
  }
  (Json::object(object), keys)
}

///|
test "bench Json::stringify Replacer::keep keys=2500 fields=5" (it : @bench.T) {
  let (json, keys) = make_replacer_bench_data(5)
  let replacer = @json.Replacer::keep(keys)
  it.bench(fn() { it.keep(json.stringify(replacer~).length()) })
}

///|
test "bench Json::stringify Replacer::keep keys=2500 fields=50" (it : @bench.T) {
  let (json, keys) = make_replacer_bench_data(50)
  let replacer = @json.Replacer::keep(keys)
  it.bench(fn() { it.keep(json.stringify(replacer~).length()) })
}

///|
test "bench Json::stringify Replacer::keep keys=2500 fields=5000" (
  it : @bench.T,
) {
  let (json, keys) = make_replacer_bench_data(5000)
  let replacer = @json.Replacer::keep(keys)
  it.bench(fn() { it.keep(json.stringify(replacer~).length()) })
}

///|
test "bench Json::stringify Replacer::keep keys=3 fields=5000" (it : @bench.T) {
  let (json, _) = make_replacer_bench_data(5000)
  let keys : Array[StringView] = ["key-0", "key-2500", "key-4999"]
  let replacer = @json.Replacer::keep(keys)
  it.bench(fn() { it.keep(json.stringify(replacer~).length()) })
}

///|
/// A replacer built fresh for each call must not pay for an index it never
/// amortizes.
test "bench Json::stringify Replacer::keep fresh keys=2500 fields=5" (
  it : @bench.T,
) {
  let (json, keys) = make_replacer_bench_data(5)
  it.bench(fn() {
    let replacer = @json.Replacer::keep(keys)
    it.keep(json.stringify(replacer~).length())
  })
}