1package db_test
 2
 3import (
 4	"context"
 5	"testing"
 6
 7	"github.com/charmbracelet/soft-serve/pkg/db"
 8	"github.com/charmbracelet/soft-serve/pkg/db/internal/test"
 9)
10
11func TestBadFromContext(t *testing.T) {
12	ctx := context.TODO()
13	if c := db.FromContext(ctx); c != nil {
14		t.Errorf("FromContext(ctx) => %v, want %v", c, nil)
15	}
16}
17
18func TestGoodFromContext(t *testing.T) {
19	ctx := context.TODO()
20	dbx, err := test.OpenSqlite(ctx, t)
21	if err != nil {
22		t.Fatal(err)
23	}
24	ctx = db.WithContext(ctx, dbx)
25	if c := db.FromContext(ctx); c == nil {
26		t.Errorf("FromContext(ctx) => %v, want %v", c, dbx)
27	}
28}