From 20fe7acb8247c48453b50f88b88d325baa47bf8a Mon Sep 17 00:00:00 2001
From: Dave Hunt <dave.hunt@gmail.com>
Date: Wed, 30 Apr 2014 14:37:58 +0100
Subject: [PATCH] Bug 999189 - Improve wait for Marketplace to be loaded.
 r=wlachance

---
 src/eideticker/eideticker/b2gtestmixins.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/eideticker/eideticker/b2gtestmixins.py b/src/eideticker/eideticker/b2gtestmixins.py
index e2da5bb..8f68dfb 100644
--- a/src/eideticker/eideticker/b2gtestmixins.py
+++ b/src/eideticker/eideticker/b2gtestmixins.py
@@ -52,9 +52,9 @@ class B2GMarketplaceTestMixin(object):
             expected.element_present(
                 By.CSS_SELECTOR, 'iframe[src*="marketplace"]'))
         self.device.marionette.switch_to_frame(iframe)
-        Wait(self.device.marionette).until(expected.element_displayed(
-            Wait(self.device.marionette).until(expected.element_present(
-                By.ID, 'site-header'))))
+        Wait(self.device.marionette).until(
+            lambda m: 'loaded' in m.find_element(
+                By.TAG_NAME, 'body').get_attribute('class').split())
 
     def prepare_app(self):
         self.launch_app()