diff --git a/ui/src/common/ContextMenus.test.jsx b/ui/src/common/ContextMenus.test.jsx
index 787e30805..45917556c 100644
--- a/ui/src/common/ContextMenus.test.jsx
+++ b/ui/src/common/ContextMenus.test.jsx
@@ -67,6 +67,17 @@ describe('AlbumContextMenu', () => {
render()
expect(capturedLoveButtonProps.resource).toBe('album')
})
+
+ it('hides LoveButton when record.missing is true', () => {
+ render()
+ expect(capturedLoveButtonProps.visible).toBe(false)
+ expect(screen.queryByTestId('love-button')).not.toBeInTheDocument()
+ })
+
+ it('renders nothing when record is undefined', () => {
+ const { container } = render()
+ expect(container).toBeEmptyDOMElement()
+ })
})
describe('ArtistContextMenu', () => {
@@ -96,4 +107,15 @@ describe('ArtistContextMenu', () => {
render()
expect(capturedLoveButtonProps.resource).toBe('artist')
})
+
+ it('hides LoveButton when record.missing is true', () => {
+ render()
+ expect(capturedLoveButtonProps.visible).toBe(false)
+ expect(screen.queryByTestId('love-button')).not.toBeInTheDocument()
+ })
+
+ it('renders nothing when record is undefined', () => {
+ const { container } = render()
+ expect(container).toBeEmptyDOMElement()
+ })
})