from unittest import TestCase
try:
    from urllib import urlopen
except ImportError:
    from urllib.request import urlopen
from test_server import TestServer

class UrllibTestCase(TestCase):
    @classmethod
    def setUpClass(cls):
        cls.server = TestServer()
        cls.server.start()

    @classmethod
    def tearDownClass(cls):
        cls.server.stop()

    def setUp(self):
        self.server.reset()

    def test_get(self):
        token = b'zorro'
        self.server.response['data'] = token
        data = urlopen(self.server.base_url).read()
        self.assertEqual(data, token)