]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
Adjust test_pgconn::test_info() for mypy
authorDenis Laxalde <denis.laxalde@dalibo.com>
Tue, 2 Nov 2021 09:26:22 +0000 (10:26 +0100)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Wed, 3 Nov 2021 15:53:05 +0000 (16:53 +0100)
Resolves the following error:
  error: Item "None" of "Optional[bytes]" has no attribute "decode"

tests/pq/test_pgconn.py

index 4f213fef524b62c273ef4e6905ee137f0c5016aa..294fed67188e283594bbf5f633e4b1e05bfe4096 100644 (file)
@@ -118,12 +118,12 @@ def test_info(dsn, pgconn):
     name = [
         o.val or os.environ.get(o.envvar.decode(), "").encode()
         for o in parsed
-        if o.keyword == b"dbname"
+        if o.keyword == b"dbname" and o.envvar
     ][0]
     user = [
         o.val or os.environ.get(o.envvar.decode(), "").encode()
         for o in parsed
-        if o.keyword == b"user"
+        if o.keyword == b"user" and o.envvar
     ][0]
     assert dbname.val == (name or user)