[Agent][Chat] Add more unit tests

src/agent/tests/InputTest.php

@@ -0,0 +1,123 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Agent\Tests;
+
+use PHPUnit\Framework\TestCase;
+use Symfony\AI\Agent\Input;
+use Symfony\AI\Platform\Message\Message;
+use Symfony\AI\Platform\Message\MessageBag;
+
+final class InputTest extends TestCase
+{
+    public function testConstructorSetsProperties()
+    {
+        $messageBag = new MessageBag();
+        $options = ['temperature' => 0.7, 'max_tokens' => 100];
+
+        $input = new Input('gpt-4', $messageBag, $options);
+
+        $this->assertSame('gpt-4', $input->getModel());
+        $this->assertSame($messageBag, $input->getMessageBag());
+        $this->assertSame($options, $input->getOptions());
+    }
+
+    public function testConstructorWithDefaultOptions()
+    {
+        $messageBag = new MessageBag();
+
+        $input = new Input('claude-3', $messageBag);
+
+        $this->assertSame('claude-3', $input->getModel());
+        $this->assertSame($messageBag, $input->getMessageBag());
+        $this->assertSame([], $input->getOptions());
+    }
+
+    public function testGetModel()
+    {
+        $messageBag = new MessageBag();
+        $input = new Input('test-model', $messageBag);
+
+        $this->assertSame('test-model', $input->getModel());
+    }
+
+    public function testSetModel()
+    {
+        $messageBag = new MessageBag();
+        $input = new Input('original-model', $messageBag);
+
+        $input->setModel('new-model');
+
+        $this->assertSame('new-model', $input->getModel());
+    }
+
+    public function testGetMessageBag()
+    {
+        $messageBag = new MessageBag();
+        $messageBag->add(Message::ofUser('Hello'));
+
+        $input = new Input('model', $messageBag);
+
+        $result = $input->getMessageBag();
+
+        $this->assertSame($messageBag, $result);
+        $this->assertCount(1, $result);
+    }
+
+    public function testSetMessageBag()
+    {
+        $originalMessageBag = new MessageBag();
+        $input = new Input('model', $originalMessageBag);
+
+        $newMessageBag = new MessageBag();
+        $newMessageBag->add(Message::ofUser('New message'));
+
+        $input->setMessageBag($newMessageBag);
+
+        $result = $input->getMessageBag();
+        $this->assertSame($newMessageBag, $result);
+        $this->assertCount(1, $result);
+    }
+
+    public function testGetOptions()
+    {
+        $messageBag = new MessageBag();
+        $options = ['foo' => 'bar', 'baz' => 42];
+
+        $input = new Input('model', $messageBag, $options);
+
+        $this->assertSame($options, $input->getOptions());
+    }
+
+    public function testSetOptions()
+    {
+        $messageBag = new MessageBag();
+        $input = new Input('model', $messageBag, ['old' => 'option']);
+
+        $newOptions = ['new' => 'options', 'count' => 3];
+        $input->setOptions($newOptions);
+
+        $this->assertSame($newOptions, $input->getOptions());
+    }
+
+    public function testSetOptionsReplacesAllOptions()
+    {
+        $messageBag = new MessageBag();
+        $input = new Input('model', $messageBag, ['a' => 1, 'b' => 2]);
+
+        $input->setOptions(['c' => 3]);
+
+        $options = $input->getOptions();
+        $this->assertArrayHasKey('c', $options);
+        $this->assertArrayNotHasKey('a', $options);
+        $this->assertArrayNotHasKey('b', $options);
+    }
+}

src/agent/tests/OutputTest.php

