mleku
869006c4c3
Some checks failed
Go / build-and-release (push) Has been cancelled
Introduce tests to validate functionality for new policy fields, including `max_expiry_duration`, `protected_required`, `identifier_regex`, and `follows_whitelist_admins`. Also, cover combinations of new and existing fields to ensure compatibility and precedence rules are correctly enforced. bump to v0.31.2
118 lines
2.9 KiB
Go
118 lines
2.9 KiB
Go
package database
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
"context"
|
|
"os"
|
|
"sort"
|
|
"testing"
|
|
|
|
"git.mleku.dev/mleku/nostr/encoders/event"
|
|
"git.mleku.dev/mleku/nostr/encoders/event/examples"
|
|
"lol.mleku.dev/chk"
|
|
)
|
|
|
|
func TestGetSerialById(t *testing.T) {
|
|
// Create a temporary directory for the database
|
|
tempDir, err := os.MkdirTemp("", "test-db-*")
|
|
if err != nil {
|
|
t.Fatalf("Failed to create temporary directory: %v", err)
|
|
}
|
|
defer os.RemoveAll(tempDir) // Clean up after the test
|
|
|
|
// Create a context and cancel function for the database
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
// Initialize the database
|
|
db, err := New(ctx, cancel, tempDir, "info")
|
|
if err != nil {
|
|
t.Fatalf("Failed to create database: %v", err)
|
|
}
|
|
defer db.Close()
|
|
|
|
// Create a scanner to read events from examples.Cache
|
|
scanner := bufio.NewScanner(bytes.NewBuffer(examples.Cache))
|
|
scanner.Buffer(make([]byte, 0, 1_000_000_000), 1_000_000_000)
|
|
|
|
// Collect all events first
|
|
var allEvents []*event.E
|
|
for scanner.Scan() {
|
|
chk.E(scanner.Err())
|
|
b := scanner.Bytes()
|
|
ev := event.New()
|
|
|
|
// Unmarshal the event
|
|
if _, err = ev.Unmarshal(b); chk.E(err) {
|
|
ev.Free()
|
|
t.Fatal(err)
|
|
}
|
|
|
|
allEvents = append(allEvents, ev)
|
|
}
|
|
|
|
// Check for scanner errors
|
|
if err = scanner.Err(); err != nil {
|
|
t.Fatalf("Scanner error: %v", err)
|
|
}
|
|
|
|
// Sort events by timestamp to ensure addressable events are processed in chronological order
|
|
sort.Slice(allEvents, func(i, j int) bool {
|
|
return allEvents[i].CreatedAt < allEvents[j].CreatedAt
|
|
})
|
|
|
|
// Now process the sorted events
|
|
eventCount := 0
|
|
skippedCount := 0
|
|
var events []*event.E
|
|
|
|
for _, ev := range allEvents {
|
|
// Save the event to the database
|
|
if _, err = db.SaveEvent(ctx, ev); err != nil {
|
|
// Skip events that fail validation (e.g., kind 3 without p tags)
|
|
skippedCount++
|
|
continue
|
|
}
|
|
|
|
events = append(events, ev)
|
|
eventCount++
|
|
}
|
|
|
|
t.Logf("Successfully saved %d events to the database (skipped %d invalid events)", eventCount, skippedCount)
|
|
|
|
// Test GetSerialById with a known event ID
|
|
if len(events) < 4 {
|
|
t.Fatalf("Need at least 4 saved events, got %d", len(events))
|
|
}
|
|
testEvent := events[3]
|
|
|
|
// Get the serial by ID
|
|
serial, err := db.GetSerialById(testEvent.ID)
|
|
if err != nil {
|
|
t.Fatalf("Failed to get serial by ID: %v", err)
|
|
}
|
|
|
|
// Verify the serial is not nil
|
|
if serial == nil {
|
|
t.Fatal("Expected serial to be non-nil, but got nil")
|
|
}
|
|
|
|
// Test with a non-existent ID
|
|
nonExistentId := make([]byte, len(testEvent.ID))
|
|
// Ensure it's different from any real ID
|
|
for i := range nonExistentId {
|
|
nonExistentId[i] = ^testEvent.ID[i]
|
|
}
|
|
|
|
serial, err = db.GetSerialById(nonExistentId)
|
|
if err == nil {
|
|
t.Fatalf("Expected error for non-existent ID, but got none")
|
|
}
|
|
|
|
// For non-existent Ids, the function should return nil serial and an error
|
|
if serial != nil {
|
|
t.Fatalf("Expected nil serial for non-existent ID, but got: %v", serial)
|
|
}
|
|
}
|