unblocker/test/unblocker_spec.js

Web proxy for evading internet censorship & general-purpose library for rewriting remote websites.

var fs = require('fs'),
    concat = require('concat-stream'),
    test = require('tap').test,
    hyperquest = require('hyperquest'),
    getServers = require('./test_utils.js').getServers;

var source = fs.readFileSync(__dirname + '/source/index.html');
var expected = fs.readFileSync(__dirname + '/expected/index.html');


test("url_rewriting should support support all kinds of links", function(t) {
    getServers(source, function(err, servers) {
        function cleanup() {
            servers.kill(function() {
                t.end();
            });
        }
        hyperquest(servers.proxiedUrl)
            .pipe(concat(function(data) {
                t.equal(data.toString(), expected.toString().replace(/<remotePort>/g, servers.remotePort));
                cleanup();
            }))
            .on('error', function(err) {
                console.error('error retrieving data from proxy', err);
                cleanup();
            });
    });
});


test("should return control to parent when route doesn't match and no referer is sent", function(t) {
    getServers(source, function(err, servers) {
        function cleanup() {
            servers.kill(function() {
                t.end();
            });
        }
        hyperquest(servers.homeUrl)
            .pipe(concat(function(data) {
                t.equal(data.toString(), 'this is the home page', servers.remotePort);
                cleanup();
            }))
            .on('error', function(err) {
                console.error('error retrieving robots.txt from proxy', err);
                cleanup();
            });
    });
});

test("should redirect root-relative urls when the correct target can be determined from the referer header", function(t) {

    getServers(source, function(err, servers) {
        function cleanup() {
            servers.kill(function() {
                t.end();
            });
        }
        hyperquest(servers.homeUrl + 'bar', {
            headers: {
                'referer': servers.proxiedUrl + 'foo'
            }
        }, function(err, res) {
            t.notOk(err);
            t.equal(res.statusCode, 307, 'http status code');
            t.equal(res.headers.location, servers.proxiedUrl + 'bar', 'redirect location');
            cleanup();
        });
    });
});


test("should redirect root-relative urls when the correct target can be determined from the referer header including for urls that the site is already serving content on", function(t) {
    getServers(source, function(err, servers) {
        function cleanup() {
            servers.kill(function() {
                t.end();
            });
        }
        hyperquest(servers.homeUrl, {
            headers: {
                'referer': servers.proxiedUrl
            }
        }, function(err, res) {
            t.notOk(err);
            t.equal(res.statusCode, 307, 'http status code');
            t.equal(res.headers.location, servers.proxiedUrl, 'redirect location');
            cleanup();
        });
    });
});


test("should redirect http urls that have had the slashes merged (http:/ instead of http://", function(t) {
    getServers(source, function(err, servers) {
        function cleanup() {
            servers.kill(function() {
                t.end();
            });
        }
        hyperquest(servers.proxiedUrl.replace('/proxy/http://','/proxy/http:/'), function(err, res) {
            t.notOk(err);
            t.equal(res.statusCode, 307, 'http status code');
            t.equal(res.headers.location, servers.proxiedUrl, 'redirect location');
            cleanup();
        });
    });
});

test("should redirect http urls that have had the have two occurrences of /prefix/http://", function(t) {
    getServers(source, function(err, servers) {
        function cleanup() {
            servers.kill(function() {
                t.end();
            });
        }
        hyperquest(servers.proxiedUrl.replace('/proxy/http://','/proxy/http://proxy/http://'), function(err, res) {
            t.notOk(err);
            t.equal(res.statusCode, 307, 'http status code');
            t.equal(res.headers.location, servers.proxiedUrl, 'redirect location');
            cleanup();
        });
    });
});