338 lines
8.0 KiB
Go
338 lines
8.0 KiB
Go
package storage
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/play-with-docker/play-with-docker/pwd/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSessionPut(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
s := &types.Session{Id: "a session"}
|
|
err = storage.SessionPut(s)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
var loadedSessions map[string]*types.Session
|
|
expectedSessions := map[string]*types.Session{}
|
|
expectedSessions[s.Id] = s
|
|
|
|
file, err := os.Open(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
defer file.Close()
|
|
|
|
decoder := json.NewDecoder(file)
|
|
err = decoder.Decode(&loadedSessions)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.EqualValues(t, expectedSessions, loadedSessions)
|
|
}
|
|
|
|
func TestSessionGet(t *testing.T) {
|
|
expectedSession := &types.Session{Id: "session1"}
|
|
sessions := map[string]*types.Session{}
|
|
sessions[expectedSession.Id] = expectedSession
|
|
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
encoder := json.NewEncoder(tmpfile)
|
|
err = encoder.Encode(&sessions)
|
|
assert.Nil(t, err)
|
|
tmpfile.Close()
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
_, err = storage.SessionGet("bad id")
|
|
assert.True(t, NotFound(err))
|
|
|
|
loadedSession, err := storage.SessionGet("session1")
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, expectedSession, loadedSession)
|
|
}
|
|
|
|
func TestSessionGetAll(t *testing.T) {
|
|
s1 := &types.Session{Id: "session1"}
|
|
s2 := &types.Session{Id: "session2"}
|
|
sessions := map[string]*types.Session{}
|
|
sessions[s1.Id] = s1
|
|
sessions[s2.Id] = s2
|
|
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
encoder := json.NewEncoder(tmpfile)
|
|
err = encoder.Encode(&sessions)
|
|
assert.Nil(t, err)
|
|
tmpfile.Close()
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
loadedSessions, err := storage.SessionGetAll()
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, s1, loadedSessions[s1.Id])
|
|
assert.Equal(t, s2, loadedSessions[s2.Id])
|
|
}
|
|
|
|
func TestInstanceFindByIP(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
i1 := &types.Instance{Name: "i1", IP: "10.0.0.1"}
|
|
i2 := &types.Instance{Name: "i2", IP: "10.1.0.1"}
|
|
s1 := &types.Session{Id: "session1", Instances: map[string]*types.Instance{"i1": i1}}
|
|
s2 := &types.Session{Id: "session2", Instances: map[string]*types.Instance{"i2": i2}}
|
|
err = storage.SessionPut(s1)
|
|
assert.Nil(t, err)
|
|
err = storage.SessionPut(s2)
|
|
assert.Nil(t, err)
|
|
|
|
foundInstance, err := storage.InstanceFindByIP("session1", "10.0.0.1")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, i1, foundInstance)
|
|
|
|
foundInstance, err = storage.InstanceFindByIP("session2", "10.1.0.1")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, i2, foundInstance)
|
|
|
|
foundInstance, err = storage.InstanceFindByIP("session3", "10.1.0.1")
|
|
assert.True(t, NotFound(err))
|
|
assert.Nil(t, foundInstance)
|
|
|
|
foundInstance, err = storage.InstanceFindByIP("session1", "10.1.0.1")
|
|
assert.True(t, NotFound(err))
|
|
assert.Nil(t, foundInstance)
|
|
|
|
foundInstance, err = storage.InstanceFindByIP("session1", "192.168.0.1")
|
|
assert.True(t, NotFound(err))
|
|
assert.Nil(t, foundInstance)
|
|
}
|
|
|
|
func TestInstanceGet(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
i1 := &types.Instance{Name: "i1", IP: "10.0.0.1"}
|
|
s1 := &types.Session{Id: "session1", Instances: map[string]*types.Instance{"i1": i1}}
|
|
err = storage.SessionPut(s1)
|
|
assert.Nil(t, err)
|
|
|
|
foundInstance, err := storage.InstanceGet("session1", "i1")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, i1, foundInstance)
|
|
}
|
|
|
|
func TestInstanceGetAllWindows(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
w1 := []*types.WindowsInstance{{ID: "one"}, {ID: "two"}}
|
|
w2 := []*types.WindowsInstance{{ID: "three"}, {ID: "four"}}
|
|
s1 := &types.Session{Id: "session1", WindowsAssigned: w1}
|
|
s2 := &types.Session{Id: "session2", WindowsAssigned: w2}
|
|
err = storage.SessionPut(s1)
|
|
err = storage.SessionPut(s2)
|
|
assert.Nil(t, err)
|
|
|
|
allw, err := storage.InstanceGetAllWindows()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, allw, append(w1, w2...))
|
|
}
|
|
|
|
func TestInstanceCreate(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
i1 := &types.Instance{Name: "i1", IP: "10.0.0.1"}
|
|
s1 := &types.Session{Id: "session1"}
|
|
err = storage.SessionPut(s1)
|
|
assert.Nil(t, err)
|
|
err = storage.InstanceCreate(s1.Id, i1)
|
|
assert.Nil(t, err)
|
|
|
|
loadedSession, err := storage.SessionGet("session1")
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, i1, loadedSession.Instances["i1"])
|
|
|
|
}
|
|
|
|
func TestInstanceCreateWindows(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
s1 := &types.Session{Id: "session1"}
|
|
i1 := &types.WindowsInstance{SessionId: s1.Id, ID: "some id"}
|
|
err = storage.SessionPut(s1)
|
|
assert.Nil(t, err)
|
|
err = storage.InstanceCreateWindows(i1)
|
|
assert.Nil(t, err)
|
|
|
|
loadedSession, err := storage.SessionGet("session1")
|
|
assert.Nil(t, err)
|
|
|
|
assert.Equal(t, i1, loadedSession.WindowsAssigned[0])
|
|
}
|
|
|
|
func TestInstanceDeleteWindows(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
s1 := &types.Session{Id: "session1", WindowsAssigned: []*types.WindowsInstance{{ID: "one"}}}
|
|
err = storage.SessionPut(s1)
|
|
assert.Nil(t, err)
|
|
|
|
err = storage.InstanceDeleteWindows(s1.Id, "one")
|
|
assert.Nil(t, err)
|
|
|
|
found, err := storage.SessionGet(s1.Id)
|
|
assert.Equal(t, 0, len(found.WindowsAssigned))
|
|
}
|
|
|
|
func TestCounts(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
c1 := &types.Client{}
|
|
i1 := &types.Instance{Name: "i1", IP: "10.0.0.1"}
|
|
i2 := &types.Instance{Name: "i2", IP: "10.1.0.1"}
|
|
s1 := &types.Session{Id: "session1", Instances: map[string]*types.Instance{"i1": i1}}
|
|
s2 := &types.Session{Id: "session2", Instances: map[string]*types.Instance{"i2": i2}}
|
|
s3 := &types.Session{Id: "session3", Clients: []*types.Client{c1}}
|
|
|
|
err = storage.SessionPut(s1)
|
|
assert.Nil(t, err)
|
|
err = storage.SessionPut(s2)
|
|
assert.Nil(t, err)
|
|
err = storage.SessionPut(s3)
|
|
assert.Nil(t, err)
|
|
|
|
num, err := storage.SessionCount()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 3, num)
|
|
|
|
num, err = storage.InstanceCount()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, 2, num)
|
|
|
|
}
|
|
|
|
func TestSessionDelete(t *testing.T) {
|
|
tmpfile, err := ioutil.TempFile("", "pwd")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
tmpfile.Close()
|
|
os.Remove(tmpfile.Name())
|
|
defer os.Remove(tmpfile.Name())
|
|
|
|
storage, err := NewFileStorage(tmpfile.Name())
|
|
|
|
assert.Nil(t, err)
|
|
|
|
s1 := &types.Session{Id: "session1"}
|
|
err = storage.SessionPut(s1)
|
|
assert.Nil(t, err)
|
|
|
|
found, err := storage.SessionGet(s1.Id)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, s1, found)
|
|
|
|
err = storage.SessionDelete(s1.Id)
|
|
assert.Nil(t, err)
|
|
|
|
found, err = storage.SessionGet(s1.Id)
|
|
assert.True(t, NotFound(err))
|
|
assert.Nil(t, found)
|
|
}
|