def test_no_password(self, dsn): dsn2 = make_conninfo(dsn, password="******") pgconn = psycopg.pq.PGconn.connect_start(dsn2.encode("utf8")) info = ConnectionInfo(pgconn) assert info.password == "the-pass-word" assert "password" not in info.get_parameters() assert info.get_parameters()["dbname"] == info.dbname
def test_dsn_no_password(self, dsn): dsn2 = make_conninfo(dsn, password="******") pgconn = psycopg.pq.PGconn.connect_start(dsn2.encode()) info = ConnectionInfo(pgconn) assert info.password == "the-pass-word" assert "password" not in info.dsn assert f"dbname={info.dbname}" in info.dsn
def __init__(self, server_dsn): cdict = conninfo.conninfo_to_dict(server_dsn) # Get server params self.server_port = cdict.get("port", "5432") if "host" not in cdict or cdict["host"].startswith("/"): self.server_host = "localhost" else: self.server_host = cdict["host"] # Get client params self.client_host = "localhost" self.client_port = self._get_random_port() # Make a connection string to the proxy cdict["host"] = self.client_host cdict["port"] = self.client_port cdict["sslmode"] = "disable" # not supported by the proxy self.client_dsn = conninfo.make_conninfo(**cdict) # The running proxy process self.proc = None
async def test_get_connection_params(dsn, kwargs, exp): params = await AsyncConnection._get_connection_params(dsn, **kwargs) conninfo = make_conninfo(**params) assert conninfo_to_dict(conninfo) == exp[0] assert params["connect_timeout"] == exp[1]
def test_get_connection_params(dsn, kwargs, exp): params = Connection._get_connection_params(dsn, **kwargs) conninfo = make_conninfo(**params) assert conninfo_to_dict(conninfo) == exp[0] assert params.get("connect_timeout") == exp[1]
def test_no_munging(): dsnin = "dbname=a host=b user=c password=d" dsnout = make_conninfo(dsnin) assert dsnin == dsnout
def test_make_conninfo_bad(conninfo, kwargs): with pytest.raises(ProgrammingError): make_conninfo(conninfo, **kwargs)
def test_make_conninfo(conninfo, kwargs, exp): out = make_conninfo(conninfo, **kwargs) assert conninfo_to_dict(out) == conninfo_to_dict(exp)