Function runTestAtLocation

lsp/src/codeLens.ts:245–320

(
  args: RunTestArgs,
  pythonEnv: PythonEnvironment,
)

}

async function runTestAtLocation(
  args: RunTestArgs,
  pythonEnv: PythonEnvironment,
): Promise<void> {
  const uri = parseUri(args.uri);
  if (!uri) {
    return;
  }
  const cwd = args.cwd;
  const className = args.className;
  const testName = args.testName;

  if (args.position && !testName && !className) {
    await runAtCursor(uri, args.position);
    return;
  }

  const interpreter = await pythonEnv.getInterpreterPath(uri);
  if (!interpreter) {
    void showRunnableCodeLensError(
      uri,
      'missing-interpreter',
      'Pyrefly could not determine a Python interpreter for this file. Ensure the correct interpreter is selected in your IDE before using runnable CodeLens.',
    );
    return;
  }
  if (args.isUnittest === true) {
    const moduleName = moduleNameFromPath(uri);
    if (!moduleName) {
      if (args.position) {
        await runAtCursor(uri, args.position);
      }
      return;
    }
    let target = moduleName;
    if (className) {
      target = `${target}.${className}`;
    }
    if (testName) {
      target = `${target}.${testName}`;
    }
    await executeProcessTask(
      uri,
      cwd,
      {type: TASK_SOURCE, action: 'runUnittest'},
      'Pyrefly: Run Test',
      interpreter,
      ['-m', 'unittest', target],
    );
    return;
  }

  let nodeId = uri.fsPath;
  if (className) {
    nodeId = `${nodeId}::${className}`;
  }
  if (testName) {
    nodeId = `${nodeId}::${testName}`;