mirror of https://github.com/ceph/go-ceph
122 lines
3.1 KiB
Go
122 lines
3.1 KiB
Go
package admin
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/ceph/go-ceph/internal/admintest"
|
|
)
|
|
|
|
var (
|
|
radosConnector = admintest.NewConnector()
|
|
|
|
// some tests are sensitive to the server version
|
|
serverVersion string
|
|
)
|
|
|
|
const (
|
|
cephNautilus = "nautilus"
|
|
cephOctopus = "octopus"
|
|
cephPacfic = "pacific"
|
|
cephQuincy = "quincy"
|
|
)
|
|
|
|
func init() {
|
|
switch vname := os.Getenv("CEPH_VERSION"); vname {
|
|
case cephNautilus, cephOctopus, cephPacfic, cephQuincy:
|
|
serverVersion = vname
|
|
}
|
|
}
|
|
|
|
func TestServerSentinel(t *testing.T) {
|
|
// there probably *is* a better way to do this but I'm doing what's easy
|
|
// and expedient at the moment. That's tying the tests to the environment
|
|
// var to tell us what version of the *server* we are testing against. The
|
|
// build tags control what version of the *client libs* we use. These
|
|
// happen to be the same for our CI tests today, but its a lousy way to
|
|
// organize things IMO.
|
|
// This check is intended to fail the test suite if you don't tell it a
|
|
// server version it expects and force us to update the tests if a new
|
|
// version of ceph is added.
|
|
if serverVersion == "" {
|
|
t.Fatalf("server must be nautilus, octopus, pacific, or quincy (do the tests need updating?)")
|
|
}
|
|
}
|
|
|
|
func getFSAdmin(t *testing.T) *FSAdmin {
|
|
return NewFromConn(radosConnector.Get(t))
|
|
}
|
|
|
|
func newFSAdmin(t *testing.T, configFile string) *FSAdmin {
|
|
return NewFromConn(
|
|
admintest.WrapConn(admintest.NewConnFromConfig(t, configFile)))
|
|
}
|
|
|
|
func TestInvalidFSAdmin(t *testing.T) {
|
|
fsa := &FSAdmin{}
|
|
res := fsa.rawMgrCommand([]byte("FOOBAR!"))
|
|
assert.Error(t, res.Unwrap())
|
|
}
|
|
|
|
type badMarshalType bool
|
|
|
|
func (badMarshalType) MarshalJSON() ([]byte, error) {
|
|
return nil, errors.New("Zowie! wow")
|
|
}
|
|
|
|
func TestBadMarshal(t *testing.T) {
|
|
fsa := getFSAdmin(t)
|
|
|
|
var bad badMarshalType
|
|
res := fsa.marshalMgrCommand(bad)
|
|
assert.Error(t, res.Unwrap())
|
|
}
|
|
|
|
func TestParseListNames(t *testing.T) {
|
|
R := newResponse
|
|
t.Run("error", func(t *testing.T) {
|
|
_, err := parseListNames(R(nil, "", errors.New("bonk")))
|
|
assert.Error(t, err)
|
|
assert.Equal(t, "bonk", err.Error())
|
|
})
|
|
t.Run("statusSet", func(t *testing.T) {
|
|
_, err := parseListNames(R(nil, "unexpected!", nil))
|
|
assert.Error(t, err)
|
|
})
|
|
t.Run("badJSON", func(t *testing.T) {
|
|
_, err := parseListNames(R([]byte("Foo[[["), "", nil))
|
|
assert.Error(t, err)
|
|
})
|
|
t.Run("ok", func(t *testing.T) {
|
|
l, err := parseListNames(R([]byte(`[{"name":"bob"}]`), "", nil))
|
|
assert.NoError(t, err)
|
|
if assert.Len(t, l, 1) {
|
|
assert.Equal(t, "bob", l[0])
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestCheckEmptyResponseExpected(t *testing.T) {
|
|
R := newResponse
|
|
t.Run("error", func(t *testing.T) {
|
|
err := R(nil, "", errors.New("bonk")).NoData().End()
|
|
assert.Error(t, err)
|
|
assert.Equal(t, "bonk", err.Error())
|
|
})
|
|
t.Run("statusSet", func(t *testing.T) {
|
|
err := R(nil, "unexpected!", nil).NoData().End()
|
|
assert.Error(t, err)
|
|
})
|
|
t.Run("someJSON", func(t *testing.T) {
|
|
err := R([]byte(`{"trouble": true}`), "", nil).NoData().End()
|
|
assert.Error(t, err)
|
|
})
|
|
t.Run("ok", func(t *testing.T) {
|
|
err := R([]byte{}, "", nil).NoData().End()
|
|
assert.NoError(t, err)
|
|
})
|
|
}
|