refactor(azure): Change class names from azure services and fix typing error (#3350)

This commit is contained in:
Pedro Martín
2024-02-05 15:43:04 +01:00
committed by GitHub
parent 8c51094df1
commit 4587a9f651
44 changed files with 581 additions and 584 deletions

View File

@@ -8,11 +8,11 @@ from azure.mgmt.sql.models import (
)
from prowler.providers.azure.services.sqlserver.sqlserver_service import (
DatabaseServer,
SQL_Server,
Database,
Server,
)
AZURE_SUSCRIPTION = str(uuid4())
AZURE_SUBSCRIPTION = str(uuid4())
class Test_sqlserver_vulnerability_assessment_enabled:
@@ -36,7 +36,7 @@ class Test_sqlserver_vulnerability_assessment_enabled:
sqlserver_client = mock.MagicMock
sql_server_name = "SQL Server Name"
sql_server_id = str(uuid4())
database = DatabaseServer(
database = Database(
id="id",
name="name",
type="type",
@@ -45,8 +45,8 @@ class Test_sqlserver_vulnerability_assessment_enabled:
tde_encryption=None,
)
sqlserver_client.sql_servers = {
AZURE_SUSCRIPTION: [
SQL_Server(
AZURE_SUBSCRIPTION: [
Server(
id=sql_server_id,
name=sql_server_name,
public_network_access="",
@@ -77,9 +77,9 @@ class Test_sqlserver_vulnerability_assessment_enabled:
assert result[0].status == "FAIL"
assert (
result[0].status_extended
== f"SQL Server {sql_server_name} from subscription {AZURE_SUSCRIPTION} has vulnerability assessment disabled."
== f"SQL Server {sql_server_name} from subscription {AZURE_SUBSCRIPTION} has vulnerability assessment disabled."
)
assert result[0].subscription == AZURE_SUSCRIPTION
assert result[0].subscription == AZURE_SUBSCRIPTION
assert result[0].resource_name == sql_server_name
assert result[0].resource_id == sql_server_id
@@ -87,7 +87,7 @@ class Test_sqlserver_vulnerability_assessment_enabled:
sqlserver_client = mock.MagicMock
sql_server_name = "SQL Server Name"
sql_server_id = str(uuid4())
database = DatabaseServer(
database = Database(
id="id",
name="name",
type="type",
@@ -96,8 +96,8 @@ class Test_sqlserver_vulnerability_assessment_enabled:
tde_encryption=TransparentDataEncryption(status="Disabled"),
)
sqlserver_client.sql_servers = {
AZURE_SUSCRIPTION: [
SQL_Server(
AZURE_SUBSCRIPTION: [
Server(
id=sql_server_id,
name=sql_server_name,
public_network_access="",
@@ -130,9 +130,9 @@ class Test_sqlserver_vulnerability_assessment_enabled:
assert result[0].status == "FAIL"
assert (
result[0].status_extended
== f"SQL Server {sql_server_name} from subscription {AZURE_SUSCRIPTION} has vulnerability assessment disabled."
== f"SQL Server {sql_server_name} from subscription {AZURE_SUBSCRIPTION} has vulnerability assessment disabled."
)
assert result[0].subscription == AZURE_SUSCRIPTION
assert result[0].subscription == AZURE_SUBSCRIPTION
assert result[0].resource_name == sql_server_name
assert result[0].resource_id == sql_server_id
@@ -140,7 +140,7 @@ class Test_sqlserver_vulnerability_assessment_enabled:
sqlserver_client = mock.MagicMock
sql_server_name = "SQL Server Name"
sql_server_id = str(uuid4())
database = DatabaseServer(
database = Database(
id="id",
name="name",
type="type",
@@ -149,8 +149,8 @@ class Test_sqlserver_vulnerability_assessment_enabled:
tde_encryption=TransparentDataEncryption(status="Enabled"),
)
sqlserver_client.sql_servers = {
AZURE_SUSCRIPTION: [
SQL_Server(
AZURE_SUBSCRIPTION: [
Server(
id=sql_server_id,
name=sql_server_name,
public_network_access="",
@@ -183,8 +183,8 @@ class Test_sqlserver_vulnerability_assessment_enabled:
assert result[0].status == "PASS"
assert (
result[0].status_extended
== f"SQL Server {sql_server_name} from subscription {AZURE_SUSCRIPTION} has vulnerability assessment enabled."
== f"SQL Server {sql_server_name} from subscription {AZURE_SUBSCRIPTION} has vulnerability assessment enabled."
)
assert result[0].subscription == AZURE_SUSCRIPTION
assert result[0].subscription == AZURE_SUBSCRIPTION
assert result[0].resource_name == sql_server_name
assert result[0].resource_id == sql_server_id