add tests for hydrating/dehydrating attributes

Author: kbond

src/LiveComponent/tests/Integration/LiveComponentHydratorTest.php

@@ -18,7 +18,9 @@
 use Symfony\UX\LiveComponent\Tests\Fixture\Component\Component2;
 use Symfony\UX\LiveComponent\Tests\Fixture\Component\Component3;
 use Symfony\UX\LiveComponent\Tests\Fixture\Entity\Entity1;
+use Symfony\UX\TwigComponent\ComponentAttributes;
 use Symfony\UX\TwigComponent\ComponentFactory;
+use Symfony\UX\TwigComponent\HasAttributesTrait;
 use function Zenstruck\Foundry\create;
 use Zenstruck\Foundry\Test\Factories;
 use Zenstruck\Foundry\Test\ResetDatabase;
@@ -239,4 +241,32 @@ public function testCorrectlyUsesCustomFrontendNameInDehydrateAndHydrate(): void
         $this->assertSame('value1', $component->prop1);
         $this->assertSame('value2', $component->prop2);
     }
+
+    public function testCanDehydrateAndHydrateComponentsWithAttributes(): void
+    {
+        if (!class_exists(ComponentAttributes::class)) {
+            $this->markTestSkipped('Attributes trait not available.');
+        }
+
+        /** @var LiveComponentHydrator $hydrator */
+        $hydrator = self::getContainer()->get('ux.live_component.component_hydrator');
+
+        $component = new class() {
+            use HasAttributesTrait;
+        };
+        $instance = clone $component;
+        $instance->attributes = new ComponentAttributes($attributes = ['class' => 'foo']);
+
+        $this->assertSame($attributes, $instance->attributes->all());
+
+        $dehydrated = $hydrator->dehydrate($instance);
+
+        $this->assertArrayHasKey('attributes', $dehydrated);
+        $this->assertSame($attributes, $dehydrated['attributes']);
+        $this->assertFalse(isset($component->prop));
+
+        $hydrator->hydrate($component, $dehydrated);
+
+        $this->assertSame($attributes, $component->attributes->all());
+    }
 }

src/TwigComponent/src/HasAttributesTrait.php

@@ -24,9 +24,9 @@ trait HasAttributesTrait
     #[LiveProp(hydrateWith: 'hydrateAttributes', dehydrateWith: 'dehydrateAttributes')]
     public ComponentAttributes $attributes;
 
-    public function setAttributes(array $attributes): void
+    public function setAttributes(array|ComponentAttributes $attributes): void
     {
-        $this->attributes = new ComponentAttributes($attributes);
+        $this->attributes = is_array($attributes) ? new ComponentAttributes($attributes) : $attributes;
     }
 
     #[PostMount(priority: -1000)]