Added defaultDiskSize setup to hyperkit driver

kubernetes · Jan 14, 2019 · 1e0e050 · 1e0e050
1 parent e6b60fb
commit 1e0e050
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/pkg/drivers/hyperkit/driver.go b/pkg/drivers/hyperkit/driver.go
@@ -36,6 +36,7 @@ import (
 	nfsexports "github.com/johanneswuerbach/nfsexports"
 	hyperkit "github.com/moby/hyperkit/go"
 	"github.com/pkg/errors"
+	"k8s.io/minikube/pkg/minikube/constants"
 	pkgdrivers "k8s.io/minikube/pkg/drivers"
 	commonutil "k8s.io/minikube/pkg/util"
 )
@@ -70,6 +71,7 @@ func NewDriver(hostName, storePath string) *Driver {
 			SSHUser: "docker",
 		},
 		CommonDriver: &pkgdrivers.CommonDriver{},
+		DiskSize:     commonutil.CalculateDiskSizeInMB(constants.DefaultDiskSize),
 	}
 }
 

diff --git a/pkg/drivers/hyperkit/driver_test.go b/pkg/drivers/hyperkit/driver_test.go
@@ -20,6 +20,10 @@ package hyperkit
 
 import (
 	"testing"
+
+	"k8s.io/minikube/pkg/minikube/constants"
+
+	commonutil "k8s.io/minikube/pkg/util"
 )
 
 func Test_portExtraction(t *testing.T) {
@@ -62,6 +66,15 @@ func Test_portExtraction(t *testing.T) {
 	}
 }
 
+func Test_defaultDiskSize(t *testing.T) {
+	expectedDefaultDiscSize := commonutil.CalculateDiskSizeInMB(constants.DefaultDiskSize)
+	driver := NewDriver("", "")
+	got := driver.DiskSize
+	if got != expectedDefaultDiscSize {
+		t.Errorf("Unexpected default disk size got: %v, want: %v", got, expectedDefaultDiscSize)
+	}
+}
+
 func testEq(a, b []int) bool {
 
 	if a == nil && b == nil {