Event refactor

This commit is contained in:
Jonathan Leibiusky @xetorthio
2017-07-18 10:45:05 -03:00
parent 6eaece99c5
commit 4731d8ec98
17 changed files with 198 additions and 125 deletions

View File

@@ -7,6 +7,7 @@ import (
"github.com/play-with-docker/play-with-docker/config"
"github.com/play-with-docker/play-with-docker/docker"
"github.com/play-with-docker/play-with-docker/event"
"github.com/play-with-docker/play-with-docker/pwd/types"
"github.com/stretchr/testify/assert"
)
@@ -36,14 +37,14 @@ func TestSessionNew(t *testing.T) {
scheduledSession = s
}
broadcast := &mockBroadcast{}
ev := event.NewLocalBroker()
storage := &mockStorage{}
storage.sessionPut = func(s *types.Session) error {
saveCalled = true
return nil
}
p := NewPWD(docker, tasks, broadcast, storage)
p := NewPWD(docker, tasks, ev, storage)
before := time.Now()
@@ -166,10 +167,10 @@ func TestSessionSetup(t *testing.T) {
return nil, nil
}
tasks := &mockTasks{}
broadcast := &mockBroadcast{}
ev := event.NewLocalBroker()
storage := &mockStorage{}
p := NewPWD(dock, tasks, broadcast, storage)
p := NewPWD(dock, tasks, ev, storage)
s, e := p.SessionNew(time.Hour, "", "", "")
assert.Nil(t, e)