1 files changed, 115 insertions, 0 deletions
diff --git a/gcc/testsuite/gcc.dg/plugin/diagnostic-test-show-locus.py b/gcc/testsuite/gcc.dg/plugin/diagnostic-test-show-locus.py
new file mode 100644
index 0000000..aca1b6c
--- /dev/null
+++ b/gcc/testsuite/gcc.dg/plugin/diagnostic-test-show-locus.py
@@ -0,0 +1,115 @@
+# Verify that diagnostic-show-locus.cc works with HTML output.
+
+from htmltest import *
+
+import pytest
+
+@pytest.fixture(scope='function', autouse=True)
+def html_tree():
+    return html_tree_from_env()
+
+#def get_tr_within_thead(thead, idx)
+
+def get_ruler_text(thead, idx):
+    trs = thead.findall('xhtml:tr', ns)
+    tr = trs[idx]
+    tds = tr.findall('xhtml:td', ns)
+    assert len(tds) == 3
+    assert_class(tds[2], 'ruler')
+    return tds[2].text
+
+def test_very_wide_line(html_tree):
+    diag = get_diag_by_index(html_tree, 2)
+    src = get_locus_within_diag(diag)
+
+    # Check ruler
+    thead = src.find('xhtml:thead', ns)
+    assert_class(thead, 'ruler')
+    trs = thead.findall('xhtml:tr', ns)
+    assert len(trs) == 3
+
+    assert get_ruler_text(thead, 0) == '       0         0         0         0         0         1         1   '
+    assert get_ruler_text(thead, 1) == '       5         6         7         8         9         0         1   '
+    assert get_ruler_text(thead, 2) == '34567890123456789012345678901234567890123456789012345678901234567890123'
+
+    # Check quoted source
+    tbody = src.find('xhtml:tbody', ns)
+    assert_class(tbody, 'line-span')
+    trs = tbody.findall('xhtml:tr', ns)
+    assert len(trs) == 5
+    assert_quoted_line(trs[0], '   43', '                                      float f = foo * bar; /* { dg-warning "95: test" } */')
+    assert_annotation_line(trs[1],      '                                                ~~~~^~~~~')
+    assert_annotation_line(trs[2],      '                                                    |')
+    assert_annotation_line(trs[3],      '                                                    label 0')
+    assert_annotation_line(trs[4],      '                                                bar * foo')
+
+def test_fixit_insert(html_tree):
+    diag = get_diag_by_index(html_tree, 3)
+    msg = get_message_within_diag(diag)
+    assert msg[0].text == 'warning: '
+    assert msg[0].tail == ' example of insertion hints'
+
+    src = get_locus_within_diag(diag)
+
+    # Check quoted source
+    tbody = src.find('xhtml:tbody', ns)
+    assert_class(tbody, 'line-span')
+    trs = tbody.findall('xhtml:tr', ns)
+    assert len(trs) == 3
+    assert_quoted_line(trs[0], '   63', '   int a[2][2] = { 0, 1 , 2, 3 }; /* { dg-warning "insertion hints" } */')
+    assert_annotation_line(trs[1],      '                   ^~~~')
+    assert_annotation_line(trs[2],      '                   {   }')
+
+def test_fixit_remove(html_tree):
+    diag = get_diag_by_index(html_tree, 4)
+    msg = get_message_within_diag(diag)
+    assert msg[0].text == 'warning: '
+    assert msg[0].tail == ' example of a removal hint'
+
+    src = get_locus_within_diag(diag)
+
+    # Check quoted source
+    tbody = src.find('xhtml:tbody', ns)
+    assert_class(tbody, 'line-span')
+    trs = tbody.findall('xhtml:tr', ns)
+    assert len(trs) == 3
+    assert_quoted_line(trs[0], '   77', '  int a;; /* { dg-warning "example of a removal hint" } */')
+    assert_annotation_line(trs[1],      '        ^')
+    assert_annotation_line(trs[2],      '        -')
+
+def test_fixit_replace(html_tree):
+    diag = get_diag_by_index(html_tree, 5)
+    msg = get_message_within_diag(diag)
+    assert msg[0].text == 'warning: '
+    assert msg[0].tail == ' example of a replacement hint'
+
+    src = get_locus_within_diag(diag)
+
+    # Check quoted source
+    tbody = src.find('xhtml:tbody', ns)
+    assert_class(tbody, 'line-span')
+    trs = tbody.findall('xhtml:tr', ns)
+    assert len(trs) == 3
+    assert_quoted_line(trs[0], '   91', '  gtk_widget_showall (dlg); /* { dg-warning "example of a replacement hint" } */')
+    assert_annotation_line(trs[1],      '  ^~~~~~~~~~~~~~~~~~')
+    assert_annotation_line(trs[2],      '  gtk_widget_show_all')
+
+def test_fixit_insert_newline(html_tree):
+    diag = get_diag_by_index(html_tree, 6)
+    msg = get_message_within_diag(diag)
+    assert msg[0].text == 'warning: '
+    assert msg[0].tail == ' example of newline insertion hint'
+
+    src = get_locus_within_diag(diag)
+
+    # Check quoted source
+    tbody = src.find('xhtml:tbody', ns)
+    assert_class(tbody, 'line-span')
+    trs = tbody.findall('xhtml:tr', ns)
+    assert len(trs) == 4
+    assert_quoted_line(trs[0], '  109', '      x = a;')
+    assert_annotation_line(trs[1],      '      break;',
+                           expected_line_num='  +++',
+                           expected_left_margin='+')
+    assert_quoted_line(trs[2], '  110', "    case 'b':  /* { dg-warning \"newline insertion\" } */")
+    assert_annotation_line(trs[3],      '    ^~~~~~~~')