searchreplacefuncs_test.go

// Copyright (c) 2025 Dell Inc. or its subsidiaries.  All rights reserved.
// Licensed to You under the Apache License, Version 2.0.   See the LICENSE file for more details.

package jsonstream_test

import (
	"testing"

	"github.com/sailfishdell/jsonstream"
)

func alwaysMatch(p jsonstream.JSONPath) jsonstream.JSONPath {
	return p
}

func neverMatch(jsonstream.JSONPath) jsonstream.JSONPath {
	return nil
}

func TestDropperByMatchFunc(t *testing.T) {
	myIter := &testTokenIterator{
		returns: []nextTokenOutput{
			{Path: jsonstream.NewJSONPath("foo"), Token: []byte(`"bar"`)},
		},
	}
	filterFn := jsonstream.DropPropertyByMatchFunc(alwaysMatch)
	ti := filterFn(myIter)
	path, tok, err := ti.NextToken()
	compareEOFCase(t, path, tok, err)
	myIter.count = 0
	myIter.isClosed = false
	filterFn = jsonstream.DropPropertyByMatchFunc(neverMatch)
	ti = filterFn(myIter)
	path, tok, err = ti.NextToken()
	compareNextTokenOutput(t, path, tok, err)
	err = ti.Close()
	if err != nil {
		t.Errorf("Expected no error from Close")
	}
}

func BenchmarkDropperByMatchFunc(b *testing.B) {
	myIter := &testTokenIterator{
		returns: []nextTokenOutput{
			{Path: jsonstream.NewJSONPath("foo"), Token: []byte(`"bar"`)},
		},
	}
	b.Run("AlwaysMatch", func(b *testing.B) {
		b.ReportAllocs()
		for i := 0; i < b.N; i++ {
			filterFn := jsonstream.DropPropertyByMatchFunc(alwaysMatch)
			ti := filterFn(myIter)
			_, _, _ = ti.NextToken()
			myIter.count = 0
		}
	})
	b.Run("NeverMatch", func(b *testing.B) {
		b.ReportAllocs()
		for i := 0; i < b.N; i++ {
			filterFn := jsonstream.DropPropertyByMatchFunc(neverMatch)
			ti := filterFn(myIter)
			_, _, _ = ti.NextToken()
			myIter.count = 0
		}
	})
}

func TestDropPropertyByPathMatch(t *testing.T) {
	myIter := &testTokenIterator{
		returns: []nextTokenOutput{
			{Path: jsonstream.NewJSONPath("foo"), Token: []byte(`"bar"`)},
		},
	}
	ti := jsonstream.DropPropertyByPathMatch(jsonstream.NewJSONPath("foo"), myIter)
	path, tok, err := ti.NextToken()
	compareEOFCase(t, path, tok, err)
	myIter.count = 0
	myIter.isClosed = false
	ti = jsonstream.DropPropertyByPathMatch(jsonstream.NewJSONPath("foo1"), myIter)
	path, tok, err = ti.NextToken()
	compareNextTokenOutput(t, path, tok, err)
	err = ti.Close()
	if err != nil {
		t.Errorf("Expected no error from Close")
	}
}

func BenchmarkDropPropertyByPathMatch(b *testing.B) {
	myIter := &testTokenIterator{
		returns: []nextTokenOutput{
			{Path: jsonstream.NewJSONPath("foo"), Token: []byte(`"bar"`)},
		},
	}
	b.Run("Match", func(b *testing.B) {
		b.ReportAllocs()
		prefix := jsonstream.NewJSONPath("foo")
		for i := 0; i < b.N; i++ {
			ti := jsonstream.DropPropertyByPathMatch(prefix, myIter)
			_, _, _ = ti.NextToken()
			myIter.count = 0
		}
	})
	b.Run("NoMatch", func(b *testing.B) {
		b.ReportAllocs()
		prefix := jsonstream.NewJSONPath("foo1")
		for i := 0; i < b.N; i++ {
			ti := jsonstream.DropPropertyByPathMatch(prefix, myIter)
			_, _, _ = ti.NextToken()
			myIter.count = 0
		}
	})
}