import unittest from aider_gitea.__main__ import generate_branch_name class TestGenerateBranchName(unittest.TestCase): def test_generate_branch_name(self): self.assertEqual(generate_branch_name("Fix bug in login"), "issue-fix-bug-in-login") self.assertEqual(generate_branch_name("Add new feature!"), "issue-add-new-feature") self.assertEqual(generate_branch_name("Refactor code (cleanup)"), "issue-refactor-code-cleanup") self.assertEqual(generate_branch_name("123 Start with numbers"), "issue-123-start-with-numbers") self.assertEqual(generate_branch_name("Handle special chars: @#$%^&*()"), "issue-handle-special-chars") if __name__ == '__main__': unittest.main()