This repository has been archived by the owner on Dec 11, 2019. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 974
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Auditors: @diracdeltas
- Loading branch information
Showing
1 changed file
with
110 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,110 @@ | ||
/* global describe, before, after, it */ | ||
const mockery = require('mockery') | ||
const assert = require('assert') | ||
const sinon = require('sinon') | ||
|
||
require('../braveUnit') | ||
|
||
describe('siteHacks unit tests', function () { | ||
let siteHacks | ||
let beforeSendCB | ||
// let beforeRequestCB | ||
let urlParse | ||
let filtering | ||
const fakeElectron = require('../lib/fakeElectron') | ||
const fakeAdBlock = require('../lib/fakeAdBlock') | ||
const fakeFiltering = { | ||
getMainFrameUrl: (details) => { | ||
return filtering.getMainFrameUrl(details) | ||
}, | ||
isResourceEnabled: (resourceName, url, isPrivate) => { | ||
return true | ||
}, | ||
registerBeforeSendHeadersFilteringCB: (cb) => { | ||
beforeSendCB = cb | ||
}, | ||
registerBeforeRequestFilteringCB: (cb) => { | ||
// beforeRequestCB = cb | ||
} | ||
} | ||
|
||
before(function () { | ||
mockery.enable({ | ||
warnOnReplace: false, | ||
warnOnUnregistered: false, | ||
useCleanCache: true | ||
}) | ||
|
||
mockery.registerMock('electron', fakeElectron) | ||
mockery.registerMock('ad-block', fakeAdBlock) | ||
urlParse = require('../../../app/common/urlParse') | ||
mockery.registerMock('./common/urlParse', urlParse) | ||
filtering = require('../../../app/filtering') | ||
mockery.registerMock('./filtering', fakeFiltering) | ||
|
||
siteHacks = require('../../../app/siteHacks') | ||
}) | ||
|
||
after(function () { | ||
mockery.disable() | ||
}) | ||
|
||
describe('init', function () { | ||
let beforeSendSpy | ||
let beforeRequestSpy | ||
|
||
before(function () { | ||
beforeSendSpy = sinon.spy(fakeFiltering, 'registerBeforeSendHeadersFilteringCB') | ||
beforeRequestSpy = sinon.spy(fakeFiltering, 'registerBeforeRequestFilteringCB') | ||
siteHacks.init() | ||
}) | ||
after(function () { | ||
beforeSendSpy.restore() | ||
beforeRequestSpy.restore() | ||
}) | ||
|
||
it('calls Filtering.registerBeforeSendHeadersFilteringCB', function () { | ||
assert.equal(beforeSendSpy.calledOnce, true) | ||
}) | ||
|
||
describe('in the callback passed into registerBeforeSendHeadersFilteringCB', function () { | ||
let getMainFrameUrlSpy | ||
// let result | ||
const details = { | ||
resourceType: 'mainFrame', | ||
requestHeaders: { | ||
'User-Agent': 'Brave Chrome' | ||
}, | ||
url: 'https://brave.com', | ||
tabId: 1 | ||
} | ||
before(function () { | ||
getMainFrameUrlSpy = sinon.spy(fakeFiltering, 'getMainFrameUrl') | ||
if (typeof beforeSendCB === 'function') { | ||
// result = beforeSendCB(details) | ||
beforeSendCB(details) | ||
} | ||
}) | ||
after(function () { | ||
getMainFrameUrlSpy.restore() | ||
}) | ||
|
||
it('calls Filtering.getMainFrameUrl', function () { | ||
assert.equal(getMainFrameUrlSpy.calledOnce, true) | ||
}) | ||
}) | ||
|
||
it('calls Filtering.registerBeforeRequestFilteringCB', function () { | ||
assert.equal(beforeRequestSpy.calledOnce, true) | ||
}) | ||
|
||
// describe('in the callback passed into registerBeforeRequestFilteringCB', function () { | ||
// let result | ||
// before(function () { | ||
// if (typeof beforeRequestCB === 'function') { | ||
// result = beforeRequestCB() | ||
// } | ||
// }) | ||
// }) | ||
}) | ||
}) |