leaf/filter_test.go

package leaf

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestFilter(t *testing.T) {
	t.Run("merge two filter documents", func(t *testing.T) {
		doc1 := Filter(
			Gt("field1", 1),
			Lt("field2", 10),
		)

		doc2 := Filter(
			Ne("field1", 2),
			Eq("field3", "test"),
		)

		expected := FilterDocument{
			"field1": {"$gt": 1, "$ne": 2},
			"field2": {"$lt": 10},
			"field3": {"$eq": "test"},
		}

		assert.Equal(t, expected, Filter(doc1, doc2))
	})

	t.Run("merge three filter documents", func(t *testing.T) {
		doc1 := Filter(
			Gt("field1", 1),
			Lt("field2", 10),
		)

		doc2 := Filter(
			Ne("field1", 2),
			Eq("field3", "test"),
		)

		doc3 := Filter(
			Gt("field1", 5),
			Ne("field2", 5),
			Eq("field4", "hello"),
		)

		expected := FilterDocument{
			"field1": {"$gt": 5, "$ne": 2},
			"field2": {"$lt": 10, "$ne": 5},
			"field3": {"$eq": "test"},
			"field4": {"$eq": "hello"},
		}

		assert.Equal(t, expected, Filter(doc1, doc2, doc3))
	})
}