Update test and qldoc

Author: joefarebrother

python/ql/lib/semmle/python/frameworks/Websockets.qll

@@ -50,7 +50,7 @@ module Websockets {
      * calls, or a special parameter that will be set when functions are called by an external
      * library.
      *
-     * Use the predicate `WebSocket::instance()` to get references to instances of `websockets.asyncio.ServerConnection` and `websockets.sync.ServerConnection`.
+     * Use the predicate `ServerConnection::instance()` to get references to instances of `websockets.asyncio.ServerConnection` and `websockets.sync.ServerConnection`.
      */
     abstract class InstanceSource extends DataFlow::LocalSourceNode { }
 

python/ql/test/library-tests/frameworks/websockets/response_test.py

@@ -34,13 +34,14 @@ def unix_route_handler(websocket, x):  # $ MISSING: requestHandler routedParamet
 if __name__ == "__main__":
     import sys
     server = s1 
-    if len(sys.argv) > 1:
-        if sys.argv[1] == "kw":
+    args = sys.argv # $ threatModelSource[commandargs]=sys.argv
+    if len(args) > 1:
+        if args[1] == "kw":
             server = s2
-        elif sys.argv[1] == "route":
+        elif args[1] == "route":
             server = s3
-        elif sys.argv[1] == "unix":
+        elif args[1] == "unix":
             server = s4
-        elif sys.argv[1] == "unix_route":
+        elif args[1] == "unix_route":
             server = s5
     server.serve_forever()