mirror of
https://github.com/mainflux/mainflux.git
synced 2025-04-29 13:49:28 +08:00
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
package grpc_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/mainflux/mainflux"
|
|
"github.com/mainflux/mainflux/things"
|
|
grpcapi "github.com/mainflux/mainflux/things/api/grpc"
|
|
"github.com/mainflux/mainflux/things/mocks"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
const (
|
|
port = 8080
|
|
token = "token"
|
|
wrong = "wrong"
|
|
email = "john.doe@email.com"
|
|
)
|
|
|
|
var (
|
|
thing = things.Thing{Type: "app", Name: "test_app", Payload: "test_payload"}
|
|
channel = things.Channel{Name: "test"}
|
|
)
|
|
|
|
func newService(tokens map[string]string) things.Service {
|
|
users := mocks.NewUsersService(tokens)
|
|
thingsRepo := mocks.NewThingRepository()
|
|
channelsRepo := mocks.NewChannelRepository(thingsRepo)
|
|
idp := mocks.NewIdentityProvider()
|
|
return things.New(users, thingsRepo, channelsRepo, idp)
|
|
}
|
|
|
|
func startGRPCServer(svc things.Service, port int) {
|
|
listener, _ := net.Listen("tcp", fmt.Sprintf(":%d", port))
|
|
server := grpc.NewServer()
|
|
mainflux.RegisterThingsServiceServer(server, grpcapi.NewServer(svc))
|
|
go server.Serve(listener)
|
|
}
|
|
|
|
func TestCanAccess(t *testing.T) {
|
|
svc := newService(map[string]string{token: email})
|
|
startGRPCServer(svc, port)
|
|
|
|
oth, _ := svc.AddThing(token, thing)
|
|
cth, _ := svc.AddThing(token, thing)
|
|
sch, _ := svc.CreateChannel(token, channel)
|
|
svc.Connect(token, sch.ID, cth.ID)
|
|
|
|
usersAddr := fmt.Sprintf("localhost:%d", port)
|
|
conn, _ := grpc.Dial(usersAddr, grpc.WithInsecure())
|
|
cli := grpcapi.NewClient(conn)
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
defer cancel()
|
|
|
|
cases := map[string]struct {
|
|
thingKey string
|
|
chanID string
|
|
id string
|
|
code codes.Code
|
|
}{
|
|
"check if connected thing can access existing channel": {cth.Key, sch.ID, cth.ID, codes.OK},
|
|
"check if unconnected thing can access existing channel": {oth.Key, sch.ID, "", codes.PermissionDenied},
|
|
"check if thing with wrong access key can access existing channel": {wrong, sch.ID, "", codes.PermissionDenied},
|
|
"check if connected thing can access non-existent channel": {cth.Key, wrong, "", codes.InvalidArgument},
|
|
}
|
|
|
|
for desc, tc := range cases {
|
|
id, err := cli.CanAccess(ctx, &mainflux.AccessReq{tc.thingKey, tc.chanID})
|
|
e, ok := status.FromError(err)
|
|
assert.True(t, ok, "OK expected to be true")
|
|
assert.Equal(t, tc.id, id.GetValue(), fmt.Sprintf("%s: expected %s got %s", desc, tc.id, id.GetValue()))
|
|
assert.Equal(t, tc.code, e.Code(), fmt.Sprintf("%s: expected %s got %s", desc, tc.code, e.Code()))
|
|
}
|
|
}
|