diff --git a/tests/framework/data/ActiveDataProviderTest.php b/tests/framework/data/ActiveDataProviderTest.php
index 789a0337260..da3e70df945 100644
--- a/tests/framework/data/ActiveDataProviderTest.php
+++ b/tests/framework/data/ActiveDataProviderTest.php
@@ -197,4 +197,23 @@ public function testDoesNotPerformQueryWhenHasNoModels()
 
         $this->assertEquals(0, $pagination->getPageCount());
     }
+
+    public function testTotalCountAfterSearch()
+    {
+        $query = Order::find();
+        $provider = new ActiveDataProvider([
+            'query' => $query,
+            'pagination' => [
+                'pageSize' => 2,
+            ],
+        ]);
+
+        $pagination = $provider->getPagination();
+        $this->assertEquals(2, $pagination->getPageCount());
+        $this->assertEquals(3, $pagination->getTotalCount());
+
+        $query->andWhere(['customer_id' => 2]);
+        $this->assertEquals(1, $pagination->getPageCount());
+        $this->assertEquals(2, $pagination->getTotalCount());
+    }
 }