1 files changed, 16 insertions, 4 deletions
diff --git a/android_webview/browser/browser_view_renderer_unittest.cc b/android_webview/browser/browser_view_renderer_unittest.cc
index 6ecf99a..0c2786b 100644
--- a/android_webview/browser/browser_view_renderer_unittest.cc
+++ b/android_webview/browser/browser_view_renderer_unittest.cc
@@ -2,13 +2,25 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+#include "android_webview/browser/browser_view_renderer.h"
 #include "android_webview/browser/test/rendering_test.h"
 
 namespace android_webview {
 
-TEST_F(RenderingTest, SmokeTest) {
-  SetUpTestHarness();
-  RunTest();
-}
+class SmokeTest : public RenderingTest {
+  void StartTest() override {
+    browser_view_renderer_->SetContinuousInvalidate(true);
+  }
+
+  void WillOnDraw() override {
+    browser_view_renderer_->SetContinuousInvalidate(false);
+  }
+
+  void DidDrawOnRT(SharedRendererState* functor) override {
+    EndTest();
+  }
+};
+
+RENDERING_TEST_F(SmokeTest);
 
 }  // namespace android_webview