Add tests for AntiSuiteActions

app/shared-hub/actions/__tests__/AntiSuiteActions.test.js

@@ -15,33 +15,33 @@ import configureStore from 'redux-mock-store';
 import thunk from 'redux-thunk';
 import * as utils from '../../utils';
 import * as AntiSuiteActions from '../AntiSuiteActions';
-import { SET_ANTI_TRACKING, SET_AD_BLOCK, SET_SMART_BLOCK } from '../../constants/AntiSuiteConstants';
+import { SET_AD_BLOCK, SET_ANTI_TRACKING, SET_SMART_BLOCK } from '../../constants/AntiSuiteConstants';
 
 const middlewares = [thunk];
 const mockStore = configureStore(middlewares);
 
 const testData = { test: true };
+
 utils.sendMessageInPromise = jest.fn((name, message) => new Promise((resolve, reject) => {
 	switch (name) {
-		case SET_ANTI_TRACKING: {
+		case 'SET_ANTI_TRACKING': {
 			resolve(testData);
 			break;
 		}
-		case SET_AD_BLOCK: {
-			console.log('going in send Message')
-			resolve(message);
+		case 'SET_AD_BLOCK': {
+			resolve(testData);
 			break;
 		}
-		case SET_SMART_BLOCK: {
+		case 'SET_SMART_BLOCK': {
 			resolve(testData);
 			break;
 		}
-		default: resolve(message);
+		default: resolve(testData);
 	}
 }));
 
-describe('app/hub/Views/OnboardingView/ actions', () => {
-	test('getHomeProps action should return correctly', () => {
+describe('app/shared-hub/actions/AntiSuiteActions', () => {
+	test('setAdBlock action should return correctly', () => {
 		const initialState = {};
 		const store = mockStore(initialState);
 
@@ -52,4 +52,28 @@ describe('app/hub/Views/OnboardingView/ actions', () => {
 			expect(actions).toEqual([expectedPayload]);
 		});
 	});
+
+	test('setAntiTracking action should return correctly', () => {
+		const initialState = {};
+		const store = mockStore(initialState);
+
+		const data = testData;
+		const expectedPayload = { data, type: SET_ANTI_TRACKING };
+		return store.dispatch(AntiSuiteActions.setAntiTracking()).then(() => {
+			const actions = store.getActions();
+			expect(actions).toEqual([expectedPayload]);
+		});
+	});
+
+	test('setSmartBlock action should return correctly', () => {
+		const initialState = {};
+		const store = mockStore(initialState);
+
+		const data = testData;
+		const expectedPayload = { data, type: SET_SMART_BLOCK };
+		return store.dispatch(AntiSuiteActions.setSmartBlocking()).then(() => {
+			const actions = store.getActions();
+			expect(actions).toEqual([expectedPayload]);
+		});
+	});
 });

app/shared-hub/utils/index.js

@@ -26,6 +26,10 @@ import { withRouter } from 'react-router-dom';
 import { log } from '../../../src/utils/common';
 import { sendMessage as importedSM, sendMessageInPromise as importedSMIP } from '../../panel/utils/msg';
 
+// Self import this file so makeDeferredDispatcher can mock the sendMessageInPromise function within the same class
+// eslint-disable-next-line import/no-self-import
+import * as utils from './index';
+
 const sendMessageInPromise = function(name, message) {
 	return importedSMIP(name, message, 'ghostery-hub');
 };
@@ -72,7 +76,7 @@ const makeStoreCreator = function(reducers, middlewares) {
  */
 function makeDeferredDispatcher(action, actionData) {
 	return function(dispatch) {
-		return sendMessageInPromise(action, actionData).then((data) => {
+		return utils.sendMessageInPromise(action, actionData).then((data) => {
 			dispatch({
 				type: action,
 				data,
@@ -113,8 +117,8 @@ function buildReduxHOC(stateKeys, actionCreators, baseComponent) {
 export {
 	buildReduxHOC,
 	makeStoreCreator,
-	makeDeferredDispatcher,
 	log,
 	sendMessage,
-	sendMessageInPromise
+	sendMessageInPromise,
+	makeDeferredDispatcher
 };