parseSourceCodeDefinitions.solidity.spec.ts

import { testParseSourceCodeDefinitions } from "./helpers"
import { solidityQuery } from "../queries"
import { sampleSolidity } from "./fixtures/sample-solidity"

describe("Solidity Source Code Definition Tests", () => {
	let parseResult: string

	beforeAll(async () => {
		const result = await testParseSourceCodeDefinitions("test.sol", sampleSolidity, {
			language: "solidity",
			wasmFile: "tree-sitter-solidity.wasm",
			queryString: solidityQuery,
			extKey: "sol",
		})
		expect(result).toBeDefined()
		expect(typeof result).toBe("string")
		parseResult = result as string
	})

	it("should parse contract declarations", () => {
		expect(parseResult).toMatch(/22--102 \| contract TestContract is ITestInterface/)
		expect(parseResult).toMatch(/5--9 \| interface ITestInterface/)
		expect(parseResult).toMatch(/11--20 \| library MathLib/)
	})

	it("should parse using directives", () => {
		expect(parseResult).toMatch(/23--23 \| {5}using MathLib for uint256;/)
	})

	it("should parse type declarations", () => {
		expect(parseResult).toMatch(/25--30 \| {5}struct UserInfo {/)
		expect(parseResult).toMatch(/32--37 \| {5}enum UserRole {/)
	})

	it("should parse state variable declarations", () => {
		expect(parseResult).toMatch(/39--39 \| {5}uint256 private immutable totalSupply;/)
		expect(parseResult).toMatch(/40--40 \| {5}mapping\(address => UserInfo\) private users;/)
		expect(parseResult).toMatch(/41--41 \| {5}UserRole\[\] private roles;/)
	})
	it("should parse function declarations", () => {
		expect(parseResult).toMatch(/70--87 \| {5}function transfer\(/)
		expect(parseResult).toMatch(/89--93 \| {5}function interfaceFunction\(/)
		expect(parseResult).toMatch(/6--6 \| {5}function interfaceFunction\(uint256 value\) external returns \(bool\);/)
		expect(parseResult).toMatch(
			/12--14 \| {5}function add\(uint256 a, uint256 b\) internal pure returns \(uint256\) {/,
		)
		expect(parseResult).toMatch(
			/16--19 \| {5}function subtract\(uint256 a, uint256 b\) internal pure returns \(uint256\) {/,
		)
	})

	it("should parse constructor declarations", () => {
		expect(parseResult).toMatch(/63--68 \| {5}constructor\(uint256 _initialSupply\) {/)
	})

	it("should parse special function declarations", () => {
		expect(parseResult).toMatch(/95--97 \| {5}fallback\(\) external payable {/)
		expect(parseResult).toMatch(/99--101 \| {5}receive\(\) external payable {/)
	})

	it("should parse event declarations", () => {
		expect(parseResult).toMatch(/43--47 \| {5}event Transfer\(/)
		expect(parseResult).toMatch(/7--7 \| {5}event InterfaceEvent\(address indexed sender, uint256 value\);/)
	})

	it("should parse error declarations", () => {
		expect(parseResult).toMatch(/49--53 \| {5}error InsufficientBalance\(/)
		expect(parseResult).toMatch(/8--8 \| {5}error InterfaceError\(string message\);/)
	})

	it("should parse modifier declarations", () => {
		expect(parseResult).toMatch(/55--61 \| {5}modifier onlyAdmin\(\) {/)
	})
})