def test_pyproject_toml_file_not_exists(self):
fake_path_class = MagicMock(spec=Path)
fake_path = fake_path_class.return_value
fake_path.__str__.return_value = 'pyproject.toml'
fake_path.exists.return_value = False
with self.assertRaisesRegex(VersionError,
'pyproject.toml file not found'):
get_version_from_pyproject_toml(fake_path)
fake_path.exists.assert_called_with()
def test_empty_poerty_section(self):
fake_path_class = MagicMock(spec=Path)
fake_path = fake_path_class.return_value
fake_path.__str__.return_value = 'pyproject.toml'
fake_path.exists.return_value = True
fake_path.read_text.return_value = '[tool.poetry]'
with self.assertRaisesRegex(
VersionError,
'Version information not found in pyproject.toml file'):
get_version_from_pyproject_toml(fake_path)
fake_path.exists.assert_called_with()
fake_path.read_text.assert_called_with()
def test_get_version(self):
fake_path_class = MagicMock(spec=Path)
fake_path = fake_path_class.return_value
fake_path.__str__.return_value = 'pyproject.toml'
fake_path.exists.return_value = True
fake_path.read_text.return_value = '[tool.poetry]\nversion = "1.2.3"'
version = get_version_from_pyproject_toml(fake_path)
self.assertEqual(version, '1.2.3')
fake_path.exists.assert_called_with()
fake_path.read_text.assert_called_with()