@Nullable
private static JstdTestCaseStructure createTestCaseStructure(
@NotNull JstdTestFileStructure jsTestFileStructure,
@NotNull JSCallExpression testCaseCallExpression) {
JSReferenceExpression referenceExpression =
ObjectUtils.tryCast(
testCaseCallExpression.getMethodExpression(), JSReferenceExpression.class);
if (referenceExpression != null) {
String referenceName = referenceExpression.getReferencedName();
if (TEST_CASE_NAME.equals(referenceName) || ASYNC_TEST_CASE_NAME.equals(referenceName)) {
JSExpression[] arguments = JsPsiUtils.getArguments(testCaseCallExpression);
if (arguments.length >= 1) {
String testCaseName = JsPsiUtils.extractStringValue(arguments[0]);
if (testCaseName != null) {
JSObjectLiteralExpression testsObjectLiteral = null;
if (arguments.length >= 2) {
testsObjectLiteral = JsPsiUtils.extractObjectLiteralExpression(arguments[1]);
}
JstdTestCaseStructure testCaseStructure =
new JstdTestCaseStructure(
jsTestFileStructure, testCaseName, testCaseCallExpression, testsObjectLiteral);
jsTestFileStructure.addTestCaseStructure(testCaseStructure);
if (testsObjectLiteral != null) {
fillTestCaseStructureByObjectLiteral(testCaseStructure, testsObjectLiteral);
}
return testCaseStructure;
}
}
}
}
return null;
}
private static void fillJsTestFileStructure(
@NotNull JstdTestFileStructure jsTestFileStructure, @NotNull JSStatement statement) {
if (statement instanceof JSExpressionStatement) {
JSExpressionStatement jsExpressionStatement = (JSExpressionStatement) statement;
JSExpression expressionOfStatement = jsExpressionStatement.getExpression();
if (expressionOfStatement instanceof JSCallExpression) {
// TestCase("testCaseName", { test1: function() {} });
JSCallExpression callExpression = (JSCallExpression) expressionOfStatement;
createTestCaseStructure(jsTestFileStructure, callExpression);
} else if (expressionOfStatement instanceof JSAssignmentExpression) {
// testCase = TestCase("testCaseName");
JSAssignmentExpression jsAssignmentExpression =
(JSAssignmentExpression) expressionOfStatement;
JSCallExpression rOperandCallExpression =
ObjectUtils.tryCast(jsAssignmentExpression.getROperand(), JSCallExpression.class);
if (rOperandCallExpression != null) {
JstdTestCaseStructure testCaseStructure =
createTestCaseStructure(jsTestFileStructure, rOperandCallExpression);
if (testCaseStructure != null) {
JSDefinitionExpression jsDefinitionExpression =
ObjectUtils.tryCast(
jsAssignmentExpression.getLOperand(), JSDefinitionExpression.class);
if (jsDefinitionExpression != null) {
JSReferenceExpression jsReferenceExpression =
ObjectUtils.tryCast(
jsDefinitionExpression.getExpression(), JSReferenceExpression.class);
if (jsReferenceExpression != null) {
String refName = jsReferenceExpression.getReferencedName();
if (refName != null) {
addPrototypeTests(testCaseStructure, refName, jsExpressionStatement);
}
}
}
}
}
}
}
if (statement instanceof JSVarStatement) {
// var testCase = TestCase("testCaseName");
JSVarStatement jsVarStatement = (JSVarStatement) statement;
JSVariable[] jsVariables =
ObjectUtils.notNull(jsVarStatement.getVariables(), JSVariable.EMPTY_ARRAY);
for (JSVariable jsVariable : jsVariables) {
JSCallExpression jsCallExpression =
ObjectUtils.tryCast(jsVariable.getInitializer(), JSCallExpression.class);
if (jsCallExpression != null) {
JstdTestCaseStructure testCaseStructure =
createTestCaseStructure(jsTestFileStructure, jsCallExpression);
if (testCaseStructure != null) {
String refName = jsVariable.getQualifiedName();
if (refName != null) {
addPrototypeTests(testCaseStructure, refName, jsVarStatement);
}
}
}
}
}
}
private static void addPrototypeTests(
@NotNull JstdTestCaseStructure testCaseStructure,
@NotNull String referenceName,
@NotNull JSStatement refStatement) {
List<JSStatement> statements = JsPsiUtils.listStatementsInExecutionOrderNextTo(refStatement);
for (JSStatement statement : statements) {
JSExpressionStatement expressionStatement =
ObjectUtils.tryCast(statement, JSExpressionStatement.class);
if (expressionStatement != null) {
JSAssignmentExpression assignmentExpr =
ObjectUtils.tryCast(expressionStatement.getExpression(), JSAssignmentExpression.class);
if (assignmentExpr != null) {
JSDefinitionExpression wholeLeftDefExpr =
ObjectUtils.tryCast(assignmentExpr.getLOperand(), JSDefinitionExpression.class);
if (wholeLeftDefExpr != null) {
JSReferenceExpression wholeLeftRefExpr =
ObjectUtils.tryCast(wholeLeftDefExpr.getExpression(), JSReferenceExpression.class);
if (wholeLeftRefExpr != null) {
JSReferenceExpression testCaseAndPrototypeRefExpr =
ObjectUtils.tryCast(wholeLeftRefExpr.getQualifier(), JSReferenceExpression.class);
if (testCaseAndPrototypeRefExpr != null) {
if ("prototype".equals(testCaseAndPrototypeRefExpr.getReferencedName())) {
JSReferenceExpression testCaseRefExpr =
ObjectUtils.tryCast(
testCaseAndPrototypeRefExpr.getQualifier(), JSReferenceExpression.class);
if (testCaseRefExpr != null && testCaseRefExpr.getQualifier() == null) {
if (referenceName.equals(testCaseRefExpr.getReferencedName())) {
addPrototypeTest(
testCaseStructure, assignmentExpr.getROperand(), wholeLeftDefExpr);
}
}
}
}
}
}
}
}
}
}