add:add a unit test for delivery push

adjust · May 15, 2024 · ccfe176 · ccfe176
1 parent b3b5057
commit ccfe176
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test_delivery_test.go b/test_delivery_test.go
@@ -34,3 +34,15 @@ func TestDeliveryReject(t *testing.T) {
 	assert.Equal(t, ErrorNotFound, delivery.Ack())
 	assert.Equal(t, Rejected, delivery.State)
 }
+
+func TestDeliveryPush(t *testing.T) {
+	delivery := NewTestDelivery("p")
+	assert.Equal(t, Unacked, delivery.State)
+	assert.NoError(t, delivery.Push())
+	assert.Equal(t, Pushed, delivery.State)
+
+	assert.Equal(t, ErrorNotFound, delivery.Push())
+	assert.Equal(t, ErrorNotFound, delivery.Reject())
+	assert.Equal(t, ErrorNotFound, delivery.Ack())
+	assert.Equal(t, Pushed, delivery.State)
+}