set deliveries on all items

message-handler_test.go

@@ -88,3 +88,42 @@ func TestItemRequest(t *testing.T) {
 		t.Errorf("Expected sku '123', got %s\n", r.Sku)
 	}
 }
+
+func TestSetDeliveryMssage(t *testing.T) {
+	h, err := GetMessageHandler(SetDeliveryType)
+	if err != nil {
+		t.Errorf("Error getting message handler: %v\n", err)
+	}
+	if h == nil {
+		t.Errorf("Expected message handler, got nil\n")
+	}
+	message := Message{
+		Type: SetDeliveryType,
+		Content: &messages.SetDelivery{
+			Provider: "test",
+			Items:    []int64{1, 2},
+		},
+	}
+	var b bytes.Buffer
+	err = h.Write(&message, &b)
+	if err != nil {
+		t.Errorf("Error writing message: %v\n", err)
+	}
+	result, err := h.Read(b.Bytes())
+	if err != nil {
+		t.Errorf("Error reading message: %v\n", err)
+	}
+	if result == nil {
+		t.Errorf("Expected result, got nil\n")
+	}
+	r, ok := result.(*messages.SetDelivery)
+	if !ok {
+		t.Errorf("Expected AddRequest, got %T\n", result)
+	}
+	if len(r.Items) != 2 {
+		t.Errorf("Expected 2 items, got %d\n", len(r.Items))
+	}
+	if r.Provider != "test" {
+		t.Errorf("Expected provider 'test', got %s\n", r.Provider)
+	}
+}