e2e_test.py

# Copyright 2015, 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.

""" Do some testing """

import urllib2 # we validate that a website responds properly
import sys # control exit codes
import json # validate json
import datetime # figure out year to have dynamic year testing

HOST = 'https://testing-dot-wasthereannhlgamelastnight.appspot.com'

YESNO = ["YES\n", "NO\n"]

NOW = datetime.datetime.now()
THIS_YEAR = NOW.year
LAST_YEAR = NOW.year - 1
NEXT_YEAR = NOW.year + 1
BOTH_YEARS = [str(THIS_YEAR), str(LAST_YEAR)]

# This is a list of the basic tests / arguments.
YESNODATES = [
    str(THIS_YEAR) + '1013',
    str(LAST_YEAR) + "1013",
    str(NEXT_YEAR) + "0316",
    "wingS/" + str(LAST_YEAR) + "1014",
    "wingS/" + str(NEXT_YEAR) + "0315",
    str(NEXT_YEAR) + "0315" + "/wingS",
    '',
    'WINGS',
    'Lak',
    'travis_e2e_test']

# First we define two special URIs where we do some extra testing
ARGS = {
    'update_schedule':  {"test": ['accounts.google.com'], "type": "in"},
    'get_schedule':     {"test": ['teamdates'], "type": "injson"},
    'version':     {"test": BOTH_YEARS, "type": "in"},
}

# Add the "basic" tests where we should only get a YES or NO
for date in YESNODATES:
    ARGS[date] = {"test": YESNO}

for arg in ARGS:
    try:
        response = urllib2.urlopen("{}/%s".format(HOST) % arg)
    except urllib2.HTTPError as urlliberror:
        print "Cannot fetch URL: %s" % urlliberror
        sys.exit(66)
    html = response.read()

    if ARGS[arg]['test'] == YESNO:
        print "asserting %s/%s - response code: %s" % (HOST, arg, response.code)
        try:
            assert html == "NO\n" or html == "YES\n"
        except AssertionError:
            print "%s/%s does not contain %s" % (HOST, arg, "YES\n or NO\n")
            sys.exit(3)

    else:
        print "asserting %s/%s contains %s - response code: %s" % (HOST, arg, ARGS[arg]['test'], response.code) # pylint: disable=line-too-long
        try:
            if ARGS[arg]['type'] == "in" or ARGS[arg]['type'] == "injson":
                # this any loops over tests in ARGS[arg]['test']). There's also an all()
                assert any(anarg in html for anarg in ARGS[arg]['test'])
        except AssertionError:
            print "%s/%s does not contain %s" % (HOST, arg, ARGS[arg])
            sys.exit(4)
        if ARGS[arg]['type'] == "injson":
            try:
                assert json.loads(html)['teamdates'].popitem()
            except KeyError:
                print "popitem of JSON on %s/%s key %s failed" % (HOST, arg, ARGS[arg]['test'])
                sys.exit(6)