import { describe, expect, test } from 'vitest'; import { parseFile } from '../../../src/core/treeSitter/parseFile.js'; import { createMockConfig } from '../../../tests/testing/testUtils.js'; describe('parseFile for Dart', () => { test('should parse Dart correctly', async () => { const fileContent = ` /// A simple greeting class class Greeter { /// The name to greet final String name; /// Constructor Greeter(this.name); /// Prints a greeting message void greet() { print('Hello, $name!'); } /// Returns a greeting message String getMessage() { return 'Hello, $name!'; } } /// Main entry point void main() { final greeter = Greeter('World'); greeter.greet(); } `; const filePath = 'dummy.dart'; const config = {}; const result = await parseFile(fileContent, filePath, createMockConfig(config)); expect(typeof result).toBe('string'); const expectContents = [ '/// A simple greeting class', 'class Greeter {', '/// The name to greet', '/// Constructor', '/// Prints a greeting message', 'void greet() {', '/// Returns a greeting message', 'String getMessage() {', '/// Main entry point', 'void main() {', ]; for (const expectContent of expectContents) { expect(result).toContain(expectContent); } }); test('should parse Dart with imports correctly', async () => { const fileContent = ` import 'dart:async'; import 'package:flutter/material.dart'; /// A simple widget class MyWidget extends StatelessWidget { @override Widget build(BuildContext context) { return Container(); } } `; const filePath = 'dummy.dart'; const config = {}; const result = await parseFile(fileContent, filePath, createMockConfig(config)); expect(typeof result).toBe('string'); const expectContents = [ "import 'dart:async';", "import 'package:flutter/material.dart';", '/// A simple widget', 'class MyWidget extends StatelessWidget {', ]; for (const expectContent of expectContents) { expect(result).toContain(expectContent); } }); test('should parse Dart with enum, extension, and constructor', async () => { const fileContent = ` /// Status enum enum Status { pending, active, completed } /// String extension extension StringExtension on String { /// Capitalizes first letter String capitalize() { return this[0].toUpperCase() + substring(1); } } /// User class class User { final String name; final Status status; /// User constructor User(this.name, this.status); /// Named constructor User.guest() : name = 'Guest', status = Status.pending; } `; const filePath = 'dummy.dart'; const config = {}; const result = await parseFile(fileContent, filePath, createMockConfig(config)); expect(typeof result).toBe('string'); const expectContents = [ '/// Status enum', 'enum Status {', '/// String extension', 'extension StringExtension on String {', '/// Capitalizes first letter', 'String capitalize() {', '/// User class', 'class User {', '/// User constructor', '/// Named constructor', ]; for (const expectContent of expectContents) { expect(result).toContain(expectContent); } }); test('should parse Dart mixin, typedef, getter, setter, and factory', async () => { const fileContent = ` /// JSON map alias typedef Json = Map; /// Walker mixin mixin Walker on Animal { /// Walk action void walk() { print('walking'); } } /// Animal base class class Animal { final String _name; Animal(this._name); /// Animal name getter String get name => _name; /// Animal name setter set nickname(String value) { print('nickname is $value'); } /// Factory constructor factory Animal.guest() { return Animal('Guest'); } /// Redirecting factory factory Animal.copy(Animal other) = Animal; } `; const filePath = 'dummy.dart'; const config = {}; const result = await parseFile(fileContent, filePath, createMockConfig(config)); expect(typeof result).toBe('string'); const expectContents = [ '/// JSON map alias', 'typedef Json = Map;', '/// Walker mixin', 'mixin Walker on Animal {', '/// Walk action', 'void walk() {', '/// Animal name getter', 'String get name => _name;', '/// Animal name setter', 'set nickname(String value) {', '/// Factory constructor', 'factory Animal.guest() {', '/// Redirecting factory', 'factory Animal.copy(Animal other) = Animal;', ]; for (const expectContent of expectContents) { expect(result).toContain(expectContent); } }); test('should parse Dart plain constructor and operator overloads', async () => { const fileContent = ` /// Vector class with operators class Vector { final double x; final double y; /// Plain constructor Vector(this.x, this.y); /// Plus operator Vector operator +(Vector other) => Vector(x + other.x, y + other.y); /// Index operator double operator [](int i) => i == 0 ? x : y; /// Index assignment operator void operator []=(int i, double v) {} /// Equality operator bool operator ==(Object other) => other is Vector && x == other.x && y == other.y; } `; const filePath = 'dummy.dart'; const config = {}; const result = await parseFile(fileContent, filePath, createMockConfig(config)); expect(typeof result).toBe('string'); const expectContents = [ '/// Plain constructor', 'Vector(this.x, this.y);', '/// Plus operator', 'Vector operator +(Vector other) => Vector(x + other.x, y + other.y);', '/// Index operator', 'double operator [](int i) => i == 0 ? x : y;', '/// Index assignment operator', 'void operator []=(int i, double v) {}', '/// Equality operator', 'bool operator ==(Object other) => other is Vector && x == other.x && y == other.y;', ]; for (const expectContent of expectContents) { expect(result).toContain(expectContent); } }); test('should parse Dart const and external constructors', async () => { const fileContent = ` /// Point with const constructors class Point { final int x; final int y; /// Const plain constructor const Point(this.x, this.y); /// Const named constructor const Point.zero() : x = 0, y = 0; /// Const redirecting factory const factory Point.alias() = Point.zero; /// External factory external factory Point.native(); } `; const filePath = 'dummy.dart'; const config = {}; const result = await parseFile(fileContent, filePath, createMockConfig(config)); expect(typeof result).toBe('string'); const expectContents = [ '/// Const plain constructor', 'const Point(this.x, this.y);', '/// Const named constructor', 'const Point.zero() : x = 0, y = 0;', '/// Const redirecting factory', 'const factory Point.alias() = Point.zero;', '/// External factory', 'external factory Point.native();', ]; for (const expectContent of expectContents) { expect(result).toContain(expectContent); } }); });