@@ -0,0 +1,146 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Agent\Tests;
+
+use PHPUnit\Framework\TestCase;
+use Symfony\AI\Agent\Output;
+use Symfony\AI\Platform\Message\Message;
+use Symfony\AI\Platform\Message\MessageBag;
+use Symfony\AI\Platform\Result\TextResult;
+
+final class OutputTest extends TestCase
+{
+    public function testConstructorSetsProperties()
+    {
+        $messageBag = new MessageBag();
+        $result = new TextResult('Test content');
+        $options = ['temperature' => 0.5];
+
+        $output = new Output('gpt-4', $result, $messageBag, $options);
+
+        $this->assertSame('gpt-4', $output->getModel());
+        $this->assertSame($result, $output->getResult());
+        $this->assertSame($messageBag, $output->getMessageBag());
+        $this->assertSame($options, $output->getOptions());
+    }
+
+    public function testConstructorWithDefaultOptions()
+    {
+        $messageBag = new MessageBag();
+        $result = new TextResult('Content');
+
+        $output = new Output('claude-3', $result, $messageBag);
+
+        $this->assertSame('claude-3', $output->getModel());
+        $this->assertSame($result, $output->getResult());
+        $this->assertSame($messageBag, $output->getMessageBag());
+        $this->assertSame([], $output->getOptions());
+    }
+
+    public function testGetModel()
+    {
+        $messageBag = new MessageBag();
+        $result = new TextResult('Test');
+
+        $output = new Output('test-model', $result, $messageBag);
+
+        $this->assertSame('test-model', $output->getModel());
+    }
+
+    public function testGetResult()
+    {
+        $messageBag = new MessageBag();
+        $result = new TextResult('Expected content');
+
+        $output = new Output('model', $result, $messageBag);
+
+        $retrievedResult = $output->getResult();
+
+        $this->assertSame($result, $retrievedResult);
+        $this->assertSame('Expected content', $retrievedResult->getContent());
+    }
+
+    public function testSetResult()
+    {
+        $messageBag = new MessageBag();
+        $originalResult = new TextResult('Original');
+        $output = new Output('model', $originalResult, $messageBag);
+
+        $newResult = new TextResult('New content');
+        $output->setResult($newResult);
+
+        $retrievedResult = $output->getResult();
+        $this->assertSame($newResult, $retrievedResult);
+        $this->assertSame('New content', $retrievedResult->getContent());
+    }
+
+    public function testGetMessageBag()
+    {
+        $messageBag = new MessageBag();
+        $messageBag->add(Message::ofUser('User message'));
+        $messageBag->add(Message::ofAssistant('Assistant reply'));
+
+        $result = new TextResult('Content');
+        $output = new Output('model', $result, $messageBag);
+
+        $retrievedMessageBag = $output->getMessageBag();
+
+        $this->assertSame($messageBag, $retrievedMessageBag);
+        $this->assertCount(2, $retrievedMessageBag);
+    }
+
+    public function testGetOptions()
+    {
+        $messageBag = new MessageBag();
+        $result = new TextResult('Content');
+        $options = ['max_tokens' => 500, 'temperature' => 0.8];
+
+        $output = new Output('model', $result, $messageBag, $options);
+
+        $this->assertSame($options, $output->getOptions());
+        $this->assertSame(500, $output->getOptions()['max_tokens']);
+        $this->assertSame(0.8, $output->getOptions()['temperature']);
+    }
+
+    public function testModelIsReadOnly()
+    {
+        $messageBag = new MessageBag();
+        $result = new TextResult('Content');
+
+        $output = new Output('original-model', $result, $messageBag);
+
+        $this->assertSame('original-model', $output->getModel());
+    }
+
+    public function testMessageBagIsReadOnly()
+    {
+        $messageBag = new MessageBag();
+        $messageBag->add(Message::ofUser('Test'));
+
+        $result = new TextResult('Content');
+        $output = new Output('model', $result, $messageBag);
+
+        $retrievedBag = $output->getMessageBag();
+        $this->assertSame($messageBag, $retrievedBag);
+    }
+
+    public function testOptionsIsReadOnly()
+    {
+        $messageBag = new MessageBag();
+        $result = new TextResult('Content');
+        $options = ['key' => 'value'];
+
+        $output = new Output('model', $result, $messageBag, $options);
+
+        $this->assertSame($options, $output->getOptions());
+    }
+}