diff --git a/queue.go b/queue.go index b41059d..0001749 100644 --- a/queue.go +++ b/queue.go @@ -11,10 +11,6 @@ type Queue struct { driver Driver } -func NewTestQueueIn(d string) Queue { - return Queue{driver: NewTestDBIn(d)} -} - func NewQueue(driver Driver) Queue { return Queue{driver: driver} } diff --git a/storage.go b/storage.go index ed972c4..32ab077 100644 --- a/storage.go +++ b/storage.go @@ -13,10 +13,6 @@ type Storage struct { driver Driver } -func NewTestStorageIn(d string) Storage { - return Storage{driver: NewTestDBIn(d)} -} - func NewStorage(driver Driver) Storage { return Storage{driver: driver} } diff --git a/storage_test.go b/storage_test.go index a512bef..a77523a 100644 --- a/storage_test.go +++ b/storage_test.go @@ -1,6 +1,36 @@ package main -import "testing" +import ( + "context" + "testing" + "time" +) 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) + } }