Add more tests for DB

MyCryptoHQ · Oct 6, 2020 · fc53143 · fc53143
1 parent 13ccbca
commit fc53143
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/src/api/db.spec.ts b/src/api/db.spec.ts
@@ -14,6 +14,16 @@ jest.mock('electron', () => ({
   }
 }));
 
+jest.mock('fs', () => ({
+  promises: {
+    stat: jest
+      .fn()
+      .mockImplementationOnce(() => Promise.resolve(true))
+      .mockImplementationOnce(() => Promise.resolve(false))
+      .mockImplementation(() => Promise.resolve(true))
+  }
+}));
+
 jest.mock('electron-store', () => {
   return jest.fn().mockImplementation(() => ({
     get: jest.fn().mockImplementation((key: string) => {
@@ -27,6 +37,11 @@ jest.mock('electron-store', () => {
 });
 
 describe('handleRequest', () => {
+  it('get login state returns logged out correctly', async () => {
+    const result = await handleRequest({ type: DBRequestType.GET_LOGIN_STATE });
+    expect(result).toBe(LoginState.LOGGED_OUT);
+  });
+
   it('init succesfully initializes the electron-store', async () => {
     const result = await handleRequest({ type: DBRequestType.INIT, password: 'password' });
     expect(result).toBe(true);
@@ -41,10 +56,16 @@ describe('handleRequest', () => {
 
   it('get login state returns new user by default', async () => {
     const result = await handleRequest({ type: DBRequestType.GET_LOGIN_STATE });
-    // @todo Test other cases
     expect(result).toBe(LoginState.NEW_USER);
   });
 
+  it('get login state returns logged in correctly', async () => {
+    const initResult = await handleRequest({ type: DBRequestType.INIT, password: 'password' });
+    expect(initResult).toBe(true);
+    const result = await handleRequest({ type: DBRequestType.GET_LOGIN_STATE });
+    expect(result).toBe(LoginState.LOGGED_IN);
+  });
+
   it('get accounts', async () => {
     const result = await handleRequest({ type: DBRequestType.GET_ACCOUNTS });
     // @todo