Skip to content

Commit b250033

Browse files
authored
all: remove MockMetricsRegistry in favor of the real one (#4427)
* all: replace mock metrics with real ones
1 parent 21e7cb3 commit b250033

File tree

23 files changed

+49
-160
lines changed

23 files changed

+49
-160
lines changed

Cargo.lock

Lines changed: 4 additions & 14 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Cargo.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ members = [
33
"core",
44
"chain/*",
55
"graphql",
6-
"mock",
76
"node",
87
"runtime/*",
98
"server/*",

chain/ethereum/Cargo.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,9 @@ graph-runtime-wasm = { path = "../../runtime/wasm" }
2525
graph-runtime-derive = { path = "../../runtime/derive" }
2626

2727
[dev-dependencies]
28+
graph-core = { path = "../../core" }
2829
test-store = { path = "../../store/test-store" }
2930
base64 = "0.20.0"
30-
graph-mock = { path = "../../mock" }
3131

3232
[build-dependencies]
3333
tonic-build = { workspace = true }

chain/ethereum/src/network.rs

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -219,11 +219,12 @@ impl EthereumNetworks {
219219

220220
#[cfg(test)]
221221
mod tests {
222-
use std::sync::Arc;
223-
224-
use graph::{firehose::SubgraphLimit, prelude::MetricsRegistry, tokio, url::Url};
225-
use graph_mock::MockMetricsRegistry;
222+
use graph::{
223+
firehose::SubgraphLimit, prelude::MetricsRegistry as MetricsRegistryTrait, tokio, url::Url,
224+
};
225+
use graph_core::MetricsRegistry;
226226
use http::HeaderMap;
227+
use std::sync::Arc;
227228

228229
use crate::{EthereumAdapter, EthereumNetworks, ProviderEthRpcMetrics, Transport};
229230

@@ -288,7 +289,7 @@ mod tests {
288289
async fn adapter_selector_selects_eth_call() {
289290
let chain = "mainnet".to_string();
290291
let logger = graph::log::logger(true);
291-
let mock_registry: Arc<dyn MetricsRegistry> = Arc::new(MockMetricsRegistry::new());
292+
let mock_registry: Arc<dyn MetricsRegistryTrait> = Arc::new(MetricsRegistry::mock());
292293
let transport =
293294
Transport::new_rpc(Url::parse("http://127.0.0.1").unwrap(), HeaderMap::new());
294295
let provider_metrics = Arc::new(ProviderEthRpcMetrics::new(mock_registry.clone()));
@@ -391,7 +392,7 @@ mod tests {
391392
async fn adapter_selector_unlimited() {
392393
let chain = "mainnet".to_string();
393394
let logger = graph::log::logger(true);
394-
let mock_registry: Arc<dyn MetricsRegistry> = Arc::new(MockMetricsRegistry::new());
395+
let mock_registry: Arc<dyn MetricsRegistryTrait> = Arc::new(MetricsRegistry::mock());
395396
let transport =
396397
Transport::new_rpc(Url::parse("http://127.0.0.1").unwrap(), HeaderMap::new());
397398
let provider_metrics = Arc::new(ProviderEthRpcMetrics::new(mock_registry.clone()));
@@ -459,7 +460,7 @@ mod tests {
459460
async fn adapter_selector_disable_call_only_fallback() {
460461
let chain = "mainnet".to_string();
461462
let logger = graph::log::logger(true);
462-
let mock_registry: Arc<dyn MetricsRegistry> = Arc::new(MockMetricsRegistry::new());
463+
let mock_registry: Arc<dyn MetricsRegistryTrait> = Arc::new(MetricsRegistry::mock());
463464
let transport =
464465
Transport::new_rpc(Url::parse("http://127.0.0.1").unwrap(), HeaderMap::new());
465466
let provider_metrics = Arc::new(ProviderEthRpcMetrics::new(mock_registry.clone()));
@@ -525,7 +526,7 @@ mod tests {
525526
async fn adapter_selector_no_call_only_fallback() {
526527
let chain = "mainnet".to_string();
527528
let logger = graph::log::logger(true);
528-
let mock_registry: Arc<dyn MetricsRegistry> = Arc::new(MockMetricsRegistry::new());
529+
let mock_registry: Arc<dyn MetricsRegistryTrait> = Arc::new(MetricsRegistry::mock());
529530
let transport =
530531
Transport::new_rpc(Url::parse("http://127.0.0.1").unwrap(), HeaderMap::new());
531532
let provider_metrics = Arc::new(ProviderEthRpcMetrics::new(mock_registry.clone()));

core/Cargo.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@ anyhow = "1.0"
3232

3333
[dev-dependencies]
3434
tower-test = { git = "https://github.com/tower-rs/tower.git" }
35-
graph-mock = { path = "../mock" }
3635
test-store = { path = "../store/test-store" }
3736
hex = "0.4.3"
3837
graphql-parser = "0.4.0"

core/src/metrics/registry.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,10 @@ impl MetricsRegistry {
4242
}
4343
}
4444

45+
pub fn mock() -> Self {
46+
MetricsRegistry::new(Logger::root(slog::Discard, o!()), Arc::new(Registry::new()))
47+
}
48+
4549
fn gen_register_errors_counter(registry: Arc<Registry>) -> Box<Counter> {
4650
let opts = Opts::new(
4751
String::from("metrics_register_errors"),

mock/Cargo.toml

Lines changed: 0 additions & 7 deletions
This file was deleted.

mock/src/lib.rs

Lines changed: 0 additions & 3 deletions
This file was deleted.

mock/src/metrics_registry.rs

Lines changed: 0 additions & 87 deletions
This file was deleted.

runtime/test/Cargo.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,4 +16,3 @@ rand = "0.8.5"
1616

1717
[dev-dependencies]
1818
test-store = { path = "../../store/test-store" }
19-
graph-mock = { path = "../../mock" }

0 commit comments

Comments
 (0)