1 files changed, 32 insertions, 0 deletions
diff --git a/tests/test_tomlify.py b/tests/test_tomlify.py
new file mode 100644
index 0000000..65e6178
--- /dev/null
+++ b/tests/test_tomlify.py
@@ -0,0 +1,32 @@
+import os
+from pathlib import Path
+try:
+    import tomllib
+except ImportError:
+    import tomli as tomllib
+from shutil import copy
+from testpath import assert_isfile
+
+from flit import tomlify
+
+samples_dir = Path(__file__).parent / 'samples'
+
+def test_tomlify(copy_sample, monkeypatch):
+    td = copy_sample('with_flit_ini')
+    monkeypatch.chdir(td)
+
+    tomlify.main(argv=[])
+
+    pyproject_toml = (td / 'pyproject.toml')
+    assert_isfile(pyproject_toml)
+
+    with pyproject_toml.open('rb') as f:
+        content = tomllib.load(f)
+
+    assert 'build-system' in content
+    assert 'tool' in content
+    assert 'flit' in content['tool']
+    flit_table = content['tool']['flit']
+    assert 'metadata' in flit_table
+    assert 'scripts' in flit_table
+    assert 'entrypoints' in flit_table