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 006916f commit 96961d2
Showing 1 changed file with 84 additions and 85 deletions.
169 changes: 84 additions & 85 deletions tests/mocked/test_databases.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def test_databases_promote_replica(mock_client: Client, mock_client_url):
assert resp is None


@responses.activate
@responses.activate
def test_databases_patch_config(mock_client: Client, mock_client_url):
"""Mocks the databases patch config operation."""

Expand All @@ -75,83 +75,85 @@ def test_databases_patch_config(mock_client: Client, mock_client_url):
responses.add(
responses.PATCH,
f"{mock_client_url}/v2/databases/{cluster_uuid}/config",
status = 200,
status=200,
)

resp = mock_client.databases.patch_config(cluster_uuid,{"config": {}})
resp = mock_client.databases.patch_config(cluster_uuid, {"config": {}})

assert resp is None


@responses.activate
@responses.activate
def test_databases_reset_auth(mock_client: Client, mock_client_url):
"""Mocks the databases patch config 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"
user_name = "app-01"

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

resp = mock_client.databases.reset_auth(cluster_uuid, user_name,{"mysql_settings": {"auth_plugin": "caching_sha2_password"}})

assert expected == resp
resp = mock_client.databases.reset_auth(
cluster_uuid,
user_name,
{"mysql_settings": {"auth_plugin": "caching_sha2_password"}},
)

assert expected == resp

user_name = "app-02"
expected = {
"user": {
"name": "app-02",
"role": "normal",
"password": "wv78n3zpz42xezdk",
"mysql_settings": {
"auth_plugin": "mysql_native_password"
}
}
}
"user": {
"name": "app-02",
"role": "normal",
"password": "wv78n3zpz42xezdk",
"mysql_settings": {"auth_plugin": "mysql_native_password"},
}
}

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

resp = mock_client.databases.reset_auth(
cluster_uuid,
user_name,
{"mysql_settings": {"auth_plugin": "caching_sha2_password"}},
)

resp = mock_client.databases.reset_auth(cluster_uuid,user_name,{"mysql_settings": {"auth_plugin": "caching_sha2_password"}})

assert expected == resp


@responses.activate
@responses.activate
def test_databases_update_cluster_size(mock_client: Client, mock_client_url):
"""Mocks the databases patch config operation."""


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

responses.add(
responses.PUT,
f"{mock_client_url}/v2/databases/{cluster_uuid}/resize",
status = 202,
status=202,
)

resp = mock_client.databases.update_cluster_size(cluster_uuid,{"size": "db-s-4vcpu-8gb","num_nodes": 3})

assert resp is None
resp = mock_client.databases.update_cluster_size(
cluster_uuid, {"size": "db-s-4vcpu-8gb", "num_nodes": 3}
)

assert resp is None


@responses.activate
@responses.activate
def test_databases_update_firewall_rules(mock_client: Client, mock_client_url):
"""Mocks the databases update firewall rules operation."""

Expand All @@ -160,35 +162,25 @@ def test_databases_update_firewall_rules(mock_client: Client, mock_client_url):
responses.add(
responses.PUT,
f"{mock_client_url}/v2/databases/{cluster_uuid}/firewall",
status = 204,
status=204,
)

resp = mock_client.databases.update_firewall_rules(
cluster_uuid, {
"rules": [
{
"type": "ip_addr",
"value": "192.168.1.1"
},
{
"type": "k8s",
"value": "ff2a6c52-5a44-4b63-b99c-0e98e7a63d61"
},
{
"type": "droplet",
"value": "163973392"
},
{
"type": "tag",
"value": "backend"
}
]
})

assert resp is None
cluster_uuid,
{
"rules": [
{"type": "ip_addr", "value": "192.168.1.1"},
{"type": "k8s", "value": "ff2a6c52-5a44-4b63-b99c-0e98e7a63d61"},
{"type": "droplet", "value": "163973392"},
{"type": "tag", "value": "backend"},
]
},
)

assert resp is None


@responses.activate
@responses.activate
def test_databases_update_maintenance_window(mock_client: Client, mock_client_url):
"""Mocks the databases update firewall rules operation."""

Expand All @@ -197,50 +189,53 @@ def test_databases_update_maintenance_window(mock_client: Client, mock_client_ur
responses.add(
responses.PUT,
f"{mock_client_url}/v2/databases/{cluster_uuid}/maintenance",
status = 204,
status=204,
)

resp = mock_client.databases.update_maintenance_window(cluster_uuid, {"day": "tuesday","hour": "14:00"})

assert resp is None
resp = mock_client.databases.update_maintenance_window(
cluster_uuid, {"day": "tuesday", "hour": "14:00"}
)

@responses.activate
assert resp is None


@responses.activate
def test_databases_update_online_migration(mock_client: Client, mock_client_url):
"""Mocks the databases update firewall rules operation."""

expected = {
"id": "77b28fc8-19ff-11eb-8c9c-c68e24557488",
"status": "running",
"created_at": "2020-10-29T15:57:38Z"
}
"id": "77b28fc8-19ff-11eb-8c9c-c68e24557488",
"status": "running",
"created_at": "2020-10-29T15:57:38Z",
}

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

responses.add(
responses.PUT,
f"{mock_client_url}/v2/databases/{cluster_uuid}/online-migration",
json = expected,
status = 200,
json=expected,
status=200,
)

resp = mock_client.databases.update_online_migration(
cluster_uuid,
{
"source": {
"source": {
"host": "source-do-user-6607903-0.b.db.ondigitalocean.com",
"dbname": "defaultdb",
"port": 25060,
"username": "doadmin",
"password": "paakjnfe10rsrsmf"
"password": "paakjnfe10rsrsmf",
},
"disable_ssl": False
}
)
"disable_ssl": False,
},
)

assert expected == resp


@responses.activate
@responses.activate
def test_databases_update_region(mock_client: Client, mock_client_url):
"""Mocks the databases update firewall rules operation."""

Expand All @@ -249,15 +244,15 @@ def test_databases_update_region(mock_client: Client, mock_client_url):
responses.add(
responses.PUT,
f"{mock_client_url}/v2/databases/{cluster_uuid}/migrate",
status = 202,
status=202,
)

resp = mock_client.databases.update_region(cluster_uuid, {"region": "lon1"})

assert resp is None


@responses.activate
@responses.activate
def test_databases_update_sql_mode(mock_client: Client, mock_client_url):
"""Mocks the databases update firewall rules operation."""

Expand All @@ -266,10 +261,14 @@ def test_databases_update_sql_mode(mock_client: Client, mock_client_url):
responses.add(
responses.PUT,
f"{mock_client_url}/v2/databases/{cluster_uuid}/sql_mode",
status = 204,
status=204,
)

resp = mock_client.databases.update_sql_mode(cluster_uuid, {"sql_mode": "ANSI,ERROR_FOR_DIVISION_BY_ZERO,NO_ENGINE_SUBSTITUTION,NO_ZERO_DATE,NO_ZERO_IN_DATE"})

assert resp is None
resp = mock_client.databases.update_sql_mode(
cluster_uuid,
{
"sql_mode": "ANSI,ERROR_FOR_DIVISION_BY_ZERO,NO_ENGINE_SUBSTITUTION,NO_ZERO_DATE,NO_ZERO_IN_DATE"
},
)

assert resp is None

0 comments on commit 96961d2

Please sign in to comment.