GitHub/mermaid
1
0
Fork 0
mirror of https://github.com/mermaid-js/mermaid.git synced 2025-02-04 07:13:25 +08:00
Code Issues Projects Releases Wiki Activity

MockedD3: node() return Element; add selectAll()

Browse Source
This commit is contained in:
Ashley Engelund (weedySeaDragon @ github) 2022-11-17 12:23:01 -08:00
parent 4d7496b8dd
commit 8a3c4f64b2
1 changed files with 22 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
packages/mermaid/src/tests/MockedD3.ts
View File

@ -1,12 +1,18 @@
/** /**
* This is a mocked/stubbed version of the d3 Selection type. Each of the main functions are all * This is a mocked/stubbed version of the d3 Selection type. Each of the main functions are all
* mocked (via vi.fn()) so you can track if they have been called, etc. * mocked (via vi.fn()) so you can track if they have been called, etc.
*
* Note that node() returns a HTML Element with tag 'svg'. It is an empty element (no innerHTML, no children, etc).
* This potentially allows testing of mermaidAPI render().
*/ */
export class MockedD3 { export class MockedD3 {
public attribs = new Map<string, string | null>(); public attribs = new Map<string, string | null>();
public id: string | undefined = ''; public id: string | undefined = '';
_children: MockedD3[] = []; _children: MockedD3[] = [];
_containingHTMLdoc = new Document();
constructor(givenId = 'mock-id') { constructor(givenId = 'mock-id') {
this.id = givenId; this.id = givenId;
} }
@ -29,6 +35,11 @@ export class MockedD3 {
return new MockedD3(cleanId); return new MockedD3(cleanId);
}); });
// This has the same implementation as select(). (It calls it.)
selectAll = vi.fn().mockImplementation(({ select_str = '' }): MockedD3 => {
return this.select(select_str);
});
append = vi append = vi
.fn() .fn()
.mockImplementation(function (this: MockedD3, type: string, id = '' + '-appended'): MockedD3 { .mockImplementation(function (this: MockedD3, type: string, id = '' + '-appended'): MockedD3 {
@ -87,9 +98,18 @@ export class MockedD3 {
this.attribs.set('text', attrValue); this.attribs.set('text', attrValue);
return this; return this;
} }
// NOTE: Arbitrarily returns an empty object. The return value could be something different with a mockReturnValue() or mockImplementation()
public node = vi.fn().mockReturnValue({});
// NOTE: Returns a HTML ELement with tag 'svg' that has _another_ 'svg' element child.
// This allows different tests to succeed -- some need a top level 'svg' and some need a 'svg' element to be the firstChild
// Real implementation returns an HTML Element
public node = vi.fn().mockImplementation(() => {
const topElem = this._containingHTMLdoc.createElement('svg');
const elem_svgChild = this._containingHTMLdoc.createElement('svg'); // another svg element
topElem.appendChild(elem_svgChild);
return topElem;
});
// TODO Is this correct? shouldn't it return a list of HTML Elements?
nodes = vi.fn().mockImplementation(function (this: MockedD3): MockedD3[] { nodes = vi.fn().mockImplementation(function (this: MockedD3): MockedD3[] {
return this._children; return this._children;
}); });