framework/test/warmup_test.py

# Copyright 2017 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style
# license that can be found in the LICENSE file or at
# https://developers.google.com/open-source/licenses/bsd

"""Tests for the warmup servlet."""
from __future__ import print_function
from __future__ import division
from __future__ import absolute_import

import unittest

import flask

from framework import warmup


class WarmupTest(unittest.TestCase):

  def testHandleWarmup(self):
    app = flask.Flask(__name__)
    app.add_url_rule('/', view_func=warmup.Warmup)

    with app.test_client() as client:
      response = client.get('/')

    self.assertEqual(response.status_code, 200)
    self.assertEqual(response.data, '')

  def testHandleStart(self):
    app = flask.Flask(__name__)
    app.add_url_rule('/', view_func=warmup.Start)

    with app.test_client() as client:
      response = client.get('/')

    self.assertEqual(response.status_code, 200)
    self.assertEqual(response.data, '')

  def testHandleStop(self):
    app = flask.Flask(__name__)
    app.add_url_rule('/', view_func=warmup.Stop)

    with app.test_client() as client:
      response = client.get('/')

    self.assertEqual(response.status_code, 200)
    self.assertEqual(response.data, '')