Add timeseries, discriminator

2026-04-17 11:38:12 -04:00
parent cba3e326e8
commit 99b36e577e
18 changed files with 897 additions and 130 deletions
--- a/replace_test.go
+++ b/replace_test.go
@@ -0,0 +1,181 @@
+package mongo
+
+import (
+	"os"
+	"context"
+	"testing"
+	"encoding/json"
+
+	"go.mongodb.org/mongo-driver/v2/bson"
+)
+
+func TestReplace(t *testing.T) {
+	data := map[string]any {
+		"_id": "su_2b3c00",
+		"name": "Ava Thompson",
+		"age": int32(26),
+	}
+
+	ctx := context.Background()
+
+	o, err := client.InsertOne(ctx, "mydb", "mycollection", data)
+	if err != nil {	t.Fatalf("Failed to insertOne %#v", err) }
+
+	id := o["_id"].(string)
+
+	// first we retrieve the entity
+	fetched, err := client.GetOne(ctx, "mydb", "mycollection", id)
+	if err != nil {	t.Fatalf("Failed to fetch %#v", err) }
+	
+	fetched["name"] = "Noah Patel"
+
+	replaced, err := client.Replace(ctx, "mydb", "mycollection", id, fetched)
+	if err != nil {	t.Fatalf("Failed to replace %#v", err) }
+
+	// t.Fatalf("-> %v", replaced)
+	if replaced["_id"] != fetched["_id"] {
+		t.Fatalf("Not the same entity")
+	}
+
+	if replaced["name"] != "Noah Patel" {
+		t.Fatal("unexpected name")
+	}
+}
+
+// Make sure discrimination field is used when replacing.
+func TestReplace_Discrimination(t *testing.T) {
+	// 1. Register schemas
+	schemaStore, err := os.ReadFile("./.test/store.json")
+	if err != nil { t.Fatal(err) }
+
+	schemaOffer, err := os.ReadFile("./.test/offer.json")
+	if err != nil { t.Fatal(err) }
+
+	var store bson.M
+	if err := json.Unmarshal(schemaStore, &store); err != nil {
+		t.Fatalf("Length: %d, First bytes: %x\n", len(schemaStore), schemaStore[:4])
+	}
+
+	var offer bson.M
+	if err := json.Unmarshal(schemaOffer, &offer); err != nil {
+		t.Fatalf("Length: %d, First bytes: %x\n", len(schemaOffer), schemaOffer[:4])
+	}
+
+	client := GetMongoClient()
+	client.AddDefinition(store)
+	client.AddDefinition(offer)
+
+	// Save data
+	ctx1 := context.Background()
+	ctx1 = context.WithValue(ctx1, "account", "xxxxxx")
+	ctx1 = context.WithValue(ctx1, "store", "str_1234")
+
+	// Save offer
+  offer1 := map[string]any { "name": "OSRAM 10" }
+  saved, err := client.InsertOne(ctx1, "mydb", "offer", offer1)
+  if err != nil { t.Fatalf("Failed to insertOne %#v", err) }
+
+  id := saved["_id"].(string)
+
+  // // Now replace it
+  saved["name"] = "** OSRAM 10 CHANGED **"
+	replaced, err := client.Replace(ctx1, "mydb", "offer", id, saved)
+	if err != nil { t.Fatalf("Failed to replace %#v", err) }
+
+	if replaced["name"] != "** OSRAM 10 CHANGED **" {
+		t.Fatal("Unexpected replaced object")
+	}
+
+	found, err := client.GetOne(ctx1, "mydb", "offer", id)
+	if err != nil { t.Fatalf("Failed to find %#v", err) }
+	
+	if found["name"] != "** OSRAM 10 CHANGED **" {
+		t.Fatal("Updates did not persist")
+	}
+}
+
+func TestReplace_Discrimination_NoStore(t *testing.T) {
+	// 1. Register schemas
+	schemaStore, err := os.ReadFile("./.test/store.json")
+	if err != nil { t.Fatal(err) }
+
+	schemaOffer, err := os.ReadFile("./.test/offer.json")
+	if err != nil { t.Fatal(err) }
+
+	var store bson.M
+	if err := json.Unmarshal(schemaStore, &store); err != nil {
+		t.Fatalf("Length: %d, First bytes: %x\n", len(schemaStore), schemaStore[:4])
+	}
+
+	var offer bson.M
+	if err := json.Unmarshal(schemaOffer, &offer); err != nil {
+		t.Fatalf("Length: %d, First bytes: %x\n", len(schemaOffer), schemaOffer[:4])
+	}
+
+	client := GetMongoClient()
+	client.AddDefinition(store)
+	client.AddDefinition(offer)
+
+	// Save data
+	ctx1 := context.Background()
+	ctx1 = context.WithValue(ctx1, "store", "str_1234")
+	
+	// Save offer
+  offer1 := map[string]any { "name": "OSRAM 11" }
+  saved, err := client.InsertOne(ctx1, "mydb", "offer", offer1)
+  if err != nil { t.Fatalf("Failed to insertOne %#v", err) }
+
+  id := saved["_id"].(string)
+
+  // // Now replace it
+  saved["name"] = "** OSRAM 11 CHANGED **"
+	_, err = client.Replace(context.Background(), "mydb", "offer", id, saved)
+	if err == nil { 
+		t.Fatal("should have required store") 
+	}
+
+}
+
+// Should save the right store regardless of the input
+func TestReplace_Discrimination_EnsureStore(t *testing.T) {
+	// 1. Register schemas
+	schemaStore, err := os.ReadFile("./.test/store.json")
+	if err != nil { t.Fatal(err) }
+
+	schemaOffer, err := os.ReadFile("./.test/offer.json")
+	if err != nil { t.Fatal(err) }
+
+	var store bson.M
+	if err := json.Unmarshal(schemaStore, &store); err != nil {
+		t.Fatalf("Length: %d, First bytes: %x\n", len(schemaStore), schemaStore[:4])
+	}
+
+	var offer bson.M
+	if err := json.Unmarshal(schemaOffer, &offer); err != nil {
+		t.Fatalf("Length: %d, First bytes: %x\n", len(schemaOffer), schemaOffer[:4])
+	}
+
+	client := GetMongoClient()
+	client.AddDefinition(store)
+	client.AddDefinition(offer)
+
+	// Save data
+	ctx1 := context.Background()
+	ctx1 = context.WithValue(ctx1, "store", "str_1234")
+	
+	// Save offer
+  offer1 := map[string]any { "name": "OSRAM 12" }
+  saved, err := client.InsertOne(ctx1, "mydb", "offer", offer1)
+  if err != nil { t.Fatalf("Failed to insertOne %#v", err) }
+
+  id := saved["_id"].(string)
+
+  // // Now replace it
+  saved["store"] = "str_0000"
+	changed, err := client.Replace(ctx1, "mydb", "offer", id, saved)
+	if err != nil { t.Fatalf("Failed to replace %v", err) }
+	if changed["store"] != "str_1234" { 
+		t.Fatal("Should have ensured store") 
+	}
+
+}