summaryrefslogtreecommitdiffstats
path: root/tests/client.py
blob: 39dad6149ea1b239c42bcb07ae09733217695c26 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# Copyright (C) 2015  Custodia Project Contributors - see LICENSE file

try:
    # pylint: disable=import-error
    from httplib import HTTPConnection
except ImportError:
    # pylint: disable=import-error,no-name-in-module
    from http.client import HTTPConnection
import socket


class LocalConnection(HTTPConnection):

    def __init__(self, path):
        HTTPConnection.__init__(self, 'localhost', 0)
        self.unix_socket = path

    def connect(self):
        s = socket.socket(family=socket.AF_UNIX)
        s.connect(self.unix_socket)
        self.sock = s