// Copyright (c) 2026 Alexander Medvednikov. All rights reserved.
// Use of this source code is governed by an MIT license
// that can be found in the LICENSE file.
module ast

fn lock_expr_ident(name string) Expr {
	return Expr(Ident{
		name: name
	})
}

fn lock_expr_stmt(value string) Stmt {
	return Stmt(ExprStmt{
		expr: Expr(BasicLiteral{
			kind:  .number
			value: value
		})
	})
}

fn test_lock_expr_counts_survive_flat_roundtrip() {
	mut b := new_flat_builder()
	b.append_file(File{
		name:  'lock_expr.v'
		mod:   'lock_expr'
		stmts: [
			Stmt(ModuleStmt{
				name: 'lock_expr'
			}),
			Stmt(ExprStmt{
				expr: Expr(LockExpr{
					lock_exprs:  [lock_expr_ident('a'), lock_expr_ident('b')]
					rlock_exprs: [lock_expr_ident('c')]
					stmts:       [lock_expr_stmt('1'), lock_expr_stmt('2')]
				})
			}),
		]
	})
	files := b.flat.to_files()
	assert files.len == 1
	assert files[0].stmts.len == 2
	assert files[0].stmts[1] is ExprStmt
	expr_stmt := files[0].stmts[1] as ExprStmt
	assert expr_stmt.expr is LockExpr
	lock_expr := expr_stmt.expr as LockExpr
	assert lock_expr.lock_exprs.len == 2
	assert lock_expr.rlock_exprs.len == 1
	assert lock_expr.stmts.len == 2
}