diff --git a/.flake8 b/.flake8 new file mode 100644 index 0000000..5c86e2d --- /dev/null +++ b/.flake8 @@ -0,0 +1,5 @@ +[flake8] +max-line-length = 88 +extend-ignore = E203 +per-file-ignores= + test_dj_database_url.py: E501, E265 diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index a2d42f5..42105ab 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -5,21 +5,12 @@ jobs: runs-on: ubuntu-latest strategy: matrix: - python-version: [2.7, 3.5, 3.6, 3.7, 3.8] - django-version: [1.11, 2.0, 2.1, 2.2, 3.0] + python-version: ["3.7", "3.8", "3.9", "3.10"] + django-version: [3.2, 4.0] exclude: - # Python 2.7 is only compatible with Django 1.11 - - python-version: 2.7 - django-version: 2.0 - - python-version: 2.7 - django-version: 2.1 - - python-version: 2.7 - django-version: 2.2 - - python-version: 2.7 - django-version: 3.0 - # Python 3.5 is compatible with Django 1.11 to 2.2 but not 3.0 - - python-version: 3.5 - django-version: 3.0 + # Python 3.6 is not compatible with 4.0 + - python-version: 3.6 + django-version: 4.0 fail-fast: false steps: - uses: actions/checkout@v1 diff --git a/.isort.cfg b/.isort.cfg new file mode 100644 index 0000000..f238bf7 --- /dev/null +++ b/.isort.cfg @@ -0,0 +1,2 @@ +[settings] +profile = black diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml new file mode 100644 index 0000000..b5a6d27 --- /dev/null +++ b/.pre-commit-config.yaml @@ -0,0 +1,23 @@ +repos: + - repo: https://github.com/pre-commit/pre-commit-hooks + rev: v4.2.0 + hooks: + - id: end-of-file-fixer + - id: trailing-whitespace + + - repo: https://github.com/pycqa/isort + rev: "5.10.1" + hooks: + - id: isort + args: ["--profile", "black"] + + - repo: https://github.com/psf/black + rev: 22.3.0 + hooks: + - id: black + args: [--target-version=py38] + + - repo: https://github.com/pycqa/flake8 + rev: '4.0.1' + hooks: + - id: flake8 diff --git a/CODE_OF_CONDUCT.md b/CODE_OF_CONDUCT.md new file mode 100644 index 0000000..e0d5efa --- /dev/null +++ b/CODE_OF_CONDUCT.md @@ -0,0 +1,46 @@ +# Code of Conduct + +As contributors and maintainers of the Jazzband projects, and in the interest of +fostering an open and welcoming community, we pledge to respect all people who +contribute through reporting issues, posting feature requests, updating documentation, +submitting pull requests or patches, and other activities. + +We are committed to making participation in the Jazzband a harassment-free experience +for everyone, regardless of the level of experience, gender, gender identity and +expression, sexual orientation, disability, personal appearance, body size, race, +ethnicity, age, religion, or nationality. + +Examples of unacceptable behavior by participants include: + +- The use of sexualized language or imagery +- Personal attacks +- Trolling or insulting/derogatory comments +- Public or private harassment +- Publishing other's private information, such as physical or electronic addresses, + without explicit permission +- Other unethical or unprofessional conduct + +The Jazzband roadies have the right and responsibility to remove, edit, or reject +comments, commits, code, wiki edits, issues, and other contributions that are not +aligned to this Code of Conduct, or to ban temporarily or permanently any contributor +for other behaviors that they deem inappropriate, threatening, offensive, or harmful. + +By adopting this Code of Conduct, the roadies commit themselves to fairly and +consistently applying these principles to every aspect of managing the jazzband +projects. Roadies who do not follow or enforce the Code of Conduct may be permanently +removed from the Jazzband roadies. + +This code of conduct applies both within project spaces and in public spaces when an +individual is representing the project or its community. + +Instances of abusive, harassing, or otherwise unacceptable behavior may be reported by +contacting the roadies at `roadies@jazzband.co`. All complaints will be reviewed and +investigated and will result in a response that is deemed necessary and appropriate to +the circumstances. Roadies are obligated to maintain confidentiality with regard to the +reporter of an incident. + +This Code of Conduct is adapted from the [Contributor Covenant][homepage], version +1.3.0, available at [https://contributor-covenant.org/version/1/3/0/][version] + +[homepage]: https://contributor-covenant.org +[version]: https://contributor-covenant.org/version/1/3/0/ diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md new file mode 100644 index 0000000..829a22a --- /dev/null +++ b/CONTRIBUTING.md @@ -0,0 +1,7 @@ +[![Jazzband](https://jazzband.co/static/img/jazzband.svg)](https://jazzband.co/) + +This is a [Jazzband](https://jazzband.co/) project. By contributing you agree to abide by the [Contributor Code of Conduct](https://jazzband.co/about/conduct) and follow the [guidelines](https://jazzband.co/about/guidelines). + +Please see the +[full contributing documentation](https://django-debug-toolbar.readthedocs.io/en/stable/contributing.html) +for more help. diff --git a/LICENSE b/LICENSE index 7d98256..3aaf151 100644 --- a/LICENSE +++ b/LICENSE @@ -1,8 +1,27 @@ Copyright (c) Kenneth Reitz & individual contributors All rights reserved. -Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: +Redistribution and use in source and binary forms, with or without modification, +are permitted provided that the following conditions are met: -Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. -Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. -THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + 1. Redistributions of source code must retain the above copyright notice, + this list of conditions and the following disclaimer. + + 2. Redistributions in binary form must reproduce the above copyright + notice, this list of conditions and the following disclaimer in the + documentation and/or other materials provided with the distribution. + + 3. Neither the name of Django nor the names of its contributors may be used + to endorse or promote products derived from this software without + specific prior written permission. + +THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND +ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED +WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE +DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR +ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES +(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; +LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. diff --git a/README.rst b/README.rst index a4325a5..f8e4c48 100644 --- a/README.rst +++ b/README.rst @@ -1,6 +1,10 @@ DJ-Database-URL ~~~~~~~~~~~~~~~ +.. image:: https://jazzband.co/static/img/badge.png + :target: https://jazzband.co/ + :alt: Jazzband + .. image:: https://img.shields.io/endpoint.svg?url=https%3A%2F%2Factions-badge.atrox.dev%2Fjacobian%2Fdj-database-url%2Fbadge&style=for-the-badge :target: https://actions-badge.atrox.dev/jacobian/dj-database-url/goto @@ -92,3 +96,26 @@ URL schema from using other Oracle tools (like SQLPlus) i.e. user and password are separated by ``:`` not by ``/``. Also you can omit ``HOST`` and ``PORT`` and provide a full DSN string or TNS name in ``NAME`` part. + +Contributing +------------ + +We welcome contributions to this project. Projects can take two forms: + +1. Raising issues or helping others through the github issue tracker. +2. Contributing code. + +Raising Issues or helping others: +^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ + +When submitting an issue or helping other remember you are talking to humans who have feelings, jobs and lives of their +own. Be nice, be kind, be polite. Remember english may not be someone first language, if you do not understand or +something is not clear be polite and re-ask/ re-word. + +Contributing code: +^^^^^^^^^^^^^^^^^^ + +* Before writing code be sure to check existing PR's and issues in the tracker. +* Write code to the pylint spec. +* Large or wide sweeping changes will take longer, and may face more scrutiny than smaller confined changes. +* Code should be pass `black` and `flake8` validation. diff --git a/dj_database_url.py b/dj_database_url.py index c3ec891..9bd7661 100644 --- a/dj_database_url.py +++ b/dj_database_url.py @@ -14,51 +14,53 @@ except ImportError: # Register database schemes in URLs. -urlparse.uses_netloc.append('postgres') -urlparse.uses_netloc.append('postgresql') -urlparse.uses_netloc.append('pgsql') -urlparse.uses_netloc.append('postgis') -urlparse.uses_netloc.append('mysql') -urlparse.uses_netloc.append('mysql2') -urlparse.uses_netloc.append('mysqlgis') -urlparse.uses_netloc.append('mysql-connector') -urlparse.uses_netloc.append('mssql') -urlparse.uses_netloc.append('spatialite') -urlparse.uses_netloc.append('sqlite') -urlparse.uses_netloc.append('oracle') -urlparse.uses_netloc.append('oraclegis') -urlparse.uses_netloc.append('redshift') -urlparse.uses_netloc.append('cockroach') +urlparse.uses_netloc.append("postgres") +urlparse.uses_netloc.append("postgresql") +urlparse.uses_netloc.append("pgsql") +urlparse.uses_netloc.append("postgis") +urlparse.uses_netloc.append("mysql") +urlparse.uses_netloc.append("mysql2") +urlparse.uses_netloc.append("mysqlgis") +urlparse.uses_netloc.append("mysql-connector") +urlparse.uses_netloc.append("mssql") +urlparse.uses_netloc.append("spatialite") +urlparse.uses_netloc.append("sqlite") +urlparse.uses_netloc.append("oracle") +urlparse.uses_netloc.append("oraclegis") +urlparse.uses_netloc.append("redshift") +urlparse.uses_netloc.append("cockroach") -DEFAULT_ENV = 'DATABASE_URL' +DEFAULT_ENV = "DATABASE_URL" SCHEMES = { - 'postgis': 'django.contrib.gis.db.backends.postgis', - 'mysql': 'django.db.backends.mysql', - 'mysql2': 'django.db.backends.mysql', - 'mysqlgis': 'django.contrib.gis.db.backends.mysql', - 'mysql-connector': 'mysql.connector.django', - 'mssql': 'sql_server.pyodbc', - 'spatialite': 'django.contrib.gis.db.backends.spatialite', - 'sqlite': 'django.db.backends.sqlite3', - 'oracle': 'django.db.backends.oracle', - 'oraclegis': 'django.contrib.gis.db.backends.oracle', - 'redshift': 'django_redshift_backend', - 'cockroach': 'django_cockroachdb' + "postgis": "django.contrib.gis.db.backends.postgis", + "mysql": "django.db.backends.mysql", + "mysql2": "django.db.backends.mysql", + "mysqlgis": "django.contrib.gis.db.backends.mysql", + "mysql-connector": "mysql.connector.django", + "mssql": "sql_server.pyodbc", + "spatialite": "django.contrib.gis.db.backends.spatialite", + "sqlite": "django.db.backends.sqlite3", + "oracle": "django.db.backends.oracle", + "oraclegis": "django.contrib.gis.db.backends.oracle", + "redshift": "django_redshift_backend", + "cockroach": "django_cockroachdb", } # https://docs.djangoproject.com/en/2.0/releases/2.0/#id1 if DJANGO_VERSION and DJANGO_VERSION < (2, 0): - SCHEMES['postgres'] = 'django.db.backends.postgresql_psycopg2' - SCHEMES['postgresql'] = 'django.db.backends.postgresql_psycopg2' - SCHEMES['pgsql'] = 'django.db.backends.postgresql_psycopg2' + SCHEMES["postgres"] = "django.db.backends.postgresql_psycopg2" + SCHEMES["postgresql"] = "django.db.backends.postgresql_psycopg2" + SCHEMES["pgsql"] = "django.db.backends.postgresql_psycopg2" else: - SCHEMES['postgres'] = 'django.db.backends.postgresql' - SCHEMES['postgresql'] = 'django.db.backends.postgresql' - SCHEMES['pgsql'] = 'django.db.backends.postgresql' + SCHEMES["postgres"] = "django.db.backends.postgresql" + SCHEMES["postgresql"] = "django.db.backends.postgresql" + SCHEMES["pgsql"] = "django.db.backends.postgresql" -def config(env=DEFAULT_ENV, default=None, engine=None, conn_max_age=0, ssl_require=False): +def config( + env=DEFAULT_ENV, default=None, engine=None, conn_max_age=0, ssl_require=False +): """Returns configured DATABASE dictionary from DATABASE_URL.""" config = {} @@ -74,14 +76,11 @@ def config(env=DEFAULT_ENV, default=None, engine=None, conn_max_age=0, ssl_requi def parse(url, engine=None, conn_max_age=0, ssl_require=False): """Parses a database URL.""" - if url == 'sqlite://:memory:': + if url == "sqlite://:memory:": # this is a special case, because if we pass this URL into # urlparse, urlparse will choke trying to interpret "memory" # as a port number - return { - 'ENGINE': SCHEMES['sqlite'], - 'NAME': ':memory:' - } + return {"ENGINE": SCHEMES["sqlite"], "NAME": ":memory:"} # note: no other settings are required for sqlite # otherwise parse the url as normal @@ -91,69 +90,74 @@ def parse(url, engine=None, conn_max_age=0, ssl_require=False): # Split query strings from path. path = url.path[1:] - if '?' in path and not url.query: - path, query = path.split('?', 2) + if "?" in path and not url.query: + path, query = path.split("?", 2) else: path, query = path, url.query query = urlparse.parse_qs(query) # If we are using sqlite and we have no path, then assume we # want an in-memory database (this is the behaviour of sqlalchemy) - if url.scheme == 'sqlite' and path == '': - path = ':memory:' + if url.scheme == "sqlite" and path == "": + path = ":memory:" # Handle postgres percent-encoded paths. - hostname = url.hostname or '' - if '%2f' in hostname.lower(): + hostname = url.hostname or "" + if "%2f" in hostname.lower(): # Switch to url.netloc to avoid lower cased paths hostname = url.netloc if "@" in hostname: hostname = hostname.rsplit("@", 1)[1] if ":" in hostname: hostname = hostname.split(":", 1)[0] - hostname = hostname.replace('%2f', '/').replace('%2F', '/') + hostname = hostname.replace("%2f", "/").replace("%2F", "/") # Lookup specified engine. engine = SCHEMES[url.scheme] if engine is None else engine - port = (str(url.port) if url.port and engine in [SCHEMES['oracle'], SCHEMES['mssql']] - else url.port) + port = ( + str(url.port) + if url.port and engine in [SCHEMES["oracle"], SCHEMES["mssql"]] + else url.port + ) # Update with environment configuration. - config.update({ - 'NAME': urlparse.unquote(path or ''), - 'USER': urlparse.unquote(url.username or ''), - 'PASSWORD': urlparse.unquote(url.password or ''), - 'HOST': hostname, - 'PORT': port or '', - 'CONN_MAX_AGE': conn_max_age, - }) + config.update( + { + "NAME": urlparse.unquote(path or ""), + "USER": urlparse.unquote(url.username or ""), + "PASSWORD": urlparse.unquote(url.password or ""), + "HOST": hostname, + "PORT": port or "", + "CONN_MAX_AGE": conn_max_age, + } + ) # Pass the query string into OPTIONS. options = {} for key, values in query.items(): - if url.scheme == 'mysql' and key == 'ssl-ca': - options['ssl'] = {'ca': values[-1]} + if url.scheme == "mysql" and key == "ssl-ca": + options["ssl"] = {"ca": values[-1]} continue options[key] = values[-1] if ssl_require: - options['sslmode'] = 'require' + options["sslmode"] = "require" # Support for Postgres Schema URLs - if 'currentSchema' in options and engine in ( - 'django.contrib.gis.db.backends.postgis', - 'django.db.backends.postgresql_psycopg2', - 'django.db.backends.postgresql', - 'django_redshift_backend', + if "currentSchema" in options and engine in ( + "django.contrib.gis.db.backends.postgis", + "django.db.backends.postgresql_psycopg2", + "django.db.backends.postgresql", + "django_redshift_backend", ): - options['options'] = '-c search_path={0}'.format(options.pop('currentSchema')) + options["options"] = "-c search_path={0}".format(options.pop("currentSchema")) if options: - config['OPTIONS'] = options + config["OPTIONS"] = options if engine: - config['ENGINE'] = engine + config["ENGINE"] = engine return config diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..08ca2f5 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,2 @@ +[tool.black] +skip-string-normalization = 1 diff --git a/setup.py b/setup.py index 17f2d3a..5cafa50 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,7 @@ # -*- coding: utf-8 -*- from setuptools import setup -with open('README.rst') as readme_rst: +with open("README.rst") as readme_rst: readme = readme_rst.read() setup( diff --git a/test_dj_database_url.py b/test_dj_database_url.py index 3bab378..9aa6f23 100644 --- a/test_dj_database_url.py +++ b/test_dj_database_url.py @@ -1,6 +1,3 @@ -# -*- coding: utf-8 -*- -#!/usr/bin/env python - import os import unittest @@ -8,333 +5,341 @@ from django import VERSION as DJANGO_VERSION import dj_database_url - -POSTGIS_URL = 'postgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn' +POSTGIS_URL = "postgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn" # Django deprecated the `django.db.backends.postgresql_psycopg2` in 2.0. # https://docs.djangoproject.com/en/2.0/releases/2.0/#id1 -EXPECTED_POSTGRES_ENGINE = 'django.db.backends.postgresql' +EXPECTED_POSTGRES_ENGINE = "django.db.backends.postgresql" if DJANGO_VERSION < (2, 0): - EXPECTED_POSTGRES_ENGINE = 'django.db.backends.postgresql_psycopg2' + EXPECTED_POSTGRES_ENGINE = "django.db.backends.postgresql_psycopg2" class DatabaseTestSuite(unittest.TestCase): - def test_postgres_parsing(self): - url = 'postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn' + url = "postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn" url = dj_database_url.parse(url) - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 def test_postgres_unix_socket_parsing(self): - url = 'postgres://%2Fvar%2Frun%2Fpostgresql/d8r82722r2kuvn' + url = "postgres://%2Fvar%2Frun%2Fpostgresql/d8r82722r2kuvn" url = dj_database_url.parse(url) - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == '/var/run/postgresql' - assert url['USER'] == '' - assert url['PASSWORD'] == '' - assert url['PORT'] == '' + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "/var/run/postgresql" + assert url["USER"] == "" + assert url["PASSWORD"] == "" + assert url["PORT"] == "" - url = 'postgres://%2FUsers%2Fpostgres%2FRuN/d8r82722r2kuvn' + url = "postgres://%2FUsers%2Fpostgres%2FRuN/d8r82722r2kuvn" url = dj_database_url.parse(url) - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['HOST'] == '/Users/postgres/RuN' - assert url['USER'] == '' - assert url['PASSWORD'] == '' - assert url['PORT'] == '' + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["HOST"] == "/Users/postgres/RuN" + assert url["USER"] == "" + assert url["PASSWORD"] == "" + assert url["PORT"] == "" def test_ipv6_parsing(self): - url = 'postgres://ieRaekei9wilaim7:wegauwhgeuioweg@[2001:db8:1234::1234:5678:90af]:5431/d8r82722r2kuvn' + url = "postgres://ieRaekei9wilaim7:wegauwhgeuioweg@[2001:db8:1234::1234:5678:90af]:5431/d8r82722r2kuvn" url = dj_database_url.parse(url) - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == '2001:db8:1234::1234:5678:90af' - assert url['USER'] == 'ieRaekei9wilaim7' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "2001:db8:1234::1234:5678:90af" + assert url["USER"] == "ieRaekei9wilaim7" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 def test_postgres_search_path_parsing(self): - url = 'postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?currentSchema=otherschema' + url = "postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?currentSchema=otherschema" url = dj_database_url.parse(url) - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 - assert url['OPTIONS']['options'] == '-c search_path=otherschema' - assert 'currentSchema' not in url['OPTIONS'] + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 + assert url["OPTIONS"]["options"] == "-c search_path=otherschema" + assert "currentSchema" not in url["OPTIONS"] def test_postgres_parsing_with_special_characters(self): - url = 'postgres://%23user:%23password@ec2-107-21-253-135.compute-1.amazonaws.com:5431/%23database' + url = "postgres://%23user:%23password@ec2-107-21-253-135.compute-1.amazonaws.com:5431/%23database" url = dj_database_url.parse(url) - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['NAME'] == '#database' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == '#user' - assert url['PASSWORD'] == '#password' - assert url['PORT'] == 5431 + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["NAME"] == "#database" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "#user" + assert url["PASSWORD"] == "#password" + assert url["PORT"] == 5431 def test_postgis_parsing(self): - url = 'postgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn' + url = "postgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.contrib.gis.db.backends.postgis' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 + assert url["ENGINE"] == "django.contrib.gis.db.backends.postgis" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 def test_postgis_search_path_parsing(self): - url = 'postgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?currentSchema=otherschema' + url = "postgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?currentSchema=otherschema" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.contrib.gis.db.backends.postgis' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 - assert url['OPTIONS']['options'] == '-c search_path=otherschema' - assert 'currentSchema' not in url['OPTIONS'] + assert url["ENGINE"] == "django.contrib.gis.db.backends.postgis" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 + assert url["OPTIONS"]["options"] == "-c search_path=otherschema" + assert "currentSchema" not in url["OPTIONS"] def test_mysql_gis_parsing(self): - url = 'mysqlgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn' + url = "mysqlgis://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.contrib.gis.db.backends.mysql' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 + assert url["ENGINE"] == "django.contrib.gis.db.backends.mysql" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 def test_mysql_connector_parsing(self): - url = 'mysql-connector://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn' + url = "mysql-connector://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'mysql.connector.django' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 + assert url["ENGINE"] == "mysql.connector.django" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 def test_cleardb_parsing(self): - url = 'mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true' + url = "mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.db.backends.mysql' - assert url['NAME'] == 'heroku_97681db3eff7580' - assert url['HOST'] == 'us-cdbr-east.cleardb.com' - assert url['USER'] == 'bea6eb025ca0d8' - assert url['PASSWORD'] == '69772142' - assert url['PORT'] == '' + assert url["ENGINE"] == "django.db.backends.mysql" + assert url["NAME"] == "heroku_97681db3eff7580" + assert url["HOST"] == "us-cdbr-east.cleardb.com" + assert url["USER"] == "bea6eb025ca0d8" + assert url["PASSWORD"] == "69772142" + assert url["PORT"] == "" def test_database_url(self): - del os.environ['DATABASE_URL'] + del os.environ["DATABASE_URL"] a = dj_database_url.config() assert not a - os.environ['DATABASE_URL'] = 'postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn' + os.environ[ + "DATABASE_URL" + ] = "postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn" url = dj_database_url.config() - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 def test_empty_sqlite_url(self): - url = 'sqlite://' + url = "sqlite://" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.db.backends.sqlite3' - assert url['NAME'] == ':memory:' + assert url["ENGINE"] == "django.db.backends.sqlite3" + assert url["NAME"] == ":memory:" def test_memory_sqlite_url(self): - url = 'sqlite://:memory:' + url = "sqlite://:memory:" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.db.backends.sqlite3' - assert url['NAME'] == ':memory:' + assert url["ENGINE"] == "django.db.backends.sqlite3" + assert url["NAME"] == ":memory:" def test_parse_engine_setting(self): - engine = 'django_mysqlpool.backends.mysqlpool' - url = 'mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true' + engine = "django_mysqlpool.backends.mysqlpool" + url = "mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true" url = dj_database_url.parse(url, engine) - assert url['ENGINE'] == engine + assert url["ENGINE"] == engine def test_config_engine_setting(self): - engine = 'django_mysqlpool.backends.mysqlpool' - os.environ['DATABASE_URL'] = 'mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true' + engine = "django_mysqlpool.backends.mysqlpool" + os.environ[ + "DATABASE_URL" + ] = "mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true" url = dj_database_url.config(engine=engine) - assert url['ENGINE'] == engine + assert url["ENGINE"] == engine def test_parse_conn_max_age_setting(self): conn_max_age = 600 - url = 'mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true' + url = "mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true" url = dj_database_url.parse(url, conn_max_age=conn_max_age) - assert url['CONN_MAX_AGE'] == conn_max_age + assert url["CONN_MAX_AGE"] == conn_max_age def test_config_conn_max_age_setting(self): conn_max_age = 600 - os.environ['DATABASE_URL'] = 'mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true' + os.environ[ + "DATABASE_URL" + ] = "mysql://bea6eb025ca0d8:69772142@us-cdbr-east.cleardb.com/heroku_97681db3eff7580?reconnect=true" url = dj_database_url.config(conn_max_age=conn_max_age) - assert url['CONN_MAX_AGE'] == conn_max_age + assert url["CONN_MAX_AGE"] == conn_max_age def test_database_url_with_options(self): # Test full options - os.environ['DATABASE_URL'] = 'postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?sslrootcert=rds-combined-ca-bundle.pem&sslmode=verify-full' + os.environ[ + "DATABASE_URL" + ] = "postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?sslrootcert=rds-combined-ca-bundle.pem&sslmode=verify-full" url = dj_database_url.config() - assert url['ENGINE'] == EXPECTED_POSTGRES_ENGINE - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5431 - assert url['OPTIONS'] == { - 'sslrootcert': 'rds-combined-ca-bundle.pem', - 'sslmode': 'verify-full' + assert url["ENGINE"] == EXPECTED_POSTGRES_ENGINE + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5431 + assert url["OPTIONS"] == { + "sslrootcert": "rds-combined-ca-bundle.pem", + "sslmode": "verify-full", } # Test empty options - os.environ['DATABASE_URL'] = 'postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?' + os.environ[ + "DATABASE_URL" + ] = "postgres://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5431/d8r82722r2kuvn?" url = dj_database_url.config() - assert 'OPTIONS' not in url + assert "OPTIONS" not in url def test_mysql_database_url_with_sslca_options(self): - os.environ['DATABASE_URL'] = 'mysql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:3306/d8r82722r2kuvn?ssl-ca=rds-combined-ca-bundle.pem' + os.environ[ + "DATABASE_URL" + ] = "mysql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:3306/d8r82722r2kuvn?ssl-ca=rds-combined-ca-bundle.pem" url = dj_database_url.config() - assert url['ENGINE'] == 'django.db.backends.mysql' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 3306 - assert url['OPTIONS'] == { - 'ssl': { - 'ca': 'rds-combined-ca-bundle.pem' - } - } + assert url["ENGINE"] == "django.db.backends.mysql" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 3306 + assert url["OPTIONS"] == {"ssl": {"ca": "rds-combined-ca-bundle.pem"}} # Test empty options - os.environ['DATABASE_URL'] = 'mysql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:3306/d8r82722r2kuvn?' + os.environ[ + "DATABASE_URL" + ] = "mysql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:3306/d8r82722r2kuvn?" url = dj_database_url.config() - assert 'OPTIONS' not in url + assert "OPTIONS" not in url def test_oracle_parsing(self): - url = 'oracle://scott:tiger@oraclehost:1521/hr' + url = "oracle://scott:tiger@oraclehost:1521/hr" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.db.backends.oracle' - assert url['NAME'] == 'hr' - assert url['HOST'] == 'oraclehost' - assert url['USER'] == 'scott' - assert url['PASSWORD'] == 'tiger' - assert url['PORT'] == '1521' + assert url["ENGINE"] == "django.db.backends.oracle" + assert url["NAME"] == "hr" + assert url["HOST"] == "oraclehost" + assert url["USER"] == "scott" + assert url["PASSWORD"] == "tiger" + assert url["PORT"] == "1521" def test_oracle_gis_parsing(self): - url = 'oraclegis://scott:tiger@oraclehost:1521/hr' + url = "oraclegis://scott:tiger@oraclehost:1521/hr" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.contrib.gis.db.backends.oracle' - assert url['NAME'] == 'hr' - assert url['HOST'] == 'oraclehost' - assert url['USER'] == 'scott' - assert url['PASSWORD'] == 'tiger' - assert url['PORT'] == 1521 + assert url["ENGINE"] == "django.contrib.gis.db.backends.oracle" + assert url["NAME"] == "hr" + assert url["HOST"] == "oraclehost" + assert url["USER"] == "scott" + assert url["PASSWORD"] == "tiger" + assert url["PORT"] == 1521 def test_oracle_dsn_parsing(self): url = ( - 'oracle://scott:tiger@/' - '(DESCRIPTION=(ADDRESS_LIST=(ADDRESS=(PROTOCOL=TCP)' - '(HOST=oraclehost)(PORT=1521)))' - '(CONNECT_DATA=(SID=hr)))' + "oracle://scott:tiger@/" + "(DESCRIPTION=(ADDRESS_LIST=(ADDRESS=(PROTOCOL=TCP)" + "(HOST=oraclehost)(PORT=1521)))" + "(CONNECT_DATA=(SID=hr)))" ) url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.db.backends.oracle' - assert url['USER'] == 'scott' - assert url['PASSWORD'] == 'tiger' - assert url['HOST'] == '' - assert url['PORT'] == '' + assert url["ENGINE"] == "django.db.backends.oracle" + assert url["USER"] == "scott" + assert url["PASSWORD"] == "tiger" + assert url["HOST"] == "" + assert url["PORT"] == "" dsn = ( - '(DESCRIPTION=(ADDRESS_LIST=(ADDRESS=(PROTOCOL=TCP)' - '(HOST=oraclehost)(PORT=1521)))' - '(CONNECT_DATA=(SID=hr)))' + "(DESCRIPTION=(ADDRESS_LIST=(ADDRESS=(PROTOCOL=TCP)" + "(HOST=oraclehost)(PORT=1521)))" + "(CONNECT_DATA=(SID=hr)))" ) - assert url['NAME'] == dsn + assert url["NAME"] == dsn def test_oracle_tns_parsing(self): - url = 'oracle://scott:tiger@/tnsname' + url = "oracle://scott:tiger@/tnsname" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django.db.backends.oracle' - assert url['USER'] == 'scott' - assert url['PASSWORD'] == 'tiger' - assert url['NAME'] == 'tnsname' - assert url['HOST'] == '' - assert url['PORT'] == '' + assert url["ENGINE"] == "django.db.backends.oracle" + assert url["USER"] == "scott" + assert url["PASSWORD"] == "tiger" + assert url["NAME"] == "tnsname" + assert url["HOST"] == "" + assert url["PORT"] == "" def test_redshift_parsing(self): - url = 'redshift://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5439/d8r82722r2kuvn?currentSchema=otherschema' + url = "redshift://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com:5439/d8r82722r2kuvn?currentSchema=otherschema" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'django_redshift_backend' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == 5439 - assert url['OPTIONS']['options'] == '-c search_path=otherschema' - assert 'currentSchema' not in url['OPTIONS'] + assert url["ENGINE"] == "django_redshift_backend" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == 5439 + assert url["OPTIONS"]["options"] == "-c search_path=otherschema" + assert "currentSchema" not in url["OPTIONS"] def test_mssql_parsing(self): - url = 'mssql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com/d8r82722r2kuvn?driver=ODBC Driver 13 for SQL Server' + url = "mssql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com/d8r82722r2kuvn?driver=ODBC Driver 13 for SQL Server" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'sql_server.pyodbc' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == '' - assert url['OPTIONS']['driver'] == 'ODBC Driver 13 for SQL Server' - assert 'currentSchema' not in url['OPTIONS'] + assert url["ENGINE"] == "sql_server.pyodbc" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == "" + assert url["OPTIONS"]["driver"] == "ODBC Driver 13 for SQL Server" + assert "currentSchema" not in url["OPTIONS"] def test_mssql_instance_port_parsing(self): - url = 'mssql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com\\insnsnss:12345/d8r82722r2kuvn?driver=ODBC Driver 13 for SQL Server' + url = "mssql://uf07k1i6d8ia0v:wegauwhgeuioweg@ec2-107-21-253-135.compute-1.amazonaws.com\\insnsnss:12345/d8r82722r2kuvn?driver=ODBC Driver 13 for SQL Server" url = dj_database_url.parse(url) - assert url['ENGINE'] == 'sql_server.pyodbc' - assert url['NAME'] == 'd8r82722r2kuvn' - assert url['HOST'] == 'ec2-107-21-253-135.compute-1.amazonaws.com\\insnsnss' - assert url['USER'] == 'uf07k1i6d8ia0v' - assert url['PASSWORD'] == 'wegauwhgeuioweg' - assert url['PORT'] == '12345' - assert url['OPTIONS']['driver'] == 'ODBC Driver 13 for SQL Server' - assert 'currentSchema' not in url['OPTIONS'] + assert url["ENGINE"] == "sql_server.pyodbc" + assert url["NAME"] == "d8r82722r2kuvn" + assert url["HOST"] == "ec2-107-21-253-135.compute-1.amazonaws.com\\insnsnss" + assert url["USER"] == "uf07k1i6d8ia0v" + assert url["PASSWORD"] == "wegauwhgeuioweg" + assert url["PORT"] == "12345" + assert url["OPTIONS"]["driver"] == "ODBC Driver 13 for SQL Server" + assert "currentSchema" not in url["OPTIONS"] def test_cockroach(self): url = "cockroach://testuser:testpass@testhost:26257/cockroach?sslmode=verify-full&sslrootcert=/certs/ca.crt&sslcert=/certs/client.myprojectuser.crt&sslkey=/certs/client.myprojectuser.key" @@ -351,5 +356,5 @@ class DatabaseTestSuite(unittest.TestCase): assert url['OPTIONS']['sslkey'] == '/certs/client.myprojectuser.key' -if __name__ == '__main__': +if __name__ == "__main__": unittest.main()