Add tests

Author: stevebauman

tests/Integration/MessagesTest.php

@@ -134,8 +134,60 @@ function folder(): Folder
     fn (MessageQuery $query) => $query->withBody(),
     fn (MessageQuery $query) => $query->withFlags(),
     fn (MessageQuery $query) => $query->withHeaders(),
+    fn (MessageQuery $query) => $query->withSize(),
 ]);
 
+test('get with size', function () {
+    $folder = folder();
+
+    $uid = $folder->messages()->append(
+        new DraftMessage(
+            from: '[email protected]',
+            to: '[email protected]',
+            subject: 'Test Subject',
+            text: 'hello world',
+        ),
+    );
+
+    // Fetch without size - should be null
+    $messagesWithoutSize = $folder->messages()->get();
+    expect($messagesWithoutSize->first()->size())->toBeNull();
+
+    // Fetch with size - should have a value
+    $messagesWithSize = $folder->messages()->withSize()->get();
+    $message = $messagesWithSize->first();
+
+    expect($message->size())->toBeInt();
+    expect($message->size())->toBeGreaterThan(0);
+    expect($message->uid())->toBe($uid);
+});
+
+test('size reflects actual message size', function () {
+    $folder = folder();
+
+    $shortMessage = new DraftMessage(
+        from: '[email protected]',
+        text: 'short',
+    );
+
+    $longMessage = new DraftMessage(
+        from: '[email protected]',
+        text: str_repeat('This is a longer message with more content. ', 100),
+    );
+
+    $uid1 = $folder->messages()->append($shortMessage);
+    $uid2 = $folder->messages()->append($longMessage);
+
+    $messages = $folder->messages()->withSize()->get();
+
+    $short = $messages->find($uid1);
+    $long = $messages->find($uid2);
+
+    expect($short->size())->toBeInt();
+    expect($long->size())->toBeInt();
+    expect($long->size())->toBeGreaterThan($short->size());
+});
+
 test('append', function () {
     $folder = folder();
 

tests/Unit/FileMessageTest.php

@@ -481,3 +481,18 @@
     // Different content
     expect($message1->is($message3))->toBeFalse();
 });
+
+test('it can determine size from contents', function () {
+    $contents = <<<'EOT'
+    From: "John Doe" <[email protected]>
+    Subject: Test Subject
+    Date: Wed, 19 Feb 2025 12:34:56 -0500
+    Content-Type: text/plain; charset="UTF-8"
+
+    Test content
+    EOT;
+
+    $message = new FileMessage($contents);
+
+    expect($message->size())->toBe(strlen($contents));
+});

tests/Unit/Testing/FakeMessageTest.php

@@ -118,3 +118,15 @@
     expect($message->isFlagged())->toBeFalse();
     expect($message->flags())->toBeEmpty();
 });
+
+test('it can get size when set', function () {
+    $message = new FakeMessage(1, [], 'Test content', 1024);
+
+    expect($message->size())->toBe(1024);
+});
+
+test('it returns size from contents when size is not set', function () {
+    $message = new FakeMessage(1, [], 'Test content');
+
+    expect($message->size())->toBe(strlen('Test content'));
+});