Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

added mocked tests for databases - part 1 #215

Merged
merged 5 commits into from
Oct 2, 2023
Merged

added mocked tests for databases - part 1 #215

Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
6175ab7
added mocked tests
work-mohit Sep 29, 2023
61498f0
fix linting issues
work-mohit Oct 2, 2023
1b506ea
Merge branch 'main' into add-mocked-tests-databases-1
work-mohit Oct 2, 2023
07890a1
fix - duplicate code
work-mohit Oct 2, 2023
44d89d7
fix- remove unused variable
work-mohit Oct 2, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
211 changes: 211 additions & 0 deletions tests/mocked/test_databases.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,72 @@
from pydo import Client


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

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,
)

resp = mock_client.databases.add(cluster_uuid, {"name": "alpha"})

assert expected == resp


@responses.activate
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,
},
}
}

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

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

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

assert expected == resp


@responses.activate
def test_databases_update_connection_pool(mock_client: Client, mock_client_url):
"""Mocks the databases update connection pool operation."""
Expand All @@ -28,6 +94,103 @@ def test_databases_update_connection_pool(mock_client: Client, mock_client_url):
assert resp is None


@responses.activate
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-cluster",
"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": 3,
"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",
}
}

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

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

assert expected == resp


@responses.activate
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"}
}
cluster_uuid = "9cc10173-e9ea-4176-9dbc-a4cee4c4ff30"

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

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

assert expected == resp


@responses.activate
def test_databases_update_major_version(mock_client: Client, mock_client_url):
"""Mocks the databases update major version."""
Expand All @@ -48,6 +211,54 @@ def test_databases_update_major_version(mock_client: Client, mock_client_url):
assert resp is None


@responses.activate
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",
}
}

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

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

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

assert expected == resp


@responses.activate
def test_databases_promote_replica(mock_client: Client, mock_client_url):
"""Mocks the databases promote replica to primary cluster function."""
Expand Down
Loading