@@ -39,6 +39,7 @@ hardware_ecc_events_total{availability_zone="us-east-1c",event_type="sram_ecc_un
39
39
const dummyClusterName = "cluster-name"
40
40
const dummyHostName = "i-000000000"
41
41
const dummyNodeName = "dummy-nodeName"
42
+ const dummyInstanceType = "instance-type"
42
43
43
44
type mockHostInfoProvider struct {
44
45
}
@@ -51,13 +52,17 @@ func (m mockHostInfoProvider) GetInstanceID() string {
51
52
return dummyHostName
52
53
}
53
54
55
+ func (m mockHostInfoProvider ) GetInstanceType () string {
56
+ return dummyInstanceType
57
+ }
58
+
54
59
func TestNewNeuronScraperEndToEnd (t * testing.T ) {
55
60
t .Setenv ("HOST_NAME" , dummyNodeName )
56
61
expectedMetrics := make (map [string ]prometheusscraper.ExpectedMetricStruct )
57
62
expectedMetrics ["neuroncore_utilization_ratio" ] = prometheusscraper.ExpectedMetricStruct {
58
63
MetricValue : 0.1 ,
59
64
MetricLabels : []prometheusscraper.MetricLabel {
60
- {LabelName : "InstanceId" , LabelValue : "i-09db9b55e0095612f" },
65
+ {LabelName : "InstanceId" , LabelValue : dummyHostName },
61
66
{LabelName : "ClusterName" , LabelValue : dummyClusterName },
62
67
{LabelName : "NeuronCore" , LabelValue : "0" },
63
68
{LabelName : "NodeName" , LabelValue : dummyNodeName },
@@ -66,7 +71,7 @@ func TestNewNeuronScraperEndToEnd(t *testing.T) {
66
71
expectedMetrics ["neurondevice_hw_ecc_events_total_mem_ecc_corrected" ] = prometheusscraper.ExpectedMetricStruct {
67
72
MetricValue : 3 ,
68
73
MetricLabels : []prometheusscraper.MetricLabel {
69
- {LabelName : "InstanceId" , LabelValue : "i-09db9b55e0095612f" },
74
+ {LabelName : "InstanceId" , LabelValue : dummyHostName },
70
75
{LabelName : "ClusterName" , LabelValue : dummyClusterName },
71
76
{LabelName : "NeuronDevice" , LabelValue : "5" },
72
77
{LabelName : "NodeName" , LabelValue : dummyNodeName },
@@ -75,7 +80,7 @@ func TestNewNeuronScraperEndToEnd(t *testing.T) {
75
80
expectedMetrics ["neuron_runtime_memory_used_bytes" ] = prometheusscraper.ExpectedMetricStruct {
76
81
MetricValue : 9.043968e+06 ,
77
82
MetricLabels : []prometheusscraper.MetricLabel {
78
- {LabelName : "InstanceId" , LabelValue : "i-09db9b55e0095612f" },
83
+ {LabelName : "InstanceId" , LabelValue : dummyHostName },
79
84
{LabelName : "ClusterName" , LabelValue : dummyClusterName },
80
85
{LabelName : "NodeName" , LabelValue : dummyNodeName },
81
86
},
@@ -84,7 +89,7 @@ func TestNewNeuronScraperEndToEnd(t *testing.T) {
84
89
expectedMetrics ["execution_errors_created" ] = prometheusscraper.ExpectedMetricStruct {
85
90
MetricValue : 1.7083389404380567e+09 ,
86
91
MetricLabels : []prometheusscraper.MetricLabel {
87
- {LabelName : "InstanceId" , LabelValue : "i-09db9b55e0095612f" },
92
+ {LabelName : "InstanceId" , LabelValue : dummyHostName },
88
93
{LabelName : "ClusterName" , LabelValue : dummyClusterName },
89
94
{LabelName : "NodeName" , LabelValue : dummyNodeName },
90
95
},
@@ -93,7 +98,7 @@ func TestNewNeuronScraperEndToEnd(t *testing.T) {
93
98
expectedMetrics ["system_memory_total_bytes" ] = prometheusscraper.ExpectedMetricStruct {
94
99
MetricValue : 5.32523487232e+011 ,
95
100
MetricLabels : []prometheusscraper.MetricLabel {
96
- {LabelName : "InstanceId" , LabelValue : "i-09db9b55e0095612f" },
101
+ {LabelName : "InstanceId" , LabelValue : dummyHostName },
97
102
{LabelName : "ClusterName" , LabelValue : dummyClusterName },
98
103
{LabelName : "NodeName" , LabelValue : dummyNodeName },
99
104
},
@@ -102,7 +107,7 @@ func TestNewNeuronScraperEndToEnd(t *testing.T) {
102
107
expectedMetrics ["hardware_ecc_events_total" ] = prometheusscraper.ExpectedMetricStruct {
103
108
MetricValue : 864.0 ,
104
109
MetricLabels : []prometheusscraper.MetricLabel {
105
- {LabelName : "InstanceId" , LabelValue : "i-09db9b55e0095612f" },
110
+ {LabelName : "InstanceId" , LabelValue : dummyHostName },
106
111
{LabelName : "ClusterName" , LabelValue : dummyClusterName },
107
112
{LabelName : "NodeName" , LabelValue : dummyNodeName },
108
113
},
0 commit comments