Skip to content
This repository
Fetching contributors…

Octocat-spinner-32-eaf2f5

Cannot retrieve contributors at this time

file 48 lines (37 sloc) 1.221 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
from base import *

MAGIC = "This QA tests the GeoIP rule module"
DIR = "GeoIP1_nomatch"
FILE = "test"

CONF = """
vserver!1!rule!1670!match = directory
vserver!1!rule!1670!match!directory = /%s
vserver!1!rule!1670!handler = cgi

vserver!1!rule!1671!match = and
vserver!1!rule!1671!match!left = geoip
vserver!1!rule!1671!match!left!countries = ES,US,UK,CA
vserver!1!rule!1671!match!right = directory
vserver!1!rule!1671!match!right!directory = /%s
vserver!1!rule!1671!handler = file
"""

CGI = """#!/bin/sh

echo "Content-Type: text/plain"
echo
echo "%s"
""" % (MAGIC)

class Test (TestBase):
    def __init__ (self):
        TestBase.__init__ (self, __file__)
        self.name = "GeoIP: no match"

        self.request = "GET /%s/%s HTTP/1.0\r\n" % (DIR, FILE)
        self.expected_error = 200
        self.expected_content = MAGIC
        self.forbidden_content = ["/bin/sh", "echo"]

    def Precondition (self):
        # Check that pam module was compiled
        if not cherokee_has_plugin("geoip"):
            return False
        return True

    def Prepare (self, www):
        self.conf = CONF % (DIR, DIR)

        d = self.Mkdir (www, DIR)
        f = self.WriteFile (d, FILE, 0755, CGI)
Something went wrong with that request. Please try again.