Fix some tests.

Several tests have incorrect/misplaced `wait` clauses. Fix them.

Author: reznikmm

testsuite/ada_lsp/T713-012.completion.generic_package/test.json

@@ -641,7 +641,15 @@
                }
             }
          },
-         "wait": []
+         "wait": [
+            {
+               "method": "textDocument/publishDiagnostics",
+               "params": {
+                  "uri": "$URI{src/inst.ads}",
+                  "diagnostics": []
+               }
+            }
+         ]
       }
    },
    {
@@ -652,13 +660,6 @@
             "method": "shutdown"
          },
          "wait": [
-            {
-               "method": "textDocument/publishDiagnostics",
-               "params": {
-                  "uri": "$URI{src/inst.ads}",
-                  "diagnostics": []
-               }
-            },
             {
                "id": 51,
                "result": null

testsuite/ada_lsp/V628-024.symbol_diff/test.json

@@ -226,11 +226,19 @@
             "method": "textDocument/didClose",
             "params": {
                "textDocument": {
-                  "uri": "file:///home/ogorodnik/adacore/tests/hello/src/main.adb"
+                  "uri": "$URI{main.adb}"
                }
             }
          },
-         "wait": []
+         "wait": [
+            {
+               "method": "textDocument/publishDiagnostics",
+               "params": {
+                  "uri": "$URI{main.adb}",
+                  "diagnostics": []
+               }
+            }
+         ]
       }
    },
    {
@@ -241,13 +249,6 @@
             "method": "shutdown"
          },
          "wait": [
-            {
-               "method": "textDocument/publishDiagnostics",
-               "params": {
-                  "uri": "file:///home/ogorodnik/adacore/tests/hello/src/main.adb",
-                  "diagnostics": []
-               }
-            },
             {
                "id": 9,
                "result": null

testsuite/ada_lsp/completion.snippet.formatting/test.json

@@ -399,7 +399,15 @@
                }
             }
          },
-         "wait": []
+         "wait": [
+            {
+               "method": "textDocument/publishDiagnostics",
+               "params": {
+                  "uri": "$URI{foo.adb}",
+                  "diagnostics": []
+               }
+            }
+         ]
       }
    },
    {
@@ -410,13 +418,6 @@
             "method": "shutdown"
          },
          "wait": [
-            {
-               "method": "textDocument/publishDiagnostics",
-               "params": {
-                  "uri": "$URI{foo.adb}",
-                  "diagnostics": []
-               }
-            },
             {
                "id": 53,
                "result": null

testsuite/ada_lsp/completion.snippet.formatting2/test.json

@@ -416,7 +416,15 @@
                }
             }
          },
-         "wait": []
+         "wait": [
+            {
+               "method": "textDocument/publishDiagnostics",
+               "params": {
+                  "uri": "$URI{foo.adb}",
+                  "diagnostics": []
+               }
+            }
+         ]
       }
    },
    {
@@ -427,13 +435,6 @@
             "method": "shutdown"
          },
          "wait": [
-            {
-               "method": "textDocument/publishDiagnostics",
-               "params": {
-                  "uri": "$URI{foo.adb}",
-                  "diagnostics": []
-               }
-            },
             {
                "id": 31,
                "result": null

testsuite/ada_lsp/completion.snippet.formatting3/test.json

@@ -357,7 +357,15 @@
                }
             }
          },
-         "wait": []
+         "wait": [
+            {
+               "method": "textDocument/publishDiagnostics",
+               "params": {
+                  "uri": "$URI{foo.adb}",
+                  "diagnostics": []
+               }
+            }
+         ]
       }
    },
    {
@@ -368,13 +376,6 @@
             "method": "shutdown"
          },
          "wait": [
-            {
-               "method": "textDocument/publishDiagnostics",
-               "params": {
-                  "uri": "$URI{foo.adb}",
-                  "diagnostics": []
-               }
-            },
             {
                "id": 16,
                "result": null

testsuite/ada_lsp/incorrect_fallback/incorrect_fallback.json

@@ -236,7 +236,12 @@
             "id": 2,
             "method": "shutdown"
          },
-         "wait": []
+         "wait": [
+            {
+               "id": 2,
+               "result": null
+            }
+         ]
       }
    },
    {
@@ -246,12 +251,7 @@
             "jsonrpc": "2.0",
             "method": "exit"
          },
-         "wait": [
-            {
-               "id": 2,
-               "result": null
-            }
-         ]
+         "wait": []
       }
    },
    {