Fix tests and formatting

RSE-Cambridge · Aug 30, 2019 · acf8ed1 · acf8ed1
1 parent e5ff590
commit acf8ed1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/internal/pkg/v2/dacd/brick_manager_impl/session_action_handler_test.go b/internal/pkg/v2/dacd/brick_manager_impl/session_action_handler_test.go
@@ -32,7 +32,7 @@ func TestSessionActionHandler_handleCreate(t *testing.T) {
 	}
 	action := datamodel.SessionAction{
 		ActionType: datamodel.SessionCreateFilesystem,
-		Session:    datamodel.Session{Name: "test"},
+		Session:    datamodel.Session{Name: "test", ActualSizeBytes: 42},
 	}
 	sessionMutex := mock_store.NewMockMutex(mockCtrl)
 	registry.EXPECT().GetSessionMutex(action.Session.Name).Return(sessionMutex, nil)
@@ -41,8 +41,9 @@ func TestSessionActionHandler_handleCreate(t *testing.T) {
 	registry.EXPECT().GetSession(action.Session.Name).Return(action.Session, nil)
 	fsProvider.EXPECT().Create(action.Session)
 	updatedSession := datamodel.Session{
-		Name:   action.Session.Name,
-		Status: datamodel.SessionStatus{FileSystemCreated: true},
+		Name:            action.Session.Name,
+		Status:          datamodel.SessionStatus{FileSystemCreated: true},
+		ActualSizeBytes: 42,
 	}
 	registry.EXPECT().UpdateSession(updatedSession).Return(updatedSession, nil)
 	updatedAction := datamodel.SessionAction{

diff --git a/internal/pkg/v2/dacd/config/config.go b/internal/pkg/v2/dacd/config/config.go
@@ -26,8 +26,8 @@ func GetBrickManagerConfig(env ReadEnvironemnt) BrickManagerConfig {
 	config := BrickManagerConfig{
 		datamodel.BrickHostName(getHostname(env)),
 		datamodel.PoolName(getString(env, "DAC_POOL_NAME", "default")),
-		getUint(env, "DAC_BRICK_CAPACITY_GB", 1400), // TODO: DAC_DEVICE_CAPACITY_GB
-		getUint(env, "DAC_BRICK_COUNT", 12), // TODO: DEVICE_COUNT
+		getUint(env, "DAC_BRICK_CAPACITY_GB", 1400),             // TODO: DAC_DEVICE_CAPACITY_GB
+		getUint(env, "DAC_BRICK_COUNT", 12),                     // TODO: DEVICE_COUNT
 		getString(env, "DAC_BRICK_ADDRESS_PATTERN", "nvme%dn1"), // TODO: DEVICE_TYPE
 		getBool(env, "DAC_HOST_ENABLED", true),
 	}