diff --git a/docs/agent-training/stage1_foundation.json b/docs/agent-training/stage1_foundation.json
index 26241bd5..0c9805fd 100644
--- a/docs/agent-training/stage1_foundation.json
+++ b/docs/agent-training/stage1_foundation.json
@@ -327,6 +327,20 @@
         "weight": 1
       }
     ],
+    "integration_tests": [
+      {
+        "name": "end_to_end_wallet_flow",
+        "operations": ["wallet_create", "wallet_fund", "wallet_send"],
+        "cleanup": true
+      }
+    ],
+    "coverage_target": 80,
+    "coverage_report": true,
     "passing_score": 80
+  },
+  "mock_data": {
+    "wallets": ["test-wallet-1", "test-wallet-2"],
+    "addresses": ["0x1234567890abcdef", "0x0987654321fedcba"],
+    "transactions": ["tx_1", "tx_2"]
   }
 }
diff --git a/docs/agent-training/training_schema.json b/docs/agent-training/training_schema.json
index 5a975301..4b042f30 100644
--- a/docs/agent-training/training_schema.json
+++ b/docs/agent-training/training_schema.json
@@ -222,6 +222,39 @@
             }
           }
         },
+        "integration_tests": {
+          "type": "array",
+          "items": {
+            "type": "object",
+            "properties": {
+              "name": {
+                "type": "string",
+                "description": "Integration test name"
+              },
+              "operations": {
+                "type": "array",
+                "items": {"type": "string"},
+                "description": "Operations to execute in sequence"
+              },
+              "cleanup": {
+                "type": "boolean",
+                "description": "Whether to cleanup after test",
+                "default": true
+              }
+            }
+          }
+        },
+        "coverage_target": {
+          "type": "number",
+          "description": "Target test coverage percentage",
+          "minimum": 0,
+          "maximum": 100
+        },
+        "coverage_report": {
+          "type": "boolean",
+          "description": "Whether to generate coverage report",
+          "default": false
+        },
         "passing_score": {
           "type": "number",
           "description": "Minimum score percentage to pass",
@@ -230,6 +263,27 @@
           "default": 80
         }
       }
+    },
+    "mock_data": {
+      "type": "object",
+      "description": "Mock data for testing",
+      "properties": {
+        "wallets": {
+          "type": "array",
+          "items": {"type": "string"},
+          "description": "Mock wallet names"
+        },
+        "addresses": {
+          "type": "array",
+          "items": {"type": "string"},
+          "description": "Mock wallet addresses"
+        },
+        "transactions": {
+          "type": "array",
+          "items": {"type": "string"},
+          "description": "Mock transaction IDs"
+        }
+      }
     }
   }
 }