From eeff29d16c74b4dacfdc2df04d388a5c32aed8b1 Mon Sep 17 00:00:00 2001
From: LRLJoe <joe@lowerrocklabs.com>
Date: Wed, 20 Nov 2024 02:38:35 +0000
Subject: [PATCH] Add ColorColumnTests

---
 tests/Unit/Views/Columns/ColorColumnTest.php | 29 ++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/tests/Unit/Views/Columns/ColorColumnTest.php b/tests/Unit/Views/Columns/ColorColumnTest.php
index 5455c9e76..8700ad190 100644
--- a/tests/Unit/Views/Columns/ColorColumnTest.php
+++ b/tests/Unit/Views/Columns/ColorColumnTest.php
@@ -24,6 +24,7 @@ public function test_can_get_the_column_view(): void
 
     }
 
+    
     public function test_can_infer_field_name_from_title_if_no_from(): void
     {
         $column = ColorColumn::make('Favorite Color');
@@ -176,4 +177,32 @@ function ($row) {
         $this->assertSame('#008000', $column->getColor($rows->slice(2, 1)->first()));
 
     }
+
+    public function test_can_get_column_get_contents_from_color(): void
+    {
+        $column = ColorColumn::make('Species Color')->color(
+            function ($row) {
+                if ($row->species_id == 1) {
+                    return '#ff0000';
+                } elseif ($row->species_id == 2) {
+                    return '#008000';
+                } else {
+                    return '#ffa500';
+                }
+
+            }
+        );
+        $rows = $this->basicTable->setAdditionalSelects(['pets.species_id as species_id'])->getRows();
+
+        $contents1 = $column->getContents($rows->first());
+        $this->assertSame($contents1['color'], '#ff0000');
+        $this->assertSame($contents1['isTailwind'], true);
+        $this->assertSame($contents1['isBootstrap'], false);
+
+        $contents2 = $column->getContents($rows->last());
+        $this->assertSame($contents2['color'], '#ffa500');
+        $this->assertSame($contents2['isTailwind'], true);
+        $this->assertSame($contents2['isBootstrap'], false);
+
+    }
 }