Python: Model sessionmaker().begin()

python/ql/src/experimental/semmle/python/frameworks/SqlAlchemy.qll

@@ -202,6 +202,14 @@ private module SqlAlchemy {
               .getMember("sessionmaker")
               .getReturn()
               .getACall()
+        or
+        this =
+          API::moduleImport("sqlalchemy")
+              .getMember("orm")
+              .getMember("sessionmaker")
+              .getReturn()
+              .getMember("begin")
+              .getACall()
       }
     }
 

python/ql/test/experimental/library-tests/frameworks/sqlalchemy/new_tests.py

@@ -144,7 +144,7 @@ with Session() as session:
     assert result.fetchall() == [("FOO",)]
 
 with Session.begin() as session:
-    result = session.execute(raw_sql) # $ MISSING: getSql=raw_sql
+    result = session.execute(raw_sql) # $ getSql=raw_sql
     assert result.fetchall() == [("FOO",)]
 
 # Querying (1.4)