{
  "output": "Successfully created and wrote to new file: /Users/bob/git/dxw/aws-stuff/tests/test_storage.py. Here is the updated code:\nimport unittest\nfrom unittest.mock import MagicMock\nfrom audits.storage import fetch_storage\n\nclass TestStorageAudit(unittest.TestCase):\n    def test_fetch_storage(self):\n        mock_ce = MagicMock()\n        \n        # Mock response\n        mock_ce.get_cost_and_usage.return_value = {\n            \"ResultsByTime\": [\n                {\n                    \"Groups\": [\n                        {\n                            \"Keys\": [\"123456789012\", \"S3: Storage - Standard\"],\n                            \"Metrics\": {\n                                \"UsageQuantity\": {\"Amount\": str(1024**3 * 100 * 720), \"Unit\": \"ByteHrs\"},\n                            }\n                        }\n                    ]\n                }\n            ]\n        }\n        \n        all_accounts = {\"123456789012\": \"Test Account\"}\n        hours_in_month = 720\n        \n        results = fetch_storage(mock_ce, \"2023-10-01\", \"2023-11-01\", all_accounts, hours_in_month)\n        \n        self.assertIn(\"123456789012\", results)\n        # Usage should be 100 GB\n        self.assertEqual(results[\"123456789012\"], 100.0)\n\nif __name__ == \"__main__\":\n    unittest.main()\n"
}