driver.phantomjs_test: Wait for load before testing snapshot

ArchiveTeam · Jan 4, 2015 · 4f91e12 · 4f91e12
1 parent 12d02e4
commit 4f91e12
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/wpull/driver/phantomjs_test.py b/wpull/driver/phantomjs_test.py
@@ -47,10 +47,6 @@ def load_finished_callback(message):
             yield From(driver.start())
             yield From(driver.open_page(self.get_url('/')))
             yield From(driver.scroll_to(0, 100))
-            yield From(driver.snapshot('asdf.png'))
-            yield From(driver.snapshot('asdf.pdf'))
-            yield From(driver.snapshot('asdf.html'))
-            page_url = yield From(driver.get_page_url())
 
             for dummy in range(50):
                 if not load_finished:
@@ -60,6 +56,11 @@ def load_finished_callback(message):
             else:
                 print('Load did not finish!')
 
+            yield From(driver.snapshot('asdf.png'))
+            yield From(driver.snapshot('asdf.pdf'))
+            yield From(driver.snapshot('asdf.html'))
+            page_url = yield From(driver.get_page_url())
+
             yield From(driver.close_page())
             driver.close()