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

@@ -4,11 +4,11 @@ from uuid import uuid4
from azure.mgmt.sql.models import TransparentDataEncryption
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_tde_encryption_enabled:
@@ -33,8 +33,8 @@ class Test_sqlserver_tde_encryption_enabled:
sql_server_name = "SQL Server Name"
sql_server_id = str(uuid4())
sqlserver_client.sql_servers = {
AZURE_SUSCRIPTION: [
SQL_Server(
AZURE_SUBSCRIPTION: [
Server(
id=sql_server_id,
name=sql_server_name,
public_network_access="",
@@ -65,7 +65,7 @@ class Test_sqlserver_tde_encryption_enabled:
sql_server_id = str(uuid4())
database_name = "Database Name"
database_id = str(uuid4())
database = DatabaseServer(
database = Database(
id=database_id,
name=database_name,
type="type",
@@ -74,8 +74,8 @@ class Test_sqlserver_tde_encryption_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="",
@@ -103,9 +103,9 @@ class Test_sqlserver_tde_encryption_enabled:
assert result[0].status == "FAIL"
assert (
result[0].status_extended
== f"Database {database_name} from SQL Server {sql_server_name} from subscription {AZURE_SUSCRIPTION} has TDE disabled"
== f"Database {database_name} from SQL Server {sql_server_name} from subscription {AZURE_SUBSCRIPTION} has TDE disabled"
)
assert result[0].subscription == AZURE_SUSCRIPTION
assert result[0].subscription == AZURE_SUBSCRIPTION
assert result[0].resource_name == database_name
assert result[0].resource_id == database_id
@@ -115,7 +115,7 @@ class Test_sqlserver_tde_encryption_enabled:
sql_server_id = str(uuid4())
database_name = "Database Name"
database_id = str(uuid4())
database = DatabaseServer(
database = Database(
id=database_id,
name=database_name,
type="type",
@@ -124,8 +124,8 @@ class Test_sqlserver_tde_encryption_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="",
@@ -153,8 +153,8 @@ class Test_sqlserver_tde_encryption_enabled:
assert result[0].status == "PASS"
assert (
result[0].status_extended
== f"Database {database_name} from SQL Server {sql_server_name} from subscription {AZURE_SUSCRIPTION} has TDE enabled"
== f"Database {database_name} from SQL Server {sql_server_name} from subscription {AZURE_SUBSCRIPTION} has TDE enabled"
)
assert result[0].subscription == AZURE_SUSCRIPTION
assert result[0].subscription == AZURE_SUBSCRIPTION
assert result[0].resource_name == database_name
assert result[0].resource_id == database_id