134 lines
3.2 KiB
Go
134 lines
3.2 KiB
Go
//
|
|
// Copyright (c) 2018
|
|
// Mainflux
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
package api_test
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mainflux/mainflux"
|
|
adapter "github.com/mainflux/mainflux/http"
|
|
"github.com/mainflux/mainflux/http/api"
|
|
"github.com/mainflux/mainflux/http/mocks"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func newService() mainflux.MessagePublisher {
|
|
pub := mocks.NewPublisher()
|
|
return adapter.New(pub)
|
|
}
|
|
|
|
func newHTTPServer(pub mainflux.MessagePublisher, cc mainflux.ThingsServiceClient) *httptest.Server {
|
|
mux := api.MakeHandler(pub, cc)
|
|
return httptest.NewServer(mux)
|
|
}
|
|
|
|
type testRequest struct {
|
|
client *http.Client
|
|
method string
|
|
url string
|
|
contentType string
|
|
token string
|
|
body io.Reader
|
|
}
|
|
|
|
func (tr testRequest) make() (*http.Response, error) {
|
|
req, err := http.NewRequest(tr.method, tr.url, tr.body)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if tr.token != "" {
|
|
req.Header.Set("Authorization", tr.token)
|
|
}
|
|
if tr.contentType != "" {
|
|
req.Header.Set("Content-Type", tr.contentType)
|
|
}
|
|
return tr.client.Do(req)
|
|
}
|
|
|
|
func TestPublish(t *testing.T) {
|
|
chanID := "1"
|
|
contentType := "application/senml+json"
|
|
token := "auth_token"
|
|
invalidToken := "invalid_token"
|
|
msg := `[{"n":"current","t":-1,"v":1.6}]`
|
|
thingsClient := mocks.NewThingsClient(map[string]string{token: chanID})
|
|
pub := newService()
|
|
ts := newHTTPServer(pub, thingsClient)
|
|
defer ts.Close()
|
|
|
|
cases := map[string]struct {
|
|
chanID string
|
|
msg string
|
|
contentType string
|
|
auth string
|
|
status int
|
|
}{
|
|
"publish message": {
|
|
chanID: chanID,
|
|
msg: msg,
|
|
contentType: contentType,
|
|
auth: token,
|
|
status: http.StatusAccepted,
|
|
},
|
|
"publish message without authorization token": {
|
|
chanID: chanID,
|
|
msg: msg,
|
|
contentType: contentType,
|
|
auth: "",
|
|
status: http.StatusForbidden,
|
|
},
|
|
"publish message with invalid authorization token": {
|
|
chanID: chanID,
|
|
msg: msg,
|
|
contentType: contentType,
|
|
auth: invalidToken,
|
|
status: http.StatusForbidden,
|
|
},
|
|
"publish message without content type": {
|
|
chanID: chanID,
|
|
msg: msg,
|
|
contentType: "",
|
|
auth: token,
|
|
status: http.StatusAccepted,
|
|
},
|
|
"publish message to invalid channel": {
|
|
chanID: "",
|
|
msg: msg,
|
|
contentType: contentType,
|
|
auth: token,
|
|
status: http.StatusBadRequest,
|
|
},
|
|
"publish message unable to authorize": {
|
|
chanID: chanID,
|
|
msg: msg,
|
|
contentType: contentType,
|
|
auth: mocks.ServiceErrToken,
|
|
status: http.StatusServiceUnavailable,
|
|
},
|
|
}
|
|
|
|
for desc, tc := range cases {
|
|
req := testRequest{
|
|
client: ts.Client(),
|
|
method: http.MethodPost,
|
|
url: fmt.Sprintf("%s/channels/%s/messages", ts.URL, tc.chanID),
|
|
contentType: tc.contentType,
|
|
token: tc.auth,
|
|
body: strings.NewReader(tc.msg),
|
|
}
|
|
res, err := req.make()
|
|
assert.Nil(t, err, fmt.Sprintf("%s: unexpected error %s", desc, err))
|
|
assert.Equal(t, tc.status, res.StatusCode, fmt.Sprintf("%s: expected status code %d got %d", desc, tc.status, res.StatusCode))
|
|
}
|
|
}
|