selenium-webdriver
Version:
The official WebDriver JavaScript bindings from the Selenium project
182 lines (155 loc) • 6.58 kB
JavaScript
// Licensed to the Software Freedom Conservancy (SFC) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The SFC licenses this file
// to you 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.
var assert = require('assert');
var http = require('http');
var url = require('url');
var HttpClient = require('../../http').HttpClient;
var HttpRequest = require('../../_base').require('webdriver.http.Request');
var Server = require('../../lib/test/httpserver').Server;
var promise = require('../..').promise;
var test = require('../../lib/test');
describe('HttpClient', function() {
this.timeout(4 * 1000);
var server = new Server(function(req, res) {
if (req.method == 'GET' && req.url == '/echo') {
res.writeHead(200, req.headers);
res.end();
} else if (req.method == 'GET' && req.url == '/redirect') {
res.writeHead(303, {'Location': server.url('/hello')});
res.end();
} else if (req.method == 'GET' && req.url == '/hello') {
res.writeHead(200, {'content-type': 'text/plain'});
res.end('hello, world!');
} else if (req.method == 'GET' && req.url == '/badredirect') {
res.writeHead(303, {});
res.end();
} else if (req.method == 'GET' && req.url == '/protected') {
var denyAccess = function() {
res.writeHead(401, {'WWW-Authenticate': 'Basic realm="test"'});
res.end('Access denied');
};
var basicAuthRegExp = /^\s*basic\s+([a-z0-9\-\._~\+\/]+)=*\s*$/i
var auth = req.headers.authorization;
var match = basicAuthRegExp.exec(auth || '');
if (!match) {
denyAccess();
return;
}
var userNameAndPass = new Buffer(match[1], 'base64').toString();
var parts = userNameAndPass.split(':', 2);
if (parts[0] !== 'genie' && parts[1] !== 'bottle') {
denyAccess();
return;
}
res.writeHead(200, {'content-type': 'text/plain'});
res.end('Access granted!');
} else if (req.method == 'GET' && req.url == '/proxy') {
res.writeHead(200, req.headers);
res.end();
} else if (req.method == 'GET' && req.url == '/proxy/redirect') {
res.writeHead(303, {'Location': '/proxy'});
res.end();
} else {
res.writeHead(404, {});
res.end();
}
});
test.before(function() {
return server.start();
});
test.after(function() {
return server.stop();
});
test.it('can send a basic HTTP request', function() {
var request = new HttpRequest('GET', '/echo');
request.headers['Foo'] = 'Bar';
var agent = new http.Agent();
agent.maxSockets = 1; // Only making 1 request.
var client = new HttpClient(server.url(), agent);
return promise.checkedNodeCall(client.send.bind(client, request))
.then(function(response) {
assert.equal(200, response.status);
assert.equal(
'application/json; charset=utf-8', response.headers['accept']);
assert.equal('Bar', response.headers['foo']);
assert.equal('0', response.headers['content-length']);
assert.equal('keep-alive', response.headers['connection']);
assert.equal(server.host(), response.headers['host']);
});
});
test.it('can use basic auth', function() {
var parsed = url.parse(server.url());
parsed.auth = 'genie:bottle';
var client = new HttpClient(url.format(parsed));
var request = new HttpRequest('GET', '/protected');
return promise.checkedNodeCall(client.send.bind(client, request))
.then(function(response) {
assert.equal(200, response.status);
assert.equal('text/plain', response.headers['content-type']);
assert.equal('Access granted!', response.body);
});
});
test.it('fails requests missing required basic auth', function() {
var client = new HttpClient(server.url());
var request = new HttpRequest('GET', '/protected');
return promise.checkedNodeCall(client.send.bind(client, request))
.then(function(response) {
assert.equal(401, response.status);
assert.equal('Access denied', response.body);
});
});
test.it('automatically follows redirects', function() {
var request = new HttpRequest('GET', '/redirect');
var client = new HttpClient(server.url());
return promise.checkedNodeCall(client.send.bind(client, request))
.then(function(response) {
assert.equal(200, response.status);
assert.equal('text/plain', response.headers['content-type']);
assert.equal('hello, world!', response.body);
});
});
test.it('handles malformed redirect responses', function() {
var request = new HttpRequest('GET', '/badredirect');
var client = new HttpClient(server.url());
return promise.checkedNodeCall(client.send.bind(client, request))
.thenCatch(function(err) {
assert.ok(/Failed to parse "Location"/.test(err.message),
'Not the expected error: ' + err.message);
});
});
test.it('proxies requests through the webdriver proxy', function() {
var request = new HttpRequest('GET', '/proxy');
var client = new HttpClient(
'http://another.server.com', undefined, server.url());
return promise.checkedNodeCall(client.send.bind(client, request))
.then(function(response) {
assert.equal(200, response.status);
assert.equal('another.server.com', response.headers['host']);
});
});
test.it(
'proxies requests through the webdriver proxy on redirect', function() {
var request = new HttpRequest('GET', '/proxy/redirect');
var client = new HttpClient(
'http://another.server.com', undefined, server.url());
return promise.checkedNodeCall(client.send.bind(client, request))
.then(function(response) {
assert.equal(200, response.status);
assert.equal('another.server.com', response.headers['host']);
});
});
});