71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package pwd
|
|
|
|
import "github.com/play-with-docker/play-with-docker/pwd/types"
|
|
|
|
type mockStorage struct {
|
|
sessionGet func(sessionId string) (*types.Session, error)
|
|
sessionPut func(s *types.Session) error
|
|
sessionCount func() (int, error)
|
|
sessionDelete func(sessionId string) error
|
|
instanceFindByAlias func(sessionPrefix, alias string) (*types.Instance, error)
|
|
instanceFindByIP func(ip string) (*types.Instance, error)
|
|
instanceFindByIPAndSession func(sessionPrefix, ip string) (*types.Instance, error)
|
|
instanceCount func() (int, error)
|
|
clientCount func() (int, error)
|
|
}
|
|
|
|
func (m *mockStorage) SessionGet(sessionId string) (*types.Session, error) {
|
|
if m.sessionGet != nil {
|
|
return m.sessionGet(sessionId)
|
|
}
|
|
return nil, nil
|
|
}
|
|
func (m *mockStorage) SessionPut(s *types.Session) error {
|
|
if m.sessionPut != nil {
|
|
return m.sessionPut(s)
|
|
}
|
|
return nil
|
|
}
|
|
func (m *mockStorage) SessionCount() (int, error) {
|
|
if m.sessionCount != nil {
|
|
return m.sessionCount()
|
|
}
|
|
return 0, nil
|
|
}
|
|
func (m *mockStorage) SessionDelete(sessionId string) error {
|
|
if m.sessionDelete != nil {
|
|
return m.sessionDelete(sessionId)
|
|
}
|
|
return nil
|
|
}
|
|
func (m *mockStorage) InstanceFindByAlias(sessionPrefix, alias string) (*types.Instance, error) {
|
|
if m.instanceFindByAlias != nil {
|
|
return m.instanceFindByAlias(sessionPrefix, alias)
|
|
}
|
|
return nil, nil
|
|
}
|
|
func (m *mockStorage) InstanceFindByIP(ip string) (*types.Instance, error) {
|
|
if m.instanceFindByIP != nil {
|
|
return m.instanceFindByIP(ip)
|
|
}
|
|
return nil, nil
|
|
}
|
|
func (m *mockStorage) InstanceFindByIPAndSession(sessionPrefix, ip string) (*types.Instance, error) {
|
|
if m.instanceFindByIPAndSession != nil {
|
|
return m.instanceFindByIPAndSession(sessionPrefix, ip)
|
|
}
|
|
return nil, nil
|
|
}
|
|
func (m *mockStorage) InstanceCount() (int, error) {
|
|
if m.instanceCount != nil {
|
|
return m.instanceCount()
|
|
}
|
|
return 0, nil
|
|
}
|
|
func (m *mockStorage) ClientCount() (int, error) {
|
|
if m.clientCount != nil {
|
|
return m.clientCount()
|
|
}
|
|
return 0, nil
|
|
}
|