another test

Signed-off-by: Chris Martin <chris@cmartinit.co.uk>
armadaproject · Jul 1, 2024 · 0962c1c · 0962c1c
1 parent 63f9638
commit 0962c1c
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/internal/common/resource/resource_test.go b/internal/common/resource/resource_test.go
@@ -181,6 +181,23 @@ func TestToProtoMap(t *testing.T) {
 	assert.Equal(t, actual, expected)
 }
 
+func TestFromProtoMap(t *testing.T) {
+	oneCpu := resource.MustParse("1")
+	oneGi := resource.MustParse("1Gi")
+
+	input := map[string]*resource.Quantity{
+		"cpu":    &oneCpu,
+		"memory": &oneGi,
+	}
+	expected := ComputeResources{
+		"cpu":    resource.MustParse("1"),
+		"memory": resource.MustParse("1Gi"),
+	}
+
+	actual := FromProtoMap(input)
+	assert.Equal(t, actual, expected)
+}
+
 func makeDefaultNodeResource() v1.ResourceList {
 	cpuResource := resource.NewQuantity(100, resource.DecimalSI)
 	memoryResource := resource.NewQuantity(50*1024*1024*1024, resource.DecimalSI)