test/test_shell.rb in rib-1.2.4 vs test/test_shell.rb in rib-1.2.5

@@ -103,12 +103,18 @@
   should 'not pollute main' do
     shell.eval_input('main').first.should.eq 'rib'
   end
 
   should 'warn on removing main' do
-    TOPLEVEL_BINDING.singleton_class.module_eval do
-      def main; end
-    end
+    TOPLEVEL_BINDING.eval <<-RUBY
+      singleton_class.module_eval do
+        def main; end
+      end
+    RUBY
     mock(Rib).warn(is_a(String)){}
     shell.eval_input('main').first.should.eq 'rib'
+  end
+
+  should 'be main' do
+    shell.eval_input('self.inspect').first.should.eq 'main'
   end
 end