bugfix: avoid nil point

projecteru2 · Nov 25, 2020 · beda46b · beda46b
1 parent c306654
commit beda46b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/Makefile b/Makefile
@@ -44,8 +44,10 @@ unit-test:
 	./rpc/. ./lock/etcdlock/... \
 	./auth/simple/... \
 	./cluster/calcium/... \
-	./resources/... \
-	./discovery/helium...
+	./discovery/helium... \
+	./resources/storage/... \
+	./resources/volume/... \
+	./resources/cpumem/...
 
 lint:
 	golangci-lint run
diff --git a/rpc/transform.go b/rpc/transform.go
@@ -531,6 +531,9 @@ func toCoreExecuteWorkloadOptions(b *pb.ExecuteWorkloadOptions) (opts *types.Exe
 }
 
 func toRPCCapacityMessage(msg *types.CapacityMessage) *pb.CapacityMessage {
+	if msg == nil {
+		return nil
+	}
 	caps := map[string]int64{}
 	for nodename, capacity := range msg.NodeCapacities {
 		caps[nodename] = int64(capacity)