Improve and expand on Figma setup script test

In this MR, various improvements are made to the Figma setup script test:

• cleaner mocking of modules
• testing that setup functions are called correctly
• testing figma.ui.onmessage invocation calls correct handler

Part of work towards #13 (closed)