test running startup file

thomasballinger · thomasballinger · commit 3fb84419ba42 · 2015-02-07T01:53:57.000-05:00
diff --git a/bpython/test/__init__.py b/bpython/test/__init__.py
@@ -12,6 +12,7 @@
 
 from bpython.translations import init
 from bpython._py3compat import py3
+from six.moves import builtins
 
 
 class FixLanguageTestCase(unittest.TestCase):
@@ -27,3 +28,8 @@ class MagicIterMock(mock.MagicMock):
         __next__ = mock.Mock(return_value=None)
     else:
         next = mock.Mock(return_value=None)
+
+
+def builtin_target(obj):
+    """Returns mock target string of a builtin"""
+    return '%s.%s' % (builtins.__name__, obj.__name__)
diff --git a/bpython/test/test_curtsies_repl.py b/bpython/test/test_curtsies_repl.py
@@ -15,11 +15,13 @@
 
 from bpython.curtsiesfrontend import repl as curtsiesrepl
 from bpython.curtsiesfrontend import interpreter
+from bpython.curtsiesfrontend import events as bpythonevents
 from bpython import autocomplete
 from bpython import config
 from bpython import args
 from bpython._py3compat import py3
-from bpython.test import FixLanguageTestCase as TestCase, MagicIterMock, mock
+from bpython.test import (FixLanguageTestCase as TestCase, MagicIterMock, mock,
+                          builtin_target)
 
 
 def setup_config(conf):
@@ -307,5 +309,20 @@ def test_show_source_formatted(self):
         self.repl.show_source()
 
 
+class TestCurtsiesStartup(TestCase):
+
+    def setUp(self):
+        self.repl = create_repl()
+        os.environ['PYTHONSTARTUP'] = 'file'
+
+    def tearDown(self):
+        del os.environ['PYTHONSTARTUP']
+
+    @mock.patch(builtin_target(open), mock.mock_open(read_data='a = 1\n'))
+    def test_startup_event(self):
+        self.repl.process_event(bpythonevents.RunStartupFileEvent())
+        self.assertIn('a', self.repl.interp.locals)
+
+
 if __name__ == '__main__':
     unittest.main()
