runbooks 1.1.4__py3-none-any.whl → 1.1.6__py3-none-any.whl

runbooks/vpc/tests/conftest.py

@@ -31,8 +31,6 @@ from runbooks.vpc.networking_wrapper import VPCNetworkingWrapper
 
 @pytest.fixture(scope="session")
 def aws_credentials():
-    # Dynamic test period for consistent test data
-    test_period = get_test_date_period(30)
     """Mock AWS credentials for VPC testing."""
     os.environ["AWS_ACCESS_KEY_ID"] = "testing"
     os.environ["AWS_SECRET_ACCESS_KEY"] = "testing"
@@ -274,11 +272,15 @@ def performance_benchmarks():
 @pytest.fixture
 def mock_cost_explorer_responses():
     """Mock Cost Explorer API responses for testing."""
+    # Dynamic test period calculation
+    end_date = datetime.now()
+    start_date = end_date - timedelta(days=30)
+
     return {
         "vpc_costs": {
             "ResultsByTime": [
                 {
-                    "TimePeriod": {"Start": test_period["Start"], "End": test_period["End"]},
+                    "TimePeriod": {"Start": start_date.strftime("%Y-%m-%d"), "End": end_date.strftime("%Y-%m-%d")},
                     "Total": {"BlendedCost": {"Amount": "145.67", "Unit": "USD"}},
                 }
             ]
@@ -286,7 +288,7 @@ def mock_cost_explorer_responses():
         "nat_gateway_costs": {
             "ResultsByTime": [
                 {
-                    "TimePeriod": {"Start": test_period["Start"], "End": test_period["End"]},
+                    "TimePeriod": {"Start": start_date.strftime("%Y-%m-%d"), "End": end_date.strftime("%Y-%m-%d")},
                     "Total": {"BlendedCost": {"Amount": "89.32", "Unit": "USD"}},
                 }
             ]
@@ -356,3 +358,311 @@ def security_test_validator():
         return True
 
     return _validate_security
+
+
+# ========================================
+# VPC Cleanup Framework Fixtures
+# ========================================
+# Added to support config-driven VPC cleanup testing
+
+
+@pytest.fixture
+def cleanup_valid_config():
+    """Load AWS-25 reference config for VPC cleanup testing"""
+    import yaml
+
+    config_path = Path(__file__).parent.parent.parent.parent.parent / "examples/vpc-cleanup/aws25-campaign-config.yaml"
+    if not config_path.exists():
+        # Fallback to minimal valid config if reference not available
+        return {
+            "campaign_metadata": {
+                "campaign_id": "TEST-01",
+                "campaign_name": "Test Campaign",
+                "execution_date": "2025-10-02",
+                "aws_billing_profile": "test-profile",
+                "description": "Test campaign description",
+            },
+            "deleted_vpcs": [
+                {
+                    "vpc_id": "vpc-test123456789abcd",
+                    "account_id": "123456789012",
+                    "region": "us-east-1",
+                    "deletion_date": "2025-09-10",
+                    "deletion_principal": "test.user@example.com",
+                    "pre_deletion_baseline_months": 3,
+                }
+            ],
+            "cost_explorer_config": {
+                "metrics": ["UnblendedCost"],
+                "group_by_dimensions": ["SERVICE", "REGION"],
+                "pre_deletion_baseline": {"granularity_monthly": "MONTHLY", "months_before_deletion": 3},
+                "pre_deletion_detailed": {"granularity_daily": "DAILY", "days_before_deletion": 10},
+                "post_deletion_validation": {"granularity_daily": "DAILY", "days_after_deletion": 30},
+            },
+            "attribution_rules": {
+                "vpc_specific_services": {
+                    "confidence_level": "HIGH (95%)",
+                    "attribution_percentage": 100,
+                    "service_patterns": ["Amazon Virtual Private Cloud"],
+                },
+                "vpc_related_services": {
+                    "confidence_level": "MEDIUM (85%)",
+                    "attribution_percentage": 70,
+                    "service_patterns": ["Amazon Elastic Compute Cloud - Compute"],
+                },
+                "other_services": {
+                    "confidence_level": "LOW (<85%)",
+                    "attribution_percentage": 30,
+                    "service_patterns": ["*"],
+                },
+            },
+            "output_config": {
+                "csv_output_file": "test_output.csv",
+                "csv_columns": ["VPC_ID", "Account_ID", "Deletion_Date", "Monthly_Savings_Realized"],
+                "json_results_file": "test_results.json",
+                "execution_summary_file": "test_summary.md",
+            },
+        }
+
+    with open(config_path) as f:
+        return yaml.safe_load(f)
+
+
+@pytest.fixture
+def cleanup_sample_vpc_deletion():
+    """Sample VPC deletion for testing"""
+    return {
+        "vpc_id": "vpc-test123456789abcd",
+        "account_id": "123456789012",
+        "region": "us-east-1",
+        "deletion_date": "2025-09-10",
+        "deletion_principal": "test.user@example.com",
+        "pre_deletion_baseline_months": 3,
+    }
+
+
+@pytest.fixture
+def cleanup_sample_vpc_deletions():
+    """Sample list of VPC deletions for multi-VPC testing"""
+    return [
+        {
+            "vpc_id": "vpc-test111111111aaaa",
+            "account_id": "111111111111",
+            "region": "us-east-1",
+            "deletion_date": "2025-09-10",
+            "deletion_principal": "user1@example.com",
+            "pre_deletion_baseline_months": 3,
+        },
+        {
+            "vpc_id": "vpc-test222222222bbbb",
+            "account_id": "222222222222",
+            "region": "us-west-2",
+            "deletion_date": "2025-08-15",
+            "deletion_principal": "user2@example.com",
+            "pre_deletion_baseline_months": 3,
+        },
+        {
+            "vpc_id": "vpc-test333333333cccc",
+            "account_id": "111111111111",
+            "region": "eu-west-1",
+            "deletion_date": "2025-09-01",
+            "deletion_principal": "user3@example.com",
+            "pre_deletion_baseline_months": 3,
+        },
+    ]
+
+
+@pytest.fixture
+def cleanup_mock_cost_explorer():
+    """Mock boto3 Cost Explorer client for cleanup testing"""
+    mock_ce_client = MagicMock()
+
+    def get_cost_side_effect(*args, **kwargs):
+        # Return different responses based on time period
+        time_period = kwargs.get("TimePeriod", {})
+        start_date = time_period.get("Start", "")
+
+        if start_date.startswith("2025-09"):
+            # Post-deletion period
+            return {
+                "ResultsByTime": [
+                    {
+                        "TimePeriod": time_period,
+                        "Groups": [
+                            {
+                                "Keys": ["Amazon Elastic Compute Cloud - Compute", "ap-southeast-2"],
+                                "Metrics": {"UnblendedCost": {"Amount": "15.00", "Unit": "USD"}},
+                            }
+                        ],
+                    }
+                ]
+            }
+        else:
+            # Pre-deletion period
+            return {
+                "ResultsByTime": [
+                    {
+                        "TimePeriod": time_period,
+                        "Groups": [
+                            {
+                                "Keys": ["Amazon Virtual Private Cloud", "ap-southeast-2"],
+                                "Metrics": {"UnblendedCost": {"Amount": "100.00", "Unit": "USD"}},
+                            },
+                            {
+                                "Keys": ["Amazon Elastic Compute Cloud - Compute", "ap-southeast-2"],
+                                "Metrics": {"UnblendedCost": {"Amount": "500.00", "Unit": "USD"}},
+                            },
+                        ],
+                    }
+                ]
+            }
+
+    mock_ce_client.get_cost_and_usage.side_effect = get_cost_side_effect
+    return mock_ce_client
+
+
+@pytest.fixture
+def cleanup_temp_config_file():
+    """Create temporary cleanup config file for testing"""
+    import yaml
+
+    minimal_config = {
+        "campaign_metadata": {
+            "campaign_id": "TEST-01",
+            "campaign_name": "Test Campaign",
+            "execution_date": "2025-10-02",
+            "aws_billing_profile": "test-profile",
+            "description": "Test campaign description",
+        },
+        "deleted_vpcs": [
+            {
+                "vpc_id": "vpc-test123456789abcd",
+                "account_id": "123456789012",
+                "region": "us-east-1",
+                "deletion_date": "2025-09-10",
+                "deletion_principal": "test.user@example.com",
+                "pre_deletion_baseline_months": 3,
+            }
+        ],
+        "cost_explorer_config": {
+            "metrics": ["UnblendedCost"],
+            "group_by_dimensions": ["SERVICE", "REGION"],
+            "pre_deletion_baseline": {"granularity_monthly": "MONTHLY", "months_before_deletion": 3},
+            "pre_deletion_detailed": {"granularity_daily": "DAILY", "days_before_deletion": 10},
+            "post_deletion_validation": {"granularity_daily": "DAILY", "days_after_deletion": 30},
+        },
+        "attribution_rules": {
+            "vpc_specific_services": {
+                "confidence_level": "HIGH (95%)",
+                "attribution_percentage": 100,
+                "service_patterns": ["Amazon Virtual Private Cloud"],
+            },
+            "vpc_related_services": {
+                "confidence_level": "MEDIUM (85%)",
+                "attribution_percentage": 70,
+                "service_patterns": ["Amazon Elastic Compute Cloud - Compute"],
+            },
+            "other_services": {
+                "confidence_level": "LOW (<85%)",
+                "attribution_percentage": 30,
+                "service_patterns": ["*"],
+            },
+        },
+        "output_config": {
+            "csv_output_file": "test_output.csv",
+            "csv_columns": ["VPC_ID", "Account_ID", "Deletion_Date", "Monthly_Savings_Realized"],
+            "json_results_file": "test_results.json",
+            "execution_summary_file": "test_summary.md",
+        },
+    }
+
+    with tempfile.NamedTemporaryFile(mode="w", suffix=".yaml", delete=False) as f:
+        yaml.dump(minimal_config, f)
+        temp_path = f.name
+
+    yield temp_path
+
+    # Cleanup
+    Path(temp_path).unlink(missing_ok=True)
+
+
+@pytest.fixture
+def cleanup_multi_lz_config():
+    """Config for multi-Landing Zone testing with different accounts and regions"""
+    return {
+        "campaign_metadata": {
+            "campaign_id": "MULTI-LZ-01",
+            "campaign_name": "Multi-LZ Test Campaign",
+            "execution_date": "2025-10-02",
+            "aws_billing_profile": "multi-lz-profile",
+            "description": "Testing multiple Landing Zones",
+        },
+        "deleted_vpcs": [
+            {
+                "vpc_id": "vpc-lz1111111111aaaa",
+                "account_id": "111111111111",
+                "region": "us-east-1",
+                "deletion_date": "2025-09-10",
+                "deletion_principal": "lz1.user@example.com",
+                "pre_deletion_baseline_months": 3,
+            },
+            {
+                "vpc_id": "vpc-lz2222222222bbbb",
+                "account_id": "222222222222",
+                "region": "eu-west-1",
+                "deletion_date": "2025-08-20",
+                "deletion_principal": "lz2.user@example.com",
+                "pre_deletion_baseline_months": 3,
+            },
+            {
+                "vpc_id": "vpc-lz3333333333cccc",
+                "account_id": "333333333333",
+                "region": "ap-southeast-2",
+                "deletion_date": "2025-09-05",
+                "deletion_principal": "lz3.user@example.com",
+                "pre_deletion_baseline_months": 3,
+            },
+        ],
+        "cost_explorer_config": {
+            "metrics": ["UnblendedCost"],
+            "group_by_dimensions": ["SERVICE", "REGION"],
+            "pre_deletion_baseline": {"granularity_monthly": "MONTHLY", "months_before_deletion": 3},
+            "pre_deletion_detailed": {"granularity_daily": "DAILY", "days_before_deletion": 10},
+            "post_deletion_validation": {"granularity_daily": "DAILY", "days_after_deletion": 30},
+        },
+        "attribution_rules": {
+            "vpc_specific_services": {
+                "confidence_level": "HIGH (95%)",
+                "attribution_percentage": 100,
+                "service_patterns": ["Amazon Virtual Private Cloud", "AWS PrivateLink"],
+            },
+            "vpc_related_services": {
+                "confidence_level": "MEDIUM (85%)",
+                "attribution_percentage": 70,
+                "service_patterns": ["Amazon Elastic Compute Cloud - Compute", "Elastic Load Balancing"],
+            },
+            "other_services": {
+                "confidence_level": "LOW (<85%)",
+                "attribution_percentage": 30,
+                "service_patterns": ["*"],
+            },
+        },
+        "output_config": {
+            "csv_output_file": "multi_lz_output.csv",
+            "csv_columns": [
+                "VPC_ID",
+                "Account_ID",
+                "Deletion_Date",
+                "Pre_Deletion_Monthly_Avg",
+                "Post_Deletion_Monthly_Avg",
+                "Monthly_Savings_Realized",
+                "Annual_Savings_Realized",
+                "Data_Quality",
+                "Confidence_Level",
+                "Notes",
+                "Service_Analysis",
+            ],
+            "json_results_file": "multi_lz_results.json",
+            "execution_summary_file": "multi_lz_summary.md",
+        },
+    }