Skip to content

Commit

Permalink
fix linting issues
Browse files Browse the repository at this point in the history
  • Loading branch information
@work-mohit
work-mohit committed Oct 2, 2023
1 parent 6175ab7 commit 61498f0
Showing 1 changed file with 128 additions and 133 deletions.
261 changes: 128 additions & 133 deletions tests/mocked/test_databases.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,23 +7,20 @@

import uuid


@responses.activate
def test_databases_add(mock_client: Client, mock_client_url):
"""Mocks the databases add database operation."""

expected = {
"db": {
"name": "alpha"
}
}
expected = {"db": {"name": "alpha"}}

cluster_uuid = "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30"

responses.add(
responses.POST,
f"{mock_client_url}/v2/databases/{cluster_uuid}/dbs",
json=expected,
status = 201,
status=201,
)

resp = mock_client.databases.add(cluster_uuid, {"name": "alpha"})
Expand All @@ -32,40 +29,46 @@ def test_databases_add(mock_client: Client, mock_client_url):


@responses.activate
def test_databases_add_connection_pool(mock_client: Client , mock_client_url):
def test_databases_add_connection_pool(mock_client: Client, mock_client_url):
"""Mocks the databases add connection pool operation."""

expected = {
"pool": {
"user": "doadmin",
"name": "backend-pool",
"size": 10,
"db": "defaultdb",
"mode": "transaction",
"connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@backend-do-user-19081923-0.db.ondigitalocean.com:25061/backend-pool?sslmode=require",
"database": "backend-pool",
"host": "backend-do-user-19081923-0.db.ondigitalocean.com",
"port": 25061,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True
}
}
}
"pool": {
"user": "doadmin",
"name": "backend-pool",
"size": 10,
"db": "defaultdb",
"mode": "transaction",
"connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@backend-do-user-19081923-0.db.ondigitalocean.com:25061/backend-pool?sslmode=require",
"database": "backend-pool",
"host": "backend-do-user-19081923-0.db.ondigitalocean.com",
"port": 25061,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True,
},
}
}

cluster_uuid = "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30"

responses.add(
responses.POST,
f"{mock_client_url}/v2/databases/{cluster_uuid}/pools",
json = expected,
status = 201
json=expected,
status=201,
)

resp = mock_client.databases.add_connection_pool(
cluster_uuid,
{"name": "backend-pool","mode": "transaction","size": 10,"db": "defaultdb","user": "doadmin"}
{
"name": "backend-pool",
"mode": "transaction",
"size": 10,
"db": "defaultdb",
"user": "doadmin",
},
)

assert expected == resp
Expand Down Expand Up @@ -94,108 +97,100 @@ def test_databases_update_connection_pool(mock_client: Client, mock_client_url):


@responses.activate
def test_databases_create_cluster(mock_client: Client , mock_client_url):
def test_databases_create_cluster(mock_client: Client, mock_client_url):
"""Mocks the databases create cluster operation."""

expected = {
"database": {
"id": "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30",
"name": "backend",
"engine": "pg",
"version": "14",
"semantic_version": "14.5",
"connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@backend-do-user-19081923-0.db.ondigitalocean.com:25060/defaultdb?sslmode=require",
"database": "",
"host": "backend-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True
},
"private_connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@private-backend-do-user-19081923-0.db.ondigitalocean.com:25060/defaultdb?sslmode=require",
"database": "",
"host": "private-backend-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True
},
"users": [
{
"name": "doadmin",
"role": "primary",
"password": "wv78n3zpz42xezdk"
}
],
"db_names": [
"defaultdb"
],
"num_nodes": 2,
"region": "nyc3",
"status": "creating",
"created_at": "2019-01-11T18:37:36Z",
"maintenance_window": {
"day": "saturday",
"hour": "08:45:12",
"pending": True,
"description": [
"Update TimescaleDB to version 1.2.1",
"Upgrade to PostgreSQL 11.2 and 10.7 bugfix releases"
]
},
"size": "db-s-2vcpu-4gb",
"tags": [
"production"
],
"private_network_uuid": "d455e75d-4858-4eec-8c95-da2f0a5f93a7",
"version_end_of_life": "2023-11-09T00:00:00Z",
"version_end_of_availability": "2023-05-09T00:00:00Z"
}
}
"database": {
"id": "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30",
"name": "backend",
"engine": "pg",
"version": "14",
"semantic_version": "14.5",
"connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@backend-do-user-19081923-0.db.ondigitalocean.com:25060/defaultdb?sslmode=require",
"database": "",
"host": "backend-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True,
},
"private_connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@private-backend-do-user-19081923-0.db.ondigitalocean.com:25060/defaultdb?sslmode=require",
"database": "",
"host": "private-backend-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True,
},
"users": [
{"name": "doadmin", "role": "primary", "password": "wv78n3zpz42xezdk"}
],
"db_names": ["defaultdb"],
"num_nodes": 2,
"region": "nyc3",
"status": "creating",
"created_at": "2019-01-11T18:37:36Z",
"maintenance_window": {
"day": "saturday",
"hour": "08:45:12",
"pending": True,
"description": [
"Update TimescaleDB to version 1.2.1",
"Upgrade to PostgreSQL 11.2 and 10.7 bugfix releases",
],
},
"size": "db-s-2vcpu-4gb",
"tags": ["production"],
"private_network_uuid": "d455e75d-4858-4eec-8c95-da2f0a5f93a7",
"version_end_of_life": "2023-11-09T00:00:00Z",
"version_end_of_availability": "2023-05-09T00:00:00Z",
}
}

