fix(poetry): Set version inside the project section #45

Since poetry >= 2.0.0 it is possible to set the version under the `[project]` section.

Co-authored-by: lennard28 <[email protected]>

Author: abichinger

lib/py/set_version.py

@@ -17,7 +17,7 @@
 
 # modifiy version (tomlkit will preserve comments)
 toml = parse(pyproject)
-if 'tool' in toml and 'poetry' in toml['tool']:
+if 'tool' in toml and 'poetry' in toml['tool'] and "version" in toml["tool"]["poetry"]:
     print('Poetry package detected')
     toml['tool']['poetry']['version'] = args.version
 else:

test/prepare.test.ts

@@ -1,3 +1,5 @@
+import fs from 'fs';
+import TOML from 'smol-toml';
 import { describe, expect, test } from 'vitest';
 import {
   bDistPackage,
@@ -52,13 +54,65 @@ test('prepare: setVersionPy', async () => {
   );
 });
 
-test('prepare: setVersionToml', async () => {
-  const { config, context } = genPackage({
-    legacyInterface: false,
-  });
-  await expect(
-    setVersionToml(config.srcDir, '2.0.0', pipe(context)),
-  ).resolves.toBe(undefined);
+describe('prepare: setVersionToml', async () => {
+  const testCases = [
+    {
+      name: 'version in project',
+      section: 'project',
+      content: `[project]
+version = "0.0.0"`,
+    },
+    {
+      name: 'no version',
+      section: 'project',
+      content: `[project]
+name = "test"`,
+    },
+    {
+      name: 'poetry: version in tool.poetry',
+      section: 'tool.poetry',
+      content: `[tool.poetry]
+version = "0.0.0"`,
+    },
+    {
+      name: 'poetry: version in project',
+      section: 'project',
+      content: `[project]
+version = "0.0.0"
+
+[tool.poetry]
+name = "test"
+
+[tool.poetry.dependencies]
+python = "^3.7"`,
+    },
+  ];
+
+  for (const t of testCases) {
+    test(
+      t.name,
+      async () => {
+        const { config, context } = genPackage({
+          legacyInterface: false,
+          content: t.content,
+        });
+        await expect(
+          setVersionToml(config.srcDir, '1.0.0', pipe(context)),
+        ).resolves.toBe(undefined);
+
+        const toml = fs.readFileSync(config.pyprojectPath, {
+          encoding: 'utf8',
+        });
+        const pyproject: any = TOML.parse(toml);
+        const version = t.section
+          .split('.')
+          .reduce((acc, key) => acc[key], pyproject).version;
+
+        await expect(version).toBe('1.0.0');
+      },
+      60000,
+    );
+  }
 });
 
 test('prepare: versionCmd', async () => {