diff --git a/lib/revert.js b/lib/revert.js index 032bf4a..220565c 100644 --- a/lib/revert.js +++ b/lib/revert.js @@ -16,7 +16,7 @@ function revert(h, node) { if (subtype === 'collapsed') { suffix += '[]' } else if (subtype === 'full') { - suffix += '[' + node.identifier + ']' + suffix += '[' + (node.reference || node.identifier) + ']' } if (node.type === 'imageReference') { diff --git a/test/image-reference.js b/test/image-reference.js index cd15737..8114af4 100644 --- a/test/image-reference.js +++ b/test/image-reference.js @@ -20,7 +20,20 @@ test('ImageReference', function(t) { }) ), u('text', '![golf][foxtrot]'), - 'should fall back on full `imageReference`s' + 'should fall back on full `imageReference`s - reference not present' + ) + + t.deepEqual( + to( + u('imageReference', { + identifier: 'foxtrot', + reference: 'Foxtrot', + referenceType: 'full', + alt: 'golf' + }) + ), + u('text', '![golf][Foxtrot]'), + 'should fall back on full `imageReference`s - reference is present' ) t.deepEqual( diff --git a/test/link-reference.js b/test/link-reference.js index 233b8c0..34b7a46 100644 --- a/test/link-reference.js +++ b/test/link-reference.js @@ -18,7 +18,19 @@ test('LinkReference', function(t) { ]) ), [u('text', '[echo][delta]')], - 'should fall back on full `linkReference`s' + 'should fall back on full `linkReference`s - original reference not present' + ) + + t.deepEqual( + to( + u('linkReference', + {identifier: 'delta', reference: 'Delta', referenceType: 'full'}, [ + u('text', 'echo') + ] + ) + ), + [u('text', '[echo][Delta]')], + 'should fall back on full `linkReference`s - original reference is present' ) t.deepEqual(