fix: network resource test cases

RSS3-Network · May 20, 2024 · c560bab · c560bab
1 parent 17e421b
commit c560bab
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/internal/node/component/network/handler_network_test.go b/internal/node/component/network/handler_network_test.go
@@ -25,18 +25,26 @@ func TestCalculateMinimumResources(t *testing.T) {
 			expected: baseResource.Mul(8),
 		},
 		{
-			name:     "Polygon Core Resource",
-			network:  network.Polygon,
-			worker:   worker.Core,
-			expected: baseResource.Mul(8),
+			name:    "Polygon Core Resource",
+			network: network.Polygon,
+			worker:  worker.Core,
+			expected: MinimumResource{
+				CPUCore:    2.0,
+				MemoryInGb: 2.0,
+				SizeInGb:   parameter.NetworkCoreSizePerMonth[network.Polygon],
+			},
 		},
 
 		// High-demand-workers + low-demand-networks
 		{
-			name:     "High Demand Worker - Low Demand Network",
-			network:  network.Linea,
-			worker:   worker.Core,
-			expected: baseResource.Mul(4),
+			name:    "High Demand Worker - Low Demand Network",
+			network: network.Linea,
+			worker:  worker.Core,
+			expected: MinimumResource{
+				CPUCore:    1.0,
+				MemoryInGb: 1.0,
+				SizeInGb:   parameter.NetworkCoreSizePerMonth[network.Linea],
+			},
 		},
 		{
 			name:     "High Demand Worker - Low Demand Network (Different)",