Cache cleanup

src/Statements/Statement.php

@@ -21,6 +21,10 @@ abstract class Statement implements IteratorAggregate
 {
     use LinkedDataFrame;
     protected GroupByIterator|CacheStatus $cache = CacheStatus::UNUSED;
+    public CacheStatus $cacheStatus {
+        get => $this->cache instanceof CacheStatus ? $this->cache : CacheStatus::SET;
+    }
+
     protected array $where = [];
     protected ?int $limit = null;
     protected int $offset = 0;
@@ -30,15 +34,6 @@ public function __construct(DataFrame $df)
         $this->setLinkedDataFrame($df);
     }
 
-    public function getCacheStatus(): CacheStatus
-    {
-        if ($this->cache instanceof CacheStatus) {
-            return $this->cache;
-        }
-
-        return CacheStatus::SET;
-    }
-
     protected function invalidateCache(): void
     {
         $this->cache = CacheStatus::UNUSED;

tests/Unit/StatementUnitTest.php

@@ -3,6 +3,7 @@
 declare(strict_types=1);
 
 use MammothPHP\WoollyM\DataFrame;
+use MammothPHP\WoollyM\Statements\CacheStatus;
 
 beforeEach(function (): void {
     $this->df = DataFrame::fromArray([
@@ -16,3 +17,24 @@
     expect(\count($this->df->select()))->toBe(3);
     expect($this->df->select())->toHaveCount(3);
 });
+
+it('has a cache status', function(): void {
+    $stmt = $this->df->select('colA')->whereColumn('colA', equal: 1)->groupBy('colA');
+
+    expect($stmt->cacheStatus)->toBe(CacheStatus::UNUSED);
+
+    $stmt->export();
+    expect($stmt->cacheStatus)->toBe(CacheStatus::SET);
+    expect($stmt)->toHaveCount(1);
+
+    $stmt->resetWhere();
+    expect($stmt->cacheStatus)->toBe(CacheStatus::UNUSED);
+    $stmt->whereKeyBetween(0,2);
+
+    expect($stmt->cacheStatus)->toBe(CacheStatus::UNUSED);
+
+    expect($stmt)->toHaveCount(3);
+    expect($stmt->cacheStatus)->toBe(CacheStatus::SET);
+
+    expect(fn() => $stmt->cacheStatus = CacheStatus::SET)->toThrow('$cacheStatus is read-only');
+});