Update locator in test_gallery.py

malini · Nov 23, 2012 · 25d59fe · 25d59fe
1 parent c31a3ac
commit 25d59fe
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/gaiatest/tests/test_gallery.py b/gaiatest/tests/test_gallery.py
@@ -10,7 +10,7 @@ class TestGallery(GaiaTestCase):
 
     _throbber_locator = ('id', 'throbber')
     _gallery_items_locator = ('css selector', 'li.thumbnail')
-    _current_photo = ('css selector', 'div.currentPhoto img[src]')
+    _current_image_locator = ('css selector', '#frame2 > img')
     _photos_toolbar_locator = ('id', 'photos-toolbar')
 
     def setUp(self):
@@ -29,9 +29,8 @@ def test_gallery_view(self):
 
         self.marionette.find_elements(*self._gallery_items_locator)[0].click()
 
-        self.wait_for_element_present(*self._current_photo)
-        self.assertTrue(
-            self.marionette.find_element(*self._current_photo).is_displayed())
+        current_image = self.wait_for_element_present(*self._current_image_locator)
+        self.assertTrue(current_image.is_displayed())
 
         # TODO
         # Add steps to view picture full screen