@@ -40,9 +40,11 @@ func (s *SpamoorSuite) SetupTest() {
4040
4141// env holds a fully-wired environment created by setupEnv.
4242type env struct {
43- traces traceProvider
44- spamoorAPI * spamoor.API
45- ethClient * ethclient.Client
43+ traces traceProvider
44+ spamoorAPI * spamoor.API
45+ ethClient * ethclient.Client
46+ evNodeServiceName string
47+ evRethServiceName string
4648}
4749
4850// TODO: temporary hardcoded tag, will be replaced with a proper release tag
@@ -151,8 +153,10 @@ func (s *SpamoorSuite) setupLocalEnv(cfg benchConfig) *env {
151153 t : t ,
152154 startTime : time .Now (),
153155 },
154- spamoorAPI : spNode .API (),
155- ethClient : ethClient ,
156+ spamoorAPI : spNode .API (),
157+ ethClient : ethClient ,
158+ evNodeServiceName : cfg .ServiceName ,
159+ evRethServiceName : "ev-reth" ,
156160 }
157161}
158162
@@ -205,44 +209,46 @@ func (s *SpamoorSuite) setupExternalEnv(cfg benchConfig, rpcURL string) *env {
205209 t : t ,
206210 startTime : time .Now (),
207211 },
208- spamoorAPI : spNode .API (),
209- ethClient : ethClient ,
212+ spamoorAPI : spNode .API (),
213+ ethClient : ethClient ,
214+ evNodeServiceName : envOrDefault ("BENCH_EVNODE_SERVICE_NAME" , "ev-node" ),
215+ evRethServiceName : envOrDefault ("BENCH_EVRETH_SERVICE_NAME" , "ev-reth" ),
210216 }
211217}
212218
213219// collectTraces fetches ev-node traces (required) and ev-reth traces (optional)
214220// from the configured trace provider, then displays flowcharts.
215- func (s * SpamoorSuite ) collectTraces (e * env , serviceName string ) * traceResult {
221+ func (s * SpamoorSuite ) collectTraces (e * env ) * traceResult {
216222 t := s .T ()
217223 ctx := t .Context ()
218224
219- evNodeSpans , err := e .traces .collectSpans (ctx , serviceName )
220- s .Require ().NoError (err , "failed to collect %s traces" , serviceName )
225+ evNodeSpans , err := e .traces .collectSpans (ctx , e . evNodeServiceName )
226+ s .Require ().NoError (err , "failed to collect %s traces" , e . evNodeServiceName )
221227
222228 tr := & traceResult {
223229 evNode : evNodeSpans ,
224- evReth : e .traces .tryCollectSpans (ctx , "ev-reth" ),
230+ evReth : e .traces .tryCollectSpans (ctx , e . evRethServiceName ),
225231 }
226232
227- if link := e .traces .uiURL (serviceName ); link != "" {
233+ if link := e .traces .uiURL (e . evNodeServiceName ); link != "" {
228234 t .Logf ("traces UI: %s" , link )
229235 }
230236
231237 if rc , ok := e .traces .(richSpanCollector ); ok {
232- if spans , err := rc .collectRichSpans (ctx , serviceName ); err == nil {
238+ if spans , err := rc .collectRichSpans (ctx , e . evNodeServiceName ); err == nil {
233239 tr .evNodeRich = spans
234240 }
235- if spans , err := rc .collectRichSpans (ctx , "ev-reth" ); err == nil {
241+ if spans , err := rc .collectRichSpans (ctx , e . evRethServiceName ); err == nil {
236242 tr .evRethRich = spans
237243 }
238244 }
239245
240246 if rac , ok := e .traces .(resourceAttrCollector ); ok {
241- tr .evNodeAttrs = rac .fetchResourceAttrs (ctx , serviceName )
242- tr .evRethAttrs = rac .fetchResourceAttrs (ctx , "ev-reth" )
247+ tr .evNodeAttrs = rac .fetchResourceAttrs (ctx , e . evNodeServiceName )
248+ tr .evRethAttrs = rac .fetchResourceAttrs (ctx , e . evRethServiceName )
243249 }
244250
245- tr .displayFlowcharts (t , serviceName )
251+ tr .displayFlowcharts (t , e . evNodeServiceName )
246252 return tr
247253}
248254
0 commit comments