cluster_uuid = "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30"

responses.add(
responses.POST,
f"{mock_client_url}/v2/databases",
json = expected,
status = 201,
json=expected,
status=201,
)

resp = mock_client.databases.create_cluster(
{"name": "backend","engine": "pg","version": "14","region": "nyc3","size": "db-s-2vcpu-4gb","num_nodes": 2,"tags": ["production"]}
{
"name": "backend",
"engine": "pg",
"version": "14",
"region": "nyc3",
"size": "db-s-2vcpu-4gb",
"num_nodes": 2,
"tags": ["production"],
}
)

assert expected == resp



@responses.activate
def test_databases_add_user(mock_client: Client , mock_client_url):
def test_databases_add_user(mock_client: Client, mock_client_url):
"""Mocks the databases add user operation."""

expected = {
"user": {
"name": "app-01",
"role": "normal",
"password": "jge5lfxtzhx42iff"
}
}
"user": {"name": "app-01", "role": "normal", "password": "jge5lfxtzhx42iff"}
}
cluster_uuid = "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30"

responses.add(
responses.POST,
f"{mock_client_url}/v2/databases/{cluster_uuid}/users",
json=expected,
status=201
status=201,
)

resp = mock_client.databases.add_user(
cluster_uuid,
{"name": "app-01"}
)
resp = mock_client.databases.add_user(cluster_uuid, {"name": "app-01"})

assert expected == resp

Expand All @@ -221,48 +216,48 @@ def test_databases_update_major_version(mock_client: Client, mock_client_url):


@responses.activate
def test_databases_create_replica(mock_client: Client , mock_client_url):
def test_databases_create_replica(mock_client: Client, mock_client_url):
"""Mocks the databases create replica operation."""

expected = {
"replica": {
"name": "read-nyc3-01",
"connection": {
"uri": "",
"database": "defaultdb",
"host": "read-nyc3-01-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True
},
"private_connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@private-read-nyc3-01-do-user-19081923-0.db.ondigitalocean.com:25060/defaultdb?sslmode=require",
"database": "",
"host": "private-read-nyc3-01-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True
},
"region": "nyc3",
"status": "online",
"created_at": "2019-01-11T18:37:36Z"
}
}
"replica": {
"name": "read-nyc3-01",
"connection": {
"uri": "",
"database": "defaultdb",
"host": "read-nyc3-01-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True,
},
"private_connection": {
"uri": "postgres://doadmin:wv78n3zpz42xezdk@private-read-nyc3-01-do-user-19081923-0.db.ondigitalocean.com:25060/defaultdb?sslmode=require",
"database": "",
"host": "private-read-nyc3-01-do-user-19081923-0.db.ondigitalocean.com",
"port": 25060,
"user": "doadmin",
"password": "wv78n3zpz42xezdk",
"ssl": True,
},
"region": "nyc3",
"status": "online",
"created_at": "2019-01-11T18:37:36Z",
}
}

cluster_uuid = "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30"

responses.add(
responses.POST,
f"{mock_client_url}/v2/databases/{cluster_uuid}/replicas",
json = expected,
status=201,
json=expected,
status=201,
)

resp = mock_client.databases.create_replica(
cluster_uuid,
{"name": "read-nyc3-01","region": "nyc3","size": "db-s-2vcpu-4gb"}
{"name": "read-nyc3-01", "region": "nyc3", "size": "db-s-2vcpu-4gb"},
)

assert expected == resp
Expand Down

0 comments on commit 61498f0

Please sign in to comment.