more tests

src/main/file.ts

@@ -83,7 +83,7 @@ export const findProgram = (app: App, program: string) => {
   return null;
 }
 
-export const writeFileContents = (app: App, payload: anyDict) => {
+export const writeFileContents = (app: App, payload: anyDict): string => {
 
   // parse properties
   const properties = payload.properties ? { ...payload.properties } : {};

tests/unit/file.test.ts

@@ -0,0 +1,69 @@
+
+import { vi, expect, test } from 'vitest'
+import * as file from '../../src/main/file'
+import { app } from 'electron'
+import path from 'path'
+import fs from 'fs'
+import os from 'os'
+
+vi.mock('electron', async() => {
+  return {
+    app: {
+      getPath: () => os.tmpdir()
+    },
+  }
+})
+
+test('Find program', async () => {
+  expect(file.findProgram(null, 'sh')).toBe('/bin/sh')
+  expect(file.findProgram(null, 'sh2')).toBeNull()
+})
+
+test('Get file contents', async () => {
+  const contents = file.getFileContents(app, './tests/fixtures/sample.txt')
+  const text = Buffer.from(contents.contents, 'base64').toString()
+  expect(contents).toStrictEqual({
+    url: 'file://./tests/fixtures/sample.txt',
+    contents: 'SGVsbG8gZnJvbSBURVhU',
+  })
+  expect(text).toContain('Hello from TEXT')
+})
+
+test('Delete file', async () => {
+  const tempFile = path.join(os.tmpdir(), 'vitest')
+  fs.writeFileSync(tempFile, 'Hello')
+  expect(fs.existsSync(tempFile)).toBeTruthy()
+  file.deleteFile(null, tempFile)
+  expect(fs.existsSync(tempFile)).toBeFalsy()
+})
+
+test('Write file contents', async () => {
+  const tempFile = path.join(os.tmpdir(), 'vitest')
+  const fileURL = file.writeFileContents(app, {
+    contents: 'SGVsbG8gZnJvbSBURVhU',
+    properties: {
+      filename: 'vitest'
+    }
+  })
+  expect(fileURL).toBe(tempFile)
+  expect(fs.existsSync(tempFile)).toBeTruthy()
+  expect(fs.readFileSync(tempFile, 'utf8')).toBe('Hello from TEXT')
+  file.deleteFile(null, `file://${tempFile}`)
+  expect(fs.existsSync(tempFile)).toBeFalsy()
+})
+
+test('Download local file', async () => {
+  const tempFile = path.join(os.tmpdir(), 'vitest')
+  const fileURL = await file.downloadFile(app, {
+    url: `file://./tests/fixtures/sample.txt`,
+    properties: {
+      filename: 'vitest',
+      prompt: false,
+    }
+  })
+  expect(fileURL).toBe(tempFile)
+  expect(fs.existsSync(tempFile)).toBeTruthy()
+  expect(fs.readFileSync(tempFile, 'utf8')).toBe('Hello from TEXT')
+  file.deleteFile(null, `file://${tempFile}`)
+  expect(fs.existsSync(tempFile)).toBeFalsy()
+})

tests/unit/mimetype.test.ts

@@ -0,0 +1,11 @@
+
+import { expect, test } from 'vitest'
+import { mimeTypeToExtension } from '../../src/main/mimetype'
+
+test('MIME type to extension', async () => {
+  expect(mimeTypeToExtension('application/vnd.openxmlformats-officedocument.wordprocessingml.document')).toBe('docx')
+  expect(mimeTypeToExtension('application/vnd.openxmlformats-officedocument.presentationml.presentation')).toBe('pptx')
+  expect(mimeTypeToExtension('application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')).toBe('xlsx')
+  expect(mimeTypeToExtension('text/plain')).toBe('txt')
+  expect(mimeTypeToExtension('application/pdf')).toBe('pdf')
+})

tests/unit/text.test.ts

@@ -1,6 +1,6 @@
 
-import { getTextContent } from '../../src/main/text'
 import { vi, expect, test } from 'vitest'
+import { getTextContent } from '../../src/main/text'
 import fs from 'fs'
 
 vi.mock('electron', async() => {