1 files changed, 35 insertions, 0 deletions
diff --git a/files/puppet/modules/pixelated/files/functional-tests/page_objects/notification.py b/files/puppet/modules/pixelated/files/functional-tests/page_objects/notification.py
new file mode 100644
index 0000000..35042dd
--- /dev/null
+++ b/files/puppet/modules/pixelated/files/functional-tests/page_objects/notification.py
@@ -0,0 +1,35 @@
+#
+# Copyright (c) 2015 ThoughtWorks, Inc.
+#
+# Pixelated is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pixelated is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
+
+from base_page_object import BasePageObject
+from selenium.webdriver.common.by import By
+
+class Notification(BasePageObject):
+    def __init__(self, context, timeout=10):
+        self._locators = {
+            'message_sent': '#user-alerts:contains("Your message was sent!")',
+            'message_deleted': '#user-alerts:contains("Your message was moved to trash!")'
+        }
+        super(Notification, self).__init__(context, timeout)
+
+    def wait_for_notification(self, notification):
+        print "Waiting for message sent notification"
+        self._find_element_by_css_locator(self._locators[notification])
+        print "Notification found"
+        return self
+
+    def wait_until_notification_is_gone(self):
+        self.wait_until_element_is_invisible_by_locator(By.ID, 'user-alerts')
+\ No newline at end of file