GetForSession now expects a *types.Session instead of a string

docker/factory.go

@@ -16,7 +16,7 @@ import (
 )
 
 type FactoryApi interface {
-	GetForSession(sessionId string) (DockerApi, error)
+	GetForSession(session *types.Session) (DockerApi, error)
 	GetForInstance(instance *types.Instance) (DockerApi, error)
 }
 

docker/factory_mock.go

@@ -9,8 +9,8 @@ type FactoryMock struct {
 	mock.Mock
 }
 
-func (m *FactoryMock) GetForSession(sessionId string) (DockerApi, error) {
-	args := m.Called(sessionId)
+func (m *FactoryMock) GetForSession(session *types.Session) (DockerApi, error) {
+	args := m.Called(session)
 	return args.Get(0).(DockerApi), args.Error(1)
 }
 

docker/local_cached_factory.go

@@ -25,7 +25,7 @@ type instanceEntry struct {
 	client DockerApi
 }
 
-func (f *localCachedFactory) GetForSession(sessionId string) (DockerApi, error) {
+func (f *localCachedFactory) GetForSession(session *types.Session) (DockerApi, error) {
 	f.rw.Lock()
 	defer f.rw.Unlock()