little things all tested

main
Bel LaPointe 2024-04-12 09:13:48 -06:00
parent 6ba23e61c3
commit 02fecb7eb2
3 changed files with 31 additions and 9 deletions

View File

@ -11,10 +11,6 @@ type Queue struct {
driver Driver driver Driver
} }
func NewTestQueueIn(d string) Queue {
return Queue{driver: NewTestDBIn(d)}
}
func NewQueue(driver Driver) Queue { func NewQueue(driver Driver) Queue {
return Queue{driver: driver} return Queue{driver: driver}
} }

View File

@ -13,10 +13,6 @@ type Storage struct {
driver Driver driver Driver
} }
func NewTestStorageIn(d string) Storage {
return Storage{driver: NewTestDBIn(d)}
}
func NewStorage(driver Driver) Storage { func NewStorage(driver Driver) Storage {
return Storage{driver: driver} return Storage{driver: driver}
} }

View File

@ -1,6 +1,36 @@
package main package main
import "testing" import (
"context"
"testing"
"time"
)
func TestStorage(t *testing.T) { func TestStorage(t *testing.T) {
ctx, can := context.WithTimeout(context.Background(), time.Second)
defer can()
db := NewTestDBIn(t.TempDir())
defer db.Close()
s := NewStorage(db)
if _, err := s.Get(ctx, "id"); err != ErrNotFound {
t.Error("failed to get 404", err)
}
m := Message{
ID: "id",
TS: 1,
}
if err := s.Upsert(ctx, m); err != nil {
t.Error("failed to upsert", err)
}
if m2, err := s.Get(ctx, "id"); err != nil {
t.Error("failed to get", err)
} else if m != m2 {
t.Error(m2)
}
} }