2 files changed, 53 insertions, 0 deletions
diff --git a/test/sign/test_sign_placement.vader b/test/sign/test_sign_placement.vader
index 707e2ce4..f8e926b6 100644
--- a/test/sign/test_sign_placement.vader
+++ b/test/sign/test_sign_placement.vader
@@ -71,6 +71,38 @@ After:
   call ale#linter#Reset()
   sign unplace *
 
+Execute(ale#sign#GetSignType should return the right sign types):
+  AssertEqual 'ALEErrorSign', ale#sign#GetSignType([{'type': 'E'}])
+  AssertEqual 'ALEStyleErrorSign', ale#sign#GetSignType([{'type': 'E', 'sub_type': 'style'}])
+  AssertEqual 'ALEWarningSign', ale#sign#GetSignType([{'type': 'W'}])
+  AssertEqual 'ALEStyleWarningSign', ale#sign#GetSignType([{'type': 'W', 'sub_type': 'style'}])
+  AssertEqual 'ALEInfoSign', ale#sign#GetSignType([{'type': 'I'}])
+  AssertEqual 'ALEErrorSign', ale#sign#GetSignType([
+  \ {'type': 'E'},
+  \ {'type': 'W'},
+  \ {'type': 'I'},
+  \ {'type': 'E', 'sub_type': 'style'},
+  \ {'type': 'W', 'sub_type': 'style'},
+  \])
+  AssertEqual 'ALEWarningSign', ale#sign#GetSignType([
+  \ {'type': 'W'},
+  \ {'type': 'I'},
+  \ {'type': 'E', 'sub_type': 'style'},
+  \ {'type': 'W', 'sub_type': 'style'},
+  \])
+  AssertEqual 'ALEInfoSign', ale#sign#GetSignType([
+  \ {'type': 'I'},
+  \ {'type': 'E', 'sub_type': 'style'},
+  \ {'type': 'W', 'sub_type': 'style'},
+  \])
+  AssertEqual 'ALEStyleErrorSign', ale#sign#GetSignType([
+  \ {'type': 'E', 'sub_type': 'style'},
+  \ {'type': 'W', 'sub_type': 'style'},
+  \])
+  AssertEqual 'ALEStyleWarningSign', ale#sign#GetSignType([
+  \ {'type': 'W', 'sub_type': 'style'},
+  \])
+
 Given testft(A file with warnings/errors):
   foo
   bar
diff --git a/test/test_loclist_corrections.vader b/test/test_loclist_corrections.vader
index 8e01dfbc..f424424d 100644
--- a/test/test_loclist_corrections.vader
+++ b/test/test_loclist_corrections.vader
@@ -164,3 +164,24 @@ Execute(FixLocList should pass on col_length values):
   \     {'text': 'a', 'lnum': '010', 'col': '011', 'end_col': 12},
   \   ],
   \ )
+
+Execute(FixLocList should allow subtypes to be set):
+  AssertEqual
+  \ [
+  \   {
+  \     'text': 'a',
+  \     'lnum': 10,
+  \     'col': 0,
+  \     'bufnr': bufnr('%'),
+  \     'vcol': 0,
+  \     'type': 'E',
+  \     'sub_type': 'style',
+  \     'nr': -1,
+  \     'linter_name': 'foobar',
+  \   },
+  \],
+  \ ale#engine#FixLocList(
+  \   bufnr('%'),
+  \   {'name': 'foobar'},
+  \   [{'text': 'a', 'lnum': 11, 'sub_type': 'style'}],
+  \ )