mirror of
https://github.com/tim-beatham/smegmesh.git
synced 2025-08-15 23:57:55 +02:00
Adding stubs and writing tests
This commit is contained in:
73
pkg/robin/requester_test.go
Normal file
73
pkg/robin/requester_test.go
Normal file
@ -0,0 +1,73 @@
|
||||
package robin
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/tim-beatham/wgmesh/pkg/ctrlserver"
|
||||
"github.com/tim-beatham/wgmesh/pkg/ipc"
|
||||
"github.com/tim-beatham/wgmesh/pkg/mesh"
|
||||
)
|
||||
|
||||
func getRequester() *IpcHandler {
|
||||
return &IpcHandler{Server: ctrlserver.NewCtrlServerStub()}
|
||||
}
|
||||
|
||||
func TestCreateMeshRepliesMeshId(t *testing.T) {
|
||||
var reply string
|
||||
requester := getRequester()
|
||||
|
||||
err := requester.CreateMesh(&ipc.NewMeshArgs{
|
||||
IfName: "wg0",
|
||||
WgPort: 5000,
|
||||
Endpoint: "abc.com",
|
||||
}, &reply)
|
||||
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
if len(reply) == 0 {
|
||||
t.Fatalf(`reply should have been returned`)
|
||||
}
|
||||
}
|
||||
|
||||
func TestListMeshesNoMeshesListsEmpty(t *testing.T) {
|
||||
var reply ipc.ListMeshReply
|
||||
requester := getRequester()
|
||||
|
||||
err := requester.ListMeshes("", &reply)
|
||||
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
if len(reply.Meshes) != 0 {
|
||||
t.Fatalf(`meshes should be empty`)
|
||||
}
|
||||
}
|
||||
|
||||
func TestListMeshesMeshesNotEmpty(t *testing.T) {
|
||||
var reply ipc.ListMeshReply
|
||||
requester := getRequester()
|
||||
|
||||
requester.Server.GetMeshManager().AddMesh(&mesh.AddMeshParams{
|
||||
MeshId: "tim123",
|
||||
DevName: "wg0",
|
||||
WgPort: 5000,
|
||||
MeshBytes: make([]byte, 0),
|
||||
})
|
||||
|
||||
err := requester.ListMeshes("", &reply)
|
||||
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
if len(reply.Meshes) != 1 {
|
||||
t.Fatalf(`only only mesh exists`)
|
||||
}
|
||||
|
||||
if reply.Meshes[0] != "tim123" {
|
||||
t.Fatalf(`meshId was %s expected %s`, reply.Meshes[0], "tim123")
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user