random_pages.py

import os
import random
import time

import requests


class Transaction(object):
    def __init__(self):
        self.custom_timers = {}
        self.base_url = 'http://blah.compute-1.amazonaws.com'
        self.headers = {'Host': 'www.treehugger.com'}

        script_dir = os.path.abspath(os.path.dirname(__file__))
        urls_file = os.path.join(script_dir, 'urls.txt')
        f = open(urls_file, 'r')
        self.urls = f.read().splitlines()
        f.close()

    def get_random_url(self, urls=None):
        if urls is None:
            urls = self.urls
        return self.base_url + random.choice(urls)

    def run(self):
        start = time.time()
        r = requests.get(self.base_url + '/', headers=self.headers)
        latency = time.time() - start
        self.custom_timers['Home_Page'] = latency

        assert (r.status_code == requests.codes.ok), 'Bad HTTP Response'

        time.sleep(1)

        url = self.get_random_url()
        start = time.time()
        r = requests.get(url, headers=self.headers)
        latency = time.time() - start
        self.custom_timers['Random_URL_Large_Pool'] = latency

        assert (r.status_code == requests.codes.ok), 'Bad HTTP Response'

        time.sleep(2)