def task1(url):
     with server() as (host, port):
         if url is None:
             url = f'http://{host}:{port}'
         r = get(url)
     r.raise_for_status()
     rs.append(r)
Example #2
0
    def test_binary_data(self):
        requests = []

        def request_sniper(x):
            requests.append(deepcopy(x))

        with server(callback=request_sniper):
            self.assertFalse(
                cli(
                    '--method',
                    'POST',
                    '--header',
                    'Content-Type: application/json',
                    '--data-binary',
                    '@{}'.format(self.binary_datafile.name),
                    '127.0.0.1',
                    'host.name/',
                ))

        self.assertIn(b'POST', requests[0])
        # Magic number is the val_size + val
        self.assertIn(b'CONTENT_LENGTH\x02\x0013', requests[0])
        self.assertIn(b'CONTENT_TYPE', requests[0])
        self.assertIn(b'application/json', requests[0])
        self.assertIn(b'binary-fooooo', requests[0])
Example #3
0
 def test_handler(self):
     req = mock.Mock(
         **{
             'makefile.return_value.readline.side_effect': [
                 b'GET / HTTP/1.1\r\n',
                 b'\r\n',
             ]
         })
     with server():
         proxy.RequestHandler(request=req,
                              client_address=['client', 0],
                              server=self.srv_mock)
Example #4
0
 def task():
     try:
         with server() as (host, port):
             r = get(f'http://{host}:{port}')
         r.raise_for_status()
         rs.append(r)
     except Exception as e:
         name = f'{e.__traceback__}{e}'
         if name in errors:
             errors[name] += 1
         else:
             errors[name] = 1
Example #5
0
    def test_post(self):
        requests = []

        def request_sniper(x):
            requests.append(deepcopy(x))

        with server(callback=request_sniper):
            self.assertFalse(
                cli('--method', 'POST', '--header',
                    'Content-Type: application/json',
                    r'''--data=\'{"foo":"bar"}\'''', '127.0.0.1',
                    'host.name/'))

        self.assertIn(b'POST', requests[0])
        # Magic number is the val_size + val
        self.assertIn(b'CONTENT_LENGTH\x02\x0017', requests[0])
        self.assertIn(b'CONTENT_TYPE', requests[0])
        self.assertIn(b'application/json', requests[0])
Example #6
0
 def test_headers(self):
     with server(callback=lambda x: self.assertIn(b'localhost', x)):
         self.assertFalse(cli('127.0.0.1:3030', '-H', 'Host: localhost'))
Example #7
0
    def test_file_socket(self):
        fname = '/tmp/unix-socket'

        with server(addr=fname, params=(socket.AF_UNIX, socket.SOCK_STREAM)):
            self.assertFalse(cli(fname))
Example #8
0
 def test_cli_nok(self):
     with server(status=300):
         self.assertTrue(cli('127.0.0.1:3030', '--timeout', '10'))
Example #9
0
 def test_cli(self):
     with server():
         self.assertFalse(cli('127.0.0.1', 'host.name/'))
 def task1():
     with server() as (host, port):
         r = get(f'http://{host}:{port}')
     r.raise_for_status()
     return r
def multi_task1():
    for _ in range(4):
        with server() as (host, port):
            r = get(f'http://{host}:{port}')
        r.raise_for_status()