unit test

Author: mabaasit

packages/compass-data-modeling/src/components/diagram-editor-toolbar.spec.tsx

@@ -12,6 +12,7 @@ function renderDiagramEditorToolbar(
       step="EDITING"
       hasUndo={true}
       hasRedo={true}
+      onDownloadClick={() => {}}
       onUndoClick={() => {}}
       onRedoClick={() => {}}
       onExportClick={() => {}}
@@ -63,12 +64,21 @@ describe('DiagramEditorToolbar', function () {
     });
   });
 
-  it('renders export buttona and calls onExportClick', function () {
+  it('renders export button and calls onExportClick', function () {
     const exportSpy = sinon.spy();
     renderDiagramEditorToolbar({ onExportClick: exportSpy });
     const exportButton = screen.getByRole('button', { name: 'Export' });
     expect(exportButton).to.exist;
     userEvent.click(exportButton);
     expect(exportSpy).to.have.been.calledOnce;
   });
+
+  it('renders download button and calls onDownloadClick', function () {
+    const downloadSpy = sinon.spy();
+    renderDiagramEditorToolbar({ onDownloadClick: downloadSpy });
+    const downloadButton = screen.getByRole('button', { name: 'Download' });
+    expect(downloadButton).to.exist;
+    userEvent.click(downloadButton);
+    expect(downloadSpy).to.have.been.calledOnce;
+  });
 });