tools/lint/test/test_black.py

   1 # -*- coding: utf-8 -*-
   2
   3 # This Source Code Form is subject to the terms of the Mozilla Public
   4 # License, v. 2.0. If a copy of the MPL was not distributed with this
   5 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   6
   7 import mozunit
   8
   9 LINTER = "black"
  10 fixed = 0
  11
  12
  13 def test_lint_fix(lint, create_temp_file):
  14     contents = """def is_unique(
  15                s
  16                ):
  17     s = list(s
  18                 )
  19     s.sort()
  20
  21
  22     for i in range(len(s) - 1):
  23         if s[i] == s[i + 1]:
  24             return 0
  25     else:
  26         return 1
  27
  28
  29 if __name__ == "__main__":
  30     print(
  31           is_unique(input())
  32          ) """
  33
  34     path = create_temp_file(contents, "bad.py")
  35     lint([path], fix=True)
  36     assert fixed == 1
  37
  38
  39 def test_lint_black(lint, paths):
  40     results = lint(paths())
  41     assert len(results) == 2
  42
  43     assert results[0].level == "error"
  44     assert results[0].relpath == "bad.py"
  45
  46     assert "EOF" in results[1].message
  47     assert results[1].level == "error"
  48     assert results[1].relpath == "invalid.py"
  49
  50
  51 if __name__ == "__main__":
  52     mozunit.main()