1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
//! Main entry point for integration tests.
//! These tests need a running instance of the AI Gateway.
//!
//! # Environment Setup
//!
//! The tests expect the following environment variables to be set:
//!
//! ```
//! # Gateway URL (default: http://localhost:3000)
//! GATEWAY_URL=http://localhost:3000
//!
//! # ElasticSearch Configuration
//! ELASTICSEARCH_URL=http://localhost:9200
//! ELASTICSEARCH_USERNAME=elastic
//! ELASTICSEARCH_PASSWORD=your_password
//! ELASTICSEARCH_INDEX=ai-gateway-metrics
//!
//! # Provider API Keys
//! OPENAI_API_KEY=your_openai_api_key
//! ANTHROPIC_API_KEY=your_anthropic_api_key
//! GROQ_API_KEY=your_groq_api_key
//! FIREWORKS_API_KEY=your_fireworks_api_key
//! TOGETHER_API_KEY=your_together_api_key
//!
//! # AWS Bedrock Credentials (if testing Bedrock)
//! AWS_ACCESS_KEY_ID=your_aws_access_key_id
//! AWS_SECRET_ACCESS_KEY=your_aws_secret_access_key
//! AWS_REGION=us-east-1
//! ```
//!
//! # Running the tests
//!
//! ```bash
//! # Start the gateway with ElasticSearch in one terminal
//! ENABLE_ELASTICSEARCH=true cargo run
//!
//! # In another terminal, run the tests
//! cargo test --test run_integration_tests -- --nocapture
//!
//! # Run tests for specific providers
//! cargo test --test run_integration_tests openai -- --nocapture
//! cargo test --test run_integration_tests anthropic -- --nocapture
//! cargo test --test run_integration_tests groq -- --nocapture
//! cargo test --test run_integration_tests fireworks -- --nocapture
//! cargo test --test run_integration_tests together -- --nocapture
//! ```