Add basic support for Qt Quick Test

plugins/autotest/testvisitor.cpp

@@ -26,11 +26,15 @@
 
 #include <cpptools/cppmodelmanager.h>
 
+#include <qmljs/parser/qmljsast_p.h>
+
 #include <QList>
 
 namespace Autotest {
 namespace Internal {
 
+/************************** Cpp Test Symbol Visitor ***************************/
+
 TestVisitor::TestVisitor(const QString &fullQualifiedClassName)
     : m_className(fullQualifiedClassName)
 {
@@ -59,10 +63,11 @@ bool TestVisitor::visit(CPlusPlus::Class *symbol)
         if (className != m_className)
             continue;
 
-        if (auto func = type->asFunctionType()) {
+        if (const auto func = type->asFunctionType()) {
             if (func->isSlot() && member->isPrivate()) {
                 const QString name = o.prettyName(func->name());
                 if (!ignoredFunctions.contains(name) && !name.endsWith(QLatin1String("_data"))) {
+                    // TODO use definition of function instead of declaration!
                     TestCodeLocation location;
                     location.m_fileName = QLatin1String(member->fileName());
                     location.m_line = member->line();
@@ -75,6 +80,8 @@ bool TestVisitor::visit(CPlusPlus::Class *symbol)
     return true;
 }
 
+/**************************** Cpp Test AST Visitor ****************************/
+
 TestAstVisitor::TestAstVisitor(CPlusPlus::Document::Ptr doc)
     : ASTVisitor(doc->translationUnit()),
       m_currentDoc(doc)
@@ -90,15 +97,15 @@ bool TestAstVisitor::visit(CPlusPlus::CallAST *ast)
     if (!m_currentScope || m_currentDoc.isNull())
         return false;
 
-    if (auto expressionAST = ast->base_expression) {
-        if (auto idExpressionAST = expressionAST->asIdExpression()) {
-            if (auto qualifiedNameAST = idExpressionAST->name->asQualifiedName()) {
+    if (const auto expressionAST = ast->base_expression) {
+        if (const auto idExpressionAST = expressionAST->asIdExpression()) {
+            if (const auto qualifiedNameAST = idExpressionAST->name->asQualifiedName()) {
                 const CPlusPlus::Overview o;
                 const QString prettyName = o.prettyName(qualifiedNameAST->name);
                 if (prettyName == QLatin1String("QTest::qExec")) {
-                    if (auto expressionListAST = ast->expression_list) {
+                    if (const auto expressionListAST = ast->expression_list) {
                         // first argument is the one we need
-                        if (auto argumentExpressionAST = expressionListAST->value) {
+                        if (const auto argumentExpressionAST = expressionListAST->value) {
                             CPlusPlus::TypeOfExpression toe;
                             CppTools::CppModelManager *cppMM = CppTools::CppModelManager::instance();
                             toe.init(m_currentDoc, cppMM->snapshot());
@@ -106,7 +113,7 @@ bool TestAstVisitor::visit(CPlusPlus::CallAST *ast)
                                     = toe(argumentExpressionAST, m_currentDoc, m_currentScope);
 
                             if (toeItems.size()) {
-                                if (auto pointerType = toeItems.first().type()->asPointerType())
+                                if (const auto pointerType = toeItems.first().type()->asPointerType())
                                     m_className = o.prettyType(pointerType->elementType());
                             }
                         }
@@ -124,5 +131,63 @@ bool TestAstVisitor::visit(CPlusPlus::CompoundStatementAST *ast)
     return true;
 }
 
+/*************************** Quick Test AST Visitor ***************************/
+
+TestQmlVisitor::TestQmlVisitor(QmlJS::Document::Ptr doc)
+    : m_currentDoc(doc)
+{
+}
+
+TestQmlVisitor::~TestQmlVisitor()
+{
+}
+
+bool TestQmlVisitor::visit(QmlJS::AST::UiObjectDefinition *ast)
+{
+    const QStringRef name = ast->qualifiedTypeNameId->name;
+    if (name != QLatin1String("TestCase"))
+        return false;
+
+    m_currentTestCaseName.clear();
+    const auto sourceLocation = ast->firstSourceLocation();
+    m_testCaseLocation.m_fileName = m_currentDoc->fileName();
+    m_testCaseLocation.m_line = sourceLocation.startLine;
+    m_testCaseLocation.m_column = sourceLocation.startColumn - 1;
+    return true;
+}
+
+bool TestQmlVisitor::visit(QmlJS::AST::ExpressionStatement *ast)
+{
+    const QmlJS::AST::ExpressionNode *expr = ast->expression;
+    return expr->kind == QmlJS::AST::Node::Kind_StringLiteral;
+}
+
+bool TestQmlVisitor::visit(QmlJS::AST::UiScriptBinding *ast)
+{
+    const QStringRef name = ast->qualifiedId->name;
+    return name == QLatin1String("name");
+}
+
+bool TestQmlVisitor::visit(QmlJS::AST::FunctionDeclaration *ast)
+{
+    const QStringRef name = ast->name;
+    if (name.startsWith(QLatin1String("test_"))) {
+        const auto sourceLocation = ast->firstSourceLocation();
+        TestCodeLocation location;
+        location.m_fileName = m_currentDoc->fileName();
+        location.m_line = sourceLocation.startLine;
+        location.m_column = sourceLocation.startColumn - 1;
+
+        m_testFunctions.insert(name.toString(), location);
+    }
+    return false;
+}
+
+bool TestQmlVisitor::visit(QmlJS::AST::StringLiteral *ast)
+{
+    m_currentTestCaseName = ast->value.toString();
+    return false;
+}
+
 } // namespace Internal
 } // namespace Autotest