Files
avante.nvim/tests/utils/fix_diff_spec.lua

719 lines
26 KiB
Lua

local Utils = require("avante.utils")
describe("Utils.fix_diff", function()
it("should not break normal diff", function()
local diff = [[------- SEARCH
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="p-6">
<div className="py-8 overflow-auto text-sm">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
<div className="text-center">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
=======
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="flex flex-col" style={{ maxHeight: '80vh' }}>
<div className="flex-1 overflow-y-auto p-6">
<div className="text-sm font-mono whitespace-pre-wrap">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
</div>
<div className="text-center mt-4">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
<div ref={(el) => {
if (el) {
el.scrollIntoView({ behavior: 'smooth', block: 'end' });
}
}} />
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end p-4 border-t">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
+++++++ REPLACE
]]
local fixed_diff = Utils.fix_diff(diff)
assert.equals(diff, fixed_diff)
end)
it("should not break normal multiple diff", function()
local diff = [[------- SEARCH
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="p-6">
<div className="py-8 overflow-auto text-sm">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
<div className="text-center">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
=======
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="flex flex-col" style={{ maxHeight: '80vh' }}>
<div className="flex-1 overflow-y-auto p-6">
<div className="text-sm font-mono whitespace-pre-wrap">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
</div>
<div className="text-center mt-4">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
<div ref={(el) => {
if (el) {
el.scrollIntoView({ behavior: 'smooth', block: 'end' });
}
}} />
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end p-4 border-t">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
+++++++ REPLACE
------- SEARCH
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="p-6">
=======
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl aaa">
<div className="p-12">
+++++++ REPLACE
]]
local fixed_diff = Utils.fix_diff(diff)
assert.equals(diff, fixed_diff)
end)
it("should fix duplicated REPLACE delimiters", function()
local diff = [[------- SEARCH
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="p-6">
<div className="py-8 overflow-auto text-sm">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
<div className="text-center">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
------- REPLACE
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="flex flex-col" style={{ maxHeight: '80vh' }}>
<div className="flex-1 overflow-y-auto p-6">
<div className="text-sm font-mono whitespace-pre-wrap">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
</div>
<div className="text-center mt-4">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
<div ref={(el) => {
if (el) {
el.scrollIntoView({ behavior: 'smooth', block: 'end' });
}
}} />
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end p-4 border-t">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
------- REPLACE
]]
local expected_diff = [[------- SEARCH
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="p-6">
<div className="py-8 overflow-auto text-sm">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
<div className="text-center">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
=======
<Modal isOpen={showLogs} onClose={() => setShowLogs(false)} title="Project PRD Logs" size="xl">
<div className="flex flex-col" style={{ maxHeight: '80vh' }}>
<div className="flex-1 overflow-y-auto p-6">
<div className="text-sm font-mono whitespace-pre-wrap">
<ReactMarkdown remarkPlugins={[remarkGfm]}>{logs.split('\n').join('\n\n')}</ReactMarkdown>
</div>
<div className="text-center mt-4">{logsLoading && <ScaleLoader color="#555" width={3} height={10} speedMultiplier={2.3} />}</div>
<div ref={(el) => {
if (el) {
el.scrollIntoView({ behavior: 'smooth', block: 'end' });
}
}} />
</div>
</div>
{logs.length > 0 && (
<div className="flex justify-end p-4 border-t">
<button
onClick={() => setShowLogs(false)}
className="bg-japanese-chigusa-600 text-white px-4 py-2 hover:bg-japanese-chigusa-700 focus:outline-none focus:ring-2 focus:ring-blue-500 focus:ring-offset-2 transition-colors"
>
Close
</button>
</div>
)}
</Modal>
+++++++ REPLACE
]]
local fixed_diff = Utils.fix_diff(diff)
assert.equals(expected_diff, fixed_diff)
end)
it("should fix the delimiter is on the same line as the content", function()
local diff = [[------- // Fetch initial stages when project changes
useEffect(() => {
if (!subscribedProject) return;
const fetchStages = async () => {
try {
const response = await fetch(`/api/projects/${subscribedProject}/stages`);
if (response.ok) {
const stagesData = await response.json();
setStages(stagesData);
}
} catch (error) {
console.error('Failed to fetch stages:', error);
}
};
fetchStages();
}, [subscribedProject, forceUpdateCounter]);
======= // Fetch initial stages when project changes
useEffect(() => {
if (!subscribedProject) return;
const fetchStages = async () => {
try {
// Use the correct API endpoint for stages by project UUID
const response = await fetch(`/api/stages?project_uuid=${subscribedProject}`);
if (response.ok) {
const stagesData = await response.json();
setStages(stagesData);
}
} catch (error) {
console.error('Failed to fetch stages:', error);
}
};
fetchStages();
}, [subscribedProject, forceUpdateCounter]);
+++++++ REPLACE
]]
local expected_diff = [[------- SEARCH
// Fetch initial stages when project changes
useEffect(() => {
if (!subscribedProject) return;
const fetchStages = async () => {
try {
const response = await fetch(`/api/projects/${subscribedProject}/stages`);
if (response.ok) {
const stagesData = await response.json();
setStages(stagesData);
}
} catch (error) {
console.error('Failed to fetch stages:', error);
}
};
fetchStages();
}, [subscribedProject, forceUpdateCounter]);
=======
// Fetch initial stages when project changes
useEffect(() => {
if (!subscribedProject) return;
const fetchStages = async () => {
try {
// Use the correct API endpoint for stages by project UUID
const response = await fetch(`/api/stages?project_uuid=${subscribedProject}`);
if (response.ok) {
const stagesData = await response.json();
setStages(stagesData);
}
} catch (error) {
console.error('Failed to fetch stages:', error);
}
};
fetchStages();
}, [subscribedProject, forceUpdateCounter]);
+++++++ REPLACE
]]
local fixed_diff = Utils.fix_diff(diff)
assert.equals(expected_diff, fixed_diff)
end)
it("should fix unified diff", function()
local diff = [[--- lua/avante/sidebar.lua
+++ lua/avante/sidebar.lua
@@ -3099,7 +3099,7 @@
function Sidebar:create_todos_container()
local history = Path.history.load(self.code.bufnr)
if not history or not history.todos or #history.todos == 0 then
- if self.containers.todos then self.containers.todos:unmount() end
+ if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end
self.containers.todos = nil
self:adjust_layout()
return
@@ -3121,7 +3121,7 @@
}),
position = "bottom",
size = {
- height = 3,
+ height = math.min(3, math.max(1, vim.o.lines - 5)),
},
})
self.containers.todos:mount()
@@ -3151,11 +3151,15 @@
self:render_header(
self.containers.todos.winid,
todos_buf,
- Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
+ Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
Highlights.SUBTITLE,
Highlights.REVERSED_SUBTITLE
)
- self:adjust_layout()
+
+ local ok, err = pcall(function()
+ self:adjust_layout()
+ end)
+ if not ok then Utils.debug("Failed to adjust layout after todos creation:", err) end
end
function Sidebar:adjust_layout()
]]
local expected_diff = [[------- SEARCH
function Sidebar:create_todos_container()
local history = Path.history.load(self.code.bufnr)
if not history or not history.todos or #history.todos == 0 then
if self.containers.todos then self.containers.todos:unmount() end
self.containers.todos = nil
self:adjust_layout()
return
=======
function Sidebar:create_todos_container()
local history = Path.history.load(self.code.bufnr)
if not history or not history.todos or #history.todos == 0 then
if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end
self.containers.todos = nil
self:adjust_layout()
return
+++++++ REPLACE
------- SEARCH
}),
position = "bottom",
size = {
height = 3,
},
})
self.containers.todos:mount()
=======
}),
position = "bottom",
size = {
height = math.min(3, math.max(1, vim.o.lines - 5)),
},
})
self.containers.todos:mount()
+++++++ REPLACE
------- SEARCH
self:render_header(
self.containers.todos.winid,
todos_buf,
Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
Highlights.SUBTITLE,
Highlights.REVERSED_SUBTITLE
)
self:adjust_layout()
end
function Sidebar:adjust_layout()
=======
self:render_header(
self.containers.todos.winid,
todos_buf,
Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
Highlights.SUBTITLE,
Highlights.REVERSED_SUBTITLE
)
local ok, err = pcall(function()
self:adjust_layout()
end)
if not ok then Utils.debug("Failed to adjust layout after todos creation:", err) end
end
function Sidebar:adjust_layout()
+++++++ REPLACE]]
local fixed_diff = Utils.fix_diff(diff)
assert.equals(expected_diff, fixed_diff)
end)
it("should fix unified diff 2", function()
local diff = [[
@@ -3099,7 +3099,7 @@
function Sidebar:create_todos_container()
local history = Path.history.load(self.code.bufnr)
if not history or not history.todos or #history.todos == 0 then
- if self.containers.todos then self.containers.todos:unmount() end
+ if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end
self.containers.todos = nil
self:adjust_layout()
return
@@ -3121,7 +3121,7 @@
}),
position = "bottom",
size = {
- height = 3,
+ height = math.min(3, math.max(1, vim.o.lines - 5)),
},
})
self.containers.todos:mount()
@@ -3151,11 +3151,15 @@
self:render_header(
self.containers.todos.winid,
todos_buf,
- Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
+ Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
Highlights.SUBTITLE,
Highlights.REVERSED_SUBTITLE
)
- self:adjust_layout()
+
+ local ok, err = pcall(function()
+ self:adjust_layout()
+ end)
+ if not ok then Utils.debug("Failed to adjust layout after todos creation:", err) end
end
function Sidebar:adjust_layout()
]]
local expected_diff = [[------- SEARCH
function Sidebar:create_todos_container()
local history = Path.history.load(self.code.bufnr)
if not history or not history.todos or #history.todos == 0 then
if self.containers.todos then self.containers.todos:unmount() end
self.containers.todos = nil
self:adjust_layout()
return
=======
function Sidebar:create_todos_container()
local history = Path.history.load(self.code.bufnr)
if not history or not history.todos or #history.todos == 0 then
if self.containers.todos and Utils.is_valid_container(self.containers.todos) then self.containers.todos:unmount() end
self.containers.todos = nil
self:adjust_layout()
return
+++++++ REPLACE
------- SEARCH
}),
position = "bottom",
size = {
height = 3,
},
})
self.containers.todos:mount()
=======
}),
position = "bottom",
size = {
height = math.min(3, math.max(1, vim.o.lines - 5)),
},
})
self.containers.todos:mount()
+++++++ REPLACE
------- SEARCH
self:render_header(
self.containers.todos.winid,
todos_buf,
Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
Highlights.SUBTITLE,
Highlights.REVERSED_SUBTITLE
)
self:adjust_layout()
end
function Sidebar:adjust_layout()
=======
self:render_header(
self.containers.todos.winid,
todos_buf,
Utils.icon(" ") .. "Todos" .. " (" .. done_count .. "/" .. total_count .. ")",
Highlights.SUBTITLE,
Highlights.REVERSED_SUBTITLE
)
local ok, err = pcall(function()
self:adjust_layout()
end)
if not ok then Utils.debug("Failed to adjust layout after todos creation:", err) end
end
function Sidebar:adjust_layout()
+++++++ REPLACE]]
local fixed_diff = Utils.fix_diff(diff)
assert.equals(expected_diff, fixed_diff)
end)
it("should fix duplicated replace blocks", function()
local diff = [[------- SEARCH
useEffect(() => {
if (!isExpanded || !textContentRef.current) {
setShowFixedCollapseButton(false);
return;
}
const observer = new IntersectionObserver(
([entry]) => {
setShowFixedCollapseButton(!entry.isIntersecting);
},
{
root: null,
rootMargin: '0px',
threshold: 1.0,
}
);
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
observer.observe(collapseButton);
}
return () => {
if (collapseButton) {
observer.unobserve(collapseButton);
}
};
}, [isExpanded, textContentRef.current]);
=======
useEffect(() => {
if (!isExpanded || !textContentRef.current) {
setShowFixedCollapseButton(false);
return;
}
// Check initial visibility of the collapse button
const checkInitialVisibility = () => {
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
const rect = collapseButton.getBoundingClientRect();
const isVisible = rect.top >= 0 && rect.bottom <= window.innerHeight;
setShowFixedCollapseButton(!isVisible);
}
};
// Small delay to ensure DOM is updated after expansion
const timeoutId = setTimeout(checkInitialVisibility, 100);
const observer = new IntersectionObserver(
([entry]) => {
setShowFixedCollapseButton(!entry.isIntersecting);
},
{
root: null,
rootMargin: '0px',
threshold: [0, 1.0], // Check both when it starts to leave and when fully visible
}
);
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
observer.observe(collapseButton);
}
return () => {
clearTimeout(timeoutId);
if (collapseButton) {
observer.unobserve(collapseButton);
}
};
}, [isExpanded, textContentRef.current]);
=======
useEffect(() => {
if (!isExpanded || !textContentRef.current) {
setShowFixedCollapseButton(false);
return;
}
// Check initial visibility of the collapse button
const checkInitialVisibility = () => {
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
const rect = collapseButton.getBoundingClientRect();
const isVisible = rect.top >= 0 && rect.bottom <= window.innerHeight;
setShowFixedCollapseButton(!isVisible);
}
};
// Small delay to ensure DOM is updated after expansion
const timeoutId = setTimeout(checkInitialVisibility, 100);
const observer = new IntersectionObserver(
([entry]) => {
setShowFixedCollapseButton(!entry.isIntersecting);
},
{
root: null,
rootMargin: '0px',
threshold: [0, 1.0], // Check both when it starts to leave and when fully visible
}
);
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
observer.observe(collapseButton);
}
return () => {
clearTimeout(timeoutId);
if (collapseButton) {
observer.unobserve(collapseButton);
}
};
}, [isExpanded, textContentRef.current]);
+++++++ REPLACE
]]
local expected_diff = [[------- SEARCH
useEffect(() => {
if (!isExpanded || !textContentRef.current) {
setShowFixedCollapseButton(false);
return;
}
const observer = new IntersectionObserver(
([entry]) => {
setShowFixedCollapseButton(!entry.isIntersecting);
},
{
root: null,
rootMargin: '0px',
threshold: 1.0,
}
);
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
observer.observe(collapseButton);
}
return () => {
if (collapseButton) {
observer.unobserve(collapseButton);
}
};
}, [isExpanded, textContentRef.current]);
=======
useEffect(() => {
if (!isExpanded || !textContentRef.current) {
setShowFixedCollapseButton(false);
return;
}
// Check initial visibility of the collapse button
const checkInitialVisibility = () => {
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
const rect = collapseButton.getBoundingClientRect();
const isVisible = rect.top >= 0 && rect.bottom <= window.innerHeight;
setShowFixedCollapseButton(!isVisible);
}
};
// Small delay to ensure DOM is updated after expansion
const timeoutId = setTimeout(checkInitialVisibility, 100);
const observer = new IntersectionObserver(
([entry]) => {
setShowFixedCollapseButton(!entry.isIntersecting);
},
{
root: null,
rootMargin: '0px',
threshold: [0, 1.0], // Check both when it starts to leave and when fully visible
}
);
const collapseButton = collapseButtonRef.current;
if (collapseButton) {
observer.observe(collapseButton);
}
return () => {
clearTimeout(timeoutId);
if (collapseButton) {
observer.unobserve(collapseButton);
}
};
}, [isExpanded, textContentRef.current]);
+++++++ REPLACE]]
local fixed_diff = Utils.fix_diff(diff)
assert.equals(expected_diff, fixed_diff)
end)
end)