Commit 839b925f authored by Cheryl Sabella's avatar Cheryl Sabella Committed by GitHub

bpo-35661: Fix failing test on buildbot (GH-12297)

parent 26c910c5
...@@ -110,18 +110,20 @@ class BasicTest(BaseTest): ...@@ -110,18 +110,20 @@ class BasicTest(BaseTest):
def test_prompt(self): def test_prompt(self):
env_name = os.path.split(self.env_dir)[1] env_name = os.path.split(self.env_dir)[1]
rmtree(self.env_dir)
builder = venv.EnvBuilder() builder = venv.EnvBuilder()
self.run_with_capture(builder.create, self.env_dir)
context = builder.ensure_directories(self.env_dir) context = builder.ensure_directories(self.env_dir)
self.assertEqual(context.prompt, '(%s) ' % env_name)
builder.create(self.env_dir)
data = self.get_text_file_contents('pyvenv.cfg') data = self.get_text_file_contents('pyvenv.cfg')
self.assertEqual(context.prompt, '(%s) ' % env_name)
self.assertNotIn("prompt = ", data) self.assertNotIn("prompt = ", data)
rmtree(self.env_dir)
builder = venv.EnvBuilder(prompt='My prompt') builder = venv.EnvBuilder(prompt='My prompt')
self.run_with_capture(builder.create, self.env_dir)
context = builder.ensure_directories(self.env_dir) context = builder.ensure_directories(self.env_dir)
self.assertEqual(context.prompt, '(My prompt) ')
builder.create(self.env_dir)
data = self.get_text_file_contents('pyvenv.cfg') data = self.get_text_file_contents('pyvenv.cfg')
self.assertEqual(context.prompt, '(My prompt) ')
self.assertIn("prompt = 'My prompt'\n", data) self.assertIn("prompt = 'My prompt'\n", data)
@skipInVenv @skipInVenv
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment