Some extra tests

risico · Oct 6, 2023 · c124f4b · c124f4b
1 parent 33103b4
commit c124f4b
Showing 1 changed file with 53 additions and 0 deletions.
diff --git a/queue_test.go b/queue_test.go
@@ -33,6 +33,59 @@ func TestSqueueLite(t *testing.T) {
 	q.Close()
 }
 
+func TestNamespacing(t *testing.T) {
+	t.Parallel()
+	q, err := squeuelite.New(squeuelite.Params{
+		DatabasePath: ":memory:",
+	})
+	assert.NoError(t, err)
+
+	id, err := q.EnqueueWithParams("something", squeuelite.EnqueueParams{
+		Namespace: "test",
+	})
+	assert.NoError(t, err)
+	assert.NotEmpty(t, id)
+	message, err := q.Dequeue("default")
+	assert.NoError(t, err)
+	assert.Nil(t, message)
+
+	message, err = q.Dequeue("test")
+	assert.NoError(t, err)
+	assert.NotNil(t, message)
+
+	if message != nil {
+		err = q.Done(id)
+		assert.NoError(t, err)
+	}
+
+	q.Close()
+}
+
+func TestPrune(t *testing.T) {
+	t.Parallel()
+	q, err := squeuelite.New(squeuelite.Params{
+		DatabasePath: ":memory:",
+	})
+	assert.NoError(t, err)
+
+	id, err := q.Enqueue("something")
+	assert.NoError(t, err)
+	assert.NotEmpty(t, id)
+	message, err := q.Dequeue("default")
+	assert.NoError(t, err)
+	assert.NotNil(t, message)
+
+	if message != nil {
+		err = q.Done(id)
+		assert.NoError(t, err)
+	}
+
+	err = q.Prune()
+	assert.NoError(t, err)
+
+	q.Close()
+}
+
 func TestSqueueLiteLoad(t *testing.T) {
 	t.Parallel()
 	defer os.Remove("test.db")