txfake server testing

test_server.py

import json
import treq

from twisted.internet.defer import inlineCallbacks
from twisted.trial.unittest import TestCase
from twisted.web.client import ProxyAgent
from twisted.web.server import Site

from txfake import FakeServer

from uritools import uricompose

from consular.server import MarathonEventServer


class MarathonEventServerTest(TestCase):

    def setUp(self):
        self.event_server = MarathonEventServer({})
        fake_server = FakeServer(Site(self.event_server.app.resource()))
        self.agent = ProxyAgent(fake_server.endpoint)

    def request(self, method, path, query=None, json_data=None):
        url = uricompose('http', 'www.example.com', path, query)
        data = json.dumps(json_data) if json_data is not None else None
        treq_kwargs = {
            'data': data,
            'agent': self.agent
        }

        return treq.request(method, url, **treq_kwargs)

    @inlineCallbacks
    def test_index(self):
        response = yield self.request('GET', '/')
        print response
        response_json = yield response.json()

        self.assertEqual(response_json, [])