import os
from pathlib import Path
def fix_error_file(filepath):
with open(filepath, 'r') as f:
content = f.read()
if 'mod property_tests' not in content:
return False
fixed_content = content.replace(
'''mod property_tests {
use super::*;
use proptest::prelude::*;
proptest! {
#[test]
fn basic_property_stability(input in ".*") {
// Basic property test for coverage
prop_assert!(true);
}
#[test]
fn module_consistency_check(x in 0u32..1000) {
// Module consistency verification
prop_assert!(x < 1001);
}
}
}''',
'''mod property_tests {
use super::*;
#[test]
fn basic_property_stability() {
// Basic property test for coverage
assert!(true);
}
#[test]
fn module_consistency_check() {
// Module consistency verification
let x = 500u32;
assert!(x < 1001);
}
}'''
)
if fixed_content != content:
with open(filepath, 'w') as f:
f.write(fixed_content)
return True
return False
def main():
error_files = [
'server/src/models/error.rs',
'server/src/scaffold/agent/error.rs',
'server/src/services/wasm/error.rs',
'server/src/unified_protocol/error.rs',
]
fixed_count = 0
for filepath in error_files:
if Path(filepath).exists():
if fix_error_file(filepath):
print(f"✅ Fixed: {filepath}")
fixed_count += 1
else:
print(f"⚠️ Not found: {filepath}")
print(f"\n📊 Summary: Fixed {fixed_count} error files")
if __name__ == "__main__":
main()