🔬 test #131

Author: chfw

templates/setup.py.jj2

@@ -24,19 +24,22 @@
   {% endif %}
 {% endfor %}
 {% block header %}
-# Template by pypi-mobans
+'''
+Template by pypi-mobans
+'''
+
 {% endblock %}
+import os
+import sys
 import codecs
 import locale
-import os
 import platform
-import sys
 from shutil import rmtree
 
 {% if external_module_library %}
 from distutils.core import setup, Extension
 {% else %}
-from setuptools import Command, find_packages, setup
+from setuptools import Command, setup, find_packages
 {%endif%}
 {% if marker_dependencies and setup_use_markers and setup_use_markers_fix %}
 from setuptools import __version__ as setuptools_version

tests/fixtures/setup.py.output

@@ -1,14 +1,17 @@
 #!/usr/bin/env python3
 
-# Template by pypi-mobans
+'''
+Template by pypi-mobans
+'''
+
+import os
+import sys
 import codecs
 import locale
-import os
 import platform
-import sys
 from shutil import rmtree
 
-from setuptools import Command, find_packages, setup
+from setuptools import Command, setup, find_packages
 
 PY2 = sys.version_info[0] == 2
 PY26 = PY2 and sys.version_info[1] < 7
@@ -63,8 +66,7 @@ INSTALL_REQUIRES = [
 ]
 SETUP_COMMANDS = {}
 
-
-PACKAGES = find_packages(exclude=["ez_setup", "examples", "tests"])
+PACKAGES = find_packages(exclude=["ez_setup", "examples", "tests", "tests.*"])
 EXTRAS_REQUIRE = {}
 # You do not need to read beyond this line
 PUBLISH_COMMAND = "{0} setup.py sdist bdist_wheel upload -r pypi".format(sys.executable)

tests/test_setup.py

@@ -11,7 +11,29 @@ def test_setup():
         content = f.read()
     yaml = YAML(typ='safe')
     context = yaml.load(content)
-    print(context)
     rendered = get_rendered_file(filename, context)
     expected = get_file_content('setup.py.output')
     assert expected == rendered
+
+
+def test_setup_use_markers_true():
+    filename = 'setup.py.jj2'
+    with open(os.path.join("tests", "fixtures", "server_use_marker_true.yml"))as f:
+        content = f.read()
+    yaml = YAML(typ='safe')
+    context = yaml.load(content)
+    rendered = get_rendered_file(filename, context)
+    print(rendered)    
+    expected = get_file_content('setup_use_markers_true.py.output')
+    assert expected == rendered
+
+
+def test_setup_use_markers_false():
+    filename = 'setup.py.jj2'
+    with open(os.path.join("tests", "fixtures", "server_use_marker_false.yml"))as f:
+        content = f.read()
+    yaml = YAML(typ='safe')
+    context = yaml.load(content)
+    rendered = get_rendered_file(filename, context)
+    expected = get_file_content('setup_use_markers_false.py.output')
+    assert expected == rendered