/**

* Copyright 2016, Google, Inc.

* Licensed under the Apache License, Version 2.0 (the "License");

* you may not use this file except in compliance with the License.

* You may obtain a copy of the License at

*

* http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing, software

* distributed under the License is distributed on an "AS IS" BASIS,

* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

* See the License for the specific language governing permissions and

* limitations under the License.

*/

'use strict';

require(`../../../test/_setup`);

const express = require(`express`);

const path = require(`path`);

const proxyquire = require(`proxyquire`).noPreserveCache();

const request = require(`supertest`);

const SAMPLE_PATH = path.join(__dirname, `../server.js`);

function getSample () {

const testApp = express();

sinon.stub(testApp, `listen`).yields();

const expressMock = sinon.stub().returns(testApp);

const resultsMock = [

{

timestamp: `1234`,

userIp: `abcd`

}

];

const connectionMock = {

query: sinon.stub()

};

connectionMock.query.onFirstCall().yields();

connectionMock.query.onSecondCall().yields(null, resultsMock);

const mysqlMock = {

createConnection: sinon.stub().returns(connectionMock)

};

const app = proxyquire(SAMPLE_PATH, {

mysql: mysqlMock,

express: expressMock

});

return {

app: app,

mocks: {

express: expressMock,

results: resultsMock,

connection: connectionMock,

mysql: mysqlMock

}

};

}

test.beforeEach(stubConsole);

test.afterEach.always(restoreConsole);

test(`sets up sample`, (t) => {

const sample = getSample();

t.true(sample.mocks.express.calledOnce);

t.true(sample.mocks.mysql.createConnection.calledOnce);

t.deepEqual(sample.mocks.mysql.createConnection.firstCall.args[0], {

user: process.env.MYSQL_USER,

password: process.env.MYSQL_PASSWORD,

database: process.env.MYSQL_DATABASE

});

t.true(sample.app.listen.calledOnce);

t.is(sample.app.listen.firstCall.args[0], process.env.PORT || 8080);

});

test.cb(`should record a visit`, (t) => {

const sample = getSample();

const expectedResult = `Last 10 visits:\nTime: 1234, AddrHash: abcd`;

request(sample.app)

.get(`/`)

.expect(200)

.expect((response) => {

t.is(response.text, expectedResult);

})

.end(t.end);

});

test.cb(`should handle insert error`, (t) => {

const sample = getSample();

const expectedResult = `insert_error`;

sample.mocks.connection.query.onFirstCall().yields(expectedResult);

request(sample.app)

.get(`/`)

.expect(500)

.expect((response) => {

t.is(response.text, `${expectedResult}\n`);

})

.end(t.end);

});

test.cb(`should handle read error`, (t) => {

const sample = getSample();

const expectedResult = `read_error`;

sample.mocks.connection.query.onSecondCall().yields(expectedResult);

request(sample.app)

.get(`/`)

.expect(500)

.expect((response) => {

t.is(response.text, `${expectedResult}\n`);

})

.end(t.end);

});