shortcut quantity change

cart-grain_test.go

@@ -16,6 +16,57 @@ func GetMessage(t uint16, data interface{}) *Message {
 	}
 }
 
+func TestAddToCartShortCut(t *testing.T) {
+	grain, err := spawn(ToCartId("kalle"))
+	if err != nil {
+		t.Errorf("Error spawning: %v\n", err)
+	}
+	if len(grain.Items) != 0 {
+		t.Errorf("Expected 0 items, got %d\n", len(grain.Items))
+	}
+	msg := GetMessage(AddItemType, &messages.AddItem{
+		Quantity: 2,
+		Price:    100,
+		Sku:      "123",
+		Name:     "Test item",
+		Image:    "test.jpg",
+	})
+
+	_, err = grain.HandleMessage(msg, false)
+	if err != nil {
+		t.Errorf("Error handling message: %v\n", err)
+	}
+	if len(grain.Items) != 1 {
+		t.Errorf("Expected 1 item, got %d\n", len(grain.Items))
+	}
+	if grain.Items[0].Quantity != 2 {
+		t.Errorf("Expected quantity 2, got %d\n", grain.Items[0].Quantity)
+	}
+	if len(grain.storageMessages) != 1 {
+		t.Errorf("Expected 1 storage message, got %d\n", len(grain.storageMessages))
+	}
+	shortCutMessage := GetMessage(AddRequestType, &messages.AddRequest{
+		Quantity: 2,
+		Sku:      "123",
+	})
+	_, err = grain.HandleMessage(shortCutMessage, false)
+	if err != nil {
+		t.Errorf("Error handling message: %v\n", err)
+	}
+	if len(grain.Items) != 1 {
+		t.Errorf("Expected 1 item, got %d\n", len(grain.Items))
+	}
+	if len(grain.storageMessages) != 2 {
+		t.Errorf("Expected 2 storage message, got %d\n", len(grain.storageMessages))
+	}
+	if grain.storageMessages[0].Type != AddItemType {
+		t.Errorf("Expected AddItemType, got %d\n", grain.storageMessages[0].Type)
+	}
+	if grain.storageMessages[1].Type != AddRequestType {
+		t.Errorf("Expected AddRequestType, got %d\n", grain.storageMessages[1].Type)
+	}
+}
+
 func TestAddToCart(t *testing.T) {
 	grain, err := spawn(ToCartId("kalle"))
 	if err != nil {