From f9bc7806e1bdd810b55e36416b2af7a1c623b462 Mon Sep 17 00:00:00 2001
From: Kyle Ellrott <kellrott@gmail.com>
Date: Tue, 21 Jan 2025 13:54:38 -0800
Subject: [PATCH] Updating sort unit test

---
 test/engine/sort_test.go | 80 +++++++++++++++++++++++++---------------
 1 file changed, 51 insertions(+), 29 deletions(-)

diff --git a/test/engine/sort_test.go b/test/engine/sort_test.go
index 1d22e83a..5ff743c5 100644
--- a/test/engine/sort_test.go
+++ b/test/engine/sort_test.go
@@ -1,15 +1,48 @@
 package main
 
 import (
-	"fmt"
+	"encoding/json"
 	"os"
-	"slices"
+	"strings"
 	"testing"
 
 	"github.com/bmeg/grip/engine/logic"
-	"github.com/cockroachdb/pebble"
 )
 
+type JSONCompare struct{}
+
+// Compare implements logic.SortConf.
+func (j *JSONCompare) Compare(a any, b any) int {
+	return logic.CompareAny(a, b)
+}
+
+// FromBytes implements logic.SortConf.
+func (j *JSONCompare) FromBytes(b []byte) any {
+	if b[0] == '"' {
+		var a string
+		json.Unmarshal(b, &a)
+		return a
+	} else if strings.Compare(string(b), "true") == 0 {
+		return true
+	} else if strings.Compare(string(b), "false") == 0 {
+		return false
+	} else if strings.Contains(string(b), ".") {
+		var a float64
+		json.Unmarshal(b, &a)
+		return a
+	} else {
+		var a int
+		json.Unmarshal(b, &a)
+		return a
+	}
+}
+
+// ToBytes implements logic.SortConf.
+func (j *JSONCompare) ToBytes(a any) []byte {
+	o, _ := json.Marshal(a)
+	return o
+}
+
 func TestKVSort(t *testing.T) {
 
 	path := "kv_sort_test.store"
@@ -25,44 +58,33 @@ func TestKVSort(t *testing.T) {
 		true,
 		false,
 	}
-	slices.SortFunc(keys, logic.CompareAny)
 
+	jConf := JSONCompare{}
+
+	memSort := logic.NewMemSorter[any](&jConf)
 	for _, i := range keys {
-		fmt.Printf("%#v\n", i)
+		memSort.Add(i)
 	}
 
-	c := *pebble.DefaultComparer
-
-	c.Compare = logic.CompareEncoded
-
-	db, err := pebble.Open(path, &pebble.Options{Comparer: &c})
-	if err != nil {
-		t.Errorf("error: %s\n", err)
+	out1 := []any{}
+	for i := range memSort.Sorted() {
+		out1 = append(out1, i)
 	}
 
-	fmt.Printf("===DB sort test===\n")
-
+	kvSort := logic.NewKVSorter(path, &jConf)
 	for _, i := range keys {
-		db.Set(logic.EncodeAny(i), []byte{}, nil)
+		kvSort.Add(i)
 	}
-
-	iter, err := db.NewIter(nil)
-	if err != nil {
-		fmt.Printf("error: %s\n", err)
-	}
-	out := []any{}
-	for iter.First(); iter.Valid(); iter.Next() {
-		j := logic.DecodeAny(iter.Key())
-		out = append(out, j)
-		fmt.Printf("%#v\n", j)
+	out2 := []any{}
+	for i := range kvSort.Sorted() {
+		out2 = append(out2, i)
 	}
 
-	for i := range keys {
-		if logic.CompareAny(keys[i], out[i]) != 0 {
+	for i := range out1 {
+		if logic.CompareAny(out1[i], out2[i]) != 0 {
 			t.Error("mismatch")
 		}
 	}
-	iter.Close()
-	db.Close()
+
 	os.RemoveAll(path)
 }