lib/calabash/page.rb in calabash-2.0.0.pre11 vs lib/calabash/page.rb in calabash-2.0.0.prelegacy

@@ -6,10 +6,13 @@
   #
   # We have a great examples of using the POM in the Calabash 2.0 repository.
   #   * https://github.com/calabash/calabash/tree/develop/samples/wordpress
   #   * https://github.com/calabash/calabash/tree/develop/samples/shared-page-logic
   class Page
+    # For auto-completion
+    include Calabash
+
     # @!visibility private
     def self.inherited(subclass)
       # Define the page into global scope
       full_name = subclass.name
 
@@ -66,10 +69,10 @@
       raise 'Implement your own trait'
     end
 
     # Waits for the page trait to appear.
     def await(options={})
-      cal.wait_for_view(trait, options)
+      wait_for_view(trait, options)
     end
 
     # @!visibility private
     class StubPage