Add test

TurboWarp · Dec 22, 2023 · 8b25fb9 · 8b25fb9
1 parent ef9466a
commit 8b25fb9
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/test/unit/tw_branch_icon_uri.js b/test/unit/tw_branch_icon_uri.js
@@ -0,0 +1,39 @@
+const {test} = require('tap');
+const VirtualMachine = require('../../src/virtual-machine');
+const BlockType = require('../../src/extension-support/block-type');
+
+test('branchIconURI', t => {
+    const vm = new VirtualMachine();
+    vm.extensionManager._registerInternalExtension({
+        getInfo: () => ({
+            id: 'testextension',
+            name: 'test',
+            blocks: [
+                {
+                    blockType: BlockType.LOOP,
+                    opcode: 'block1',
+                    text: 'no custom icon'
+                },
+                {
+                    blockType: BlockType.LOOP,
+                    opcode: 'block2',
+                    text: 'LOOP with custom icon',
+                    branchIconURI: 'data:whatever1'
+                },
+                {
+                    blockType: BlockType.CONDITIONAL,
+                    opcode: 'block2',
+                    text: 'CONDITIONAL with custom icon',
+                    branchIconURI: 'data:whatever2'
+                }
+            ]
+        })
+    });
+
+    const blocks = vm.runtime.getBlocksJSON();
+    t.equal(blocks[0].args2[0].src, './static/blocks-media/repeat.svg', 'default custom icon');
+    t.equal(blocks[1].args2[0].src, 'data:whatever1', 'LOOP with custom icon');
+    t.equal(blocks[2].args2[0].src, 'data:whatever2', 'CONDITIONAL with custom icon');
+
+    t.end();
+});