adds test for new method

matomo-org · Jul 16, 2024 · 7806cd6 · 7806cd6
1 parent b2d72b9
commit 7806cd6
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/PHPUnit/Integration/DataAccess/ModelTest.php b/tests/PHPUnit/Integration/DataAccess/ModelTest.php
@@ -125,6 +125,27 @@ private function assertAllocatedArchiveId($expectedId)
         $this->assertEquals($expectedId, $id);
     }
 
+    public function testGetAndUpdateArchiveStatus()
+    {
+        $this->insertArchiveData([
+            ['date1' => '2020-02-03', 'date2' => '2020-02-03', 'period' => 1, 'name' => 'done', 'value' => ArchiveWriter::DONE_ERROR],
+        ]);
+
+        $numericTable = ArchiveTableCreator::getNumericTable(Date::factory('2020-02-03'));
+
+        self::assertEquals(
+            ArchiveWriter::DONE_ERROR,
+            $this->model->getArchiveStatus($numericTable, '1', 'done')
+        );
+
+        $this->model->updateArchiveStatus($numericTable, '1', 'done', ArchiveWriter::DONE_ERROR_INVALIDATED);
+
+        self::assertEquals(
+            ArchiveWriter::DONE_ERROR_INVALIDATED,
+            $this->model->getArchiveStatus($numericTable, '1', 'done')
+        );
+    }
+
     /**
      * @dataProvider getTestDataForHasChildArchivesInPeriod
      */