From 5bac6b117c8dfd56d66e663a87434353faa05727 Mon Sep 17 00:00:00 2001
From: Christopher Hertel <mail@christopher-hertel.de>
Date: Sun, 29 Sep 2024 02:16:27 +0200
Subject: [PATCH] fix: makefile target and UsesClass attribute in tests

---
 Makefile                         |  2 ++
 tests/Message/MessageBagTest.php |  7 +++++++
 tests/Message/MessageTest.php    | 10 ++++++++++
 3 files changed, 19 insertions(+)

diff --git a/Makefile b/Makefile
index e5df7d18..07dfc271 100644
--- a/Makefile
+++ b/Makefile
@@ -1,3 +1,5 @@
+.PHONY: qa qa-lowest coverage run-all-examples
+
 qa:
 	composer update --prefer-stable
 	vendor/bin/php-cs-fixer fix --diff --verbose
diff --git a/tests/Message/MessageBagTest.php b/tests/Message/MessageBagTest.php
index 4348c371..9274ddc9 100644
--- a/tests/Message/MessageBagTest.php
+++ b/tests/Message/MessageBagTest.php
@@ -6,9 +6,11 @@
 
 use PhpLlm\LlmChain\Message\AssistantMessage;
 use PhpLlm\LlmChain\Message\Content\Image;
+use PhpLlm\LlmChain\Message\Content\Text;
 use PhpLlm\LlmChain\Message\Message;
 use PhpLlm\LlmChain\Message\MessageBag;
 use PhpLlm\LlmChain\Message\SystemMessage;
+use PhpLlm\LlmChain\Message\UserMessage;
 use PHPUnit\Framework\Attributes\CoversClass;
 use PHPUnit\Framework\Attributes\Small;
 use PHPUnit\Framework\Attributes\Test;
@@ -17,6 +19,11 @@
 
 #[CoversClass(MessageBag::class)]
 #[UsesClass(Message::class)]
+#[UsesClass(UserMessage::class)]
+#[UsesClass(SystemMessage::class)]
+#[UsesClass(AssistantMessage::class)]
+#[UsesClass(Image::class)]
+#[UsesClass(Text::class)]
 #[Small]
 final class MessageBagTest extends TestCase
 {
diff --git a/tests/Message/MessageTest.php b/tests/Message/MessageTest.php
index 7644b814..8dfc8e51 100644
--- a/tests/Message/MessageTest.php
+++ b/tests/Message/MessageTest.php
@@ -4,10 +4,14 @@
 
 namespace PhpLlm\LlmChain\Tests\Message;
 
+use PhpLlm\LlmChain\Message\AssistantMessage;
 use PhpLlm\LlmChain\Message\Content\Image;
 use PhpLlm\LlmChain\Message\Content\Text;
 use PhpLlm\LlmChain\Message\Message;
 use PhpLlm\LlmChain\Message\Role;
+use PhpLlm\LlmChain\Message\SystemMessage;
+use PhpLlm\LlmChain\Message\ToolCallMessage;
+use PhpLlm\LlmChain\Message\UserMessage;
 use PhpLlm\LlmChain\Response\ToolCall;
 use PHPUnit\Framework\Attributes\CoversClass;
 use PHPUnit\Framework\Attributes\Small;
@@ -16,8 +20,14 @@
 use PHPUnit\Framework\TestCase;
 
 #[CoversClass(Message::class)]
+#[UsesClass(UserMessage::class)]
+#[UsesClass(SystemMessage::class)]
+#[UsesClass(AssistantMessage::class)]
+#[UsesClass(ToolCallMessage::class)]
 #[UsesClass(Role::class)]
 #[UsesClass(ToolCall::class)]
+#[UsesClass(Image::class)]
+#[UsesClass(Text::class)]
 #[Small]
 final class MessageTest extends TestCase
 {