from twisted.trial.unittest import (
    TestCase,
)
from twisted.internet.defer import (
    inlineCallbacks,
)

from allmydata.testing.web import (
    create_tahoe_treq_client,
)


class FakeWebTest(TestCase):
    """
    Test the WebUI verified-fakes infrastucture
    """

    def setUp(self):
        super(FakeWebTest, self).setUp()
        self.http_client = create_tahoe_treq_client()

    @inlineCallbacks
    def test_create_and_download(self):
        """
        Upload some content and then download it
        """
        content = "fake data\n" * 100
        resp = yield self.http_client.put("http://example.com/uri", content)
        cap = yield resp.content()

        self.assertTrue(cap.startswith("URI:CHK:"))

        resp = yield self.http_client.get(
            "http://example.com/uri?uri={}".format(cap)
        )
        round_trip_content = yield resp.content()

        self.assertEqual(content, round_trip_content)