test_build_clib.py 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. import random
  2. from unittest import mock
  3. import pytest
  4. from setuptools.command.build_clib import build_clib
  5. from setuptools.dist import Distribution
  6. from distutils.errors import DistutilsSetupError
  7. class TestBuildCLib:
  8. @mock.patch('setuptools.command.build_clib.newer_pairwise_group')
  9. def test_build_libraries(self, mock_newer):
  10. dist = Distribution()
  11. cmd = build_clib(dist)
  12. # this will be a long section, just making sure all
  13. # exceptions are properly raised
  14. libs = [('example', {'sources': 'broken.c'})]
  15. with pytest.raises(DistutilsSetupError):
  16. cmd.build_libraries(libs)
  17. obj_deps = 'some_string'
  18. libs = [('example', {'sources': ['source.c'], 'obj_deps': obj_deps})]
  19. with pytest.raises(DistutilsSetupError):
  20. cmd.build_libraries(libs)
  21. obj_deps = {'': ''}
  22. libs = [('example', {'sources': ['source.c'], 'obj_deps': obj_deps})]
  23. with pytest.raises(DistutilsSetupError):
  24. cmd.build_libraries(libs)
  25. obj_deps = {'source.c': ''}
  26. libs = [('example', {'sources': ['source.c'], 'obj_deps': obj_deps})]
  27. with pytest.raises(DistutilsSetupError):
  28. cmd.build_libraries(libs)
  29. # with that out of the way, let's see if the crude dependency
  30. # system works
  31. cmd.compiler = mock.MagicMock(spec=cmd.compiler)
  32. mock_newer.return_value = ([], [])
  33. obj_deps = {'': ('global.h',), 'example.c': ('example.h',)}
  34. libs = [('example', {'sources': ['example.c'], 'obj_deps': obj_deps})]
  35. cmd.build_libraries(libs)
  36. assert [['example.c', 'global.h', 'example.h']] in mock_newer.call_args[0]
  37. assert not cmd.compiler.compile.called
  38. assert cmd.compiler.create_static_lib.call_count == 1
  39. # reset the call numbers so we can test again
  40. cmd.compiler.reset_mock()
  41. mock_newer.return_value = '' # anything as long as it's not ([],[])
  42. cmd.build_libraries(libs)
  43. assert cmd.compiler.compile.call_count == 1
  44. assert cmd.compiler.create_static_lib.call_count == 1
  45. @mock.patch('setuptools.command.build_clib.newer_pairwise_group')
  46. def test_build_libraries_reproducible(self, mock_newer):
  47. dist = Distribution()
  48. cmd = build_clib(dist)
  49. # with that out of the way, let's see if the crude dependency
  50. # system works
  51. cmd.compiler = mock.MagicMock(spec=cmd.compiler)
  52. mock_newer.return_value = ([], [])
  53. original_sources = ['a-example.c', 'example.c']
  54. sources = original_sources
  55. obj_deps = {'': ('global.h',), 'example.c': ('example.h',)}
  56. libs = [('example', {'sources': sources, 'obj_deps': obj_deps})]
  57. cmd.build_libraries(libs)
  58. computed_call_args = mock_newer.call_args[0]
  59. while sources == original_sources:
  60. sources = random.sample(original_sources, len(original_sources))
  61. libs = [('example', {'sources': sources, 'obj_deps': obj_deps})]
  62. cmd.build_libraries(libs)
  63. assert computed_call_args == mock_newer.call_args[0]