first commit in new package

command/command_test.go

@@ -0,0 +1,95 @@
+package command
+
+import (
+	"errors"
+	"io/ioutil"
+	"os"
+	"testing"
+
+	"github.com/boltdb/bolt"
+	"github.com/micromdm/mdm"
+	"golang.org/x/net/context"
+)
+
+func TestService_NewCommand(t *testing.T) {
+	svc := setupDB(t)
+	mock := &mockPublisher{}
+	svc.publisher = mock
+	passPublisher := func(string, []byte) error { return nil }
+	failPublisher := func(string, []byte) error {
+		return errors.New("failed")
+	}
+
+	tests := []struct {
+		name      string
+		publisher func(string, []byte) error
+		request   *mdm.CommandRequest
+		wantErr   bool
+	}{
+		{
+			name:      "happy path",
+			wantErr:   false,
+			publisher: passPublisher,
+			request: &mdm.CommandRequest{
+				RequestType: "DeviceInformation",
+				UDID:        "foobarbaz",
+				Queries:     []string{"foo", "bar", "baz"},
+			},
+		},
+		{
+			name:      "publish fail",
+			wantErr:   true,
+			publisher: failPublisher,
+			request: &mdm.CommandRequest{
+				RequestType: "DeviceInformation",
+			},
+		},
+		{
+			name:      "empty request",
+			wantErr:   true,
+			publisher: passPublisher,
+		},
+		{
+			name:      "bad payload",
+			wantErr:   true,
+			publisher: passPublisher,
+			request: &mdm.CommandRequest{
+				RequestType: "DevicePropaganda",
+				UDID:        "foobarbaz",
+			},
+		},
+	}
+	for _, tt := range tests {
+		mock.PublishFn = tt.publisher
+		_, err := svc.NewCommand(context.Background(), tt.request)
+		if (err != nil) != tt.wantErr {
+			t.Errorf("%q. CommandService.NewCommand() error = %v, wantErr %v",
+				tt.name, err, tt.wantErr)
+			continue
+		}
+	}
+}
+
+type mockPublisher struct {
+	PublishFn func(string, []byte) error
+}
+
+func (m *mockPublisher) Publish(s string, b []byte) error {
+	return m.PublishFn(s, b)
+}
+
+func setupDB(t *testing.T) *Command {
+	f, _ := ioutil.TempFile("", "bolt-")
+	f.Close()
+	os.Remove(f.Name())
+
+	db, err := bolt.Open(f.Name(), 0777, nil)
+	if err != nil {
+		t.Fatalf("couldn't open bolt, err %s\n", err)
+	}
+	svc, err := New(db, nil)
+	if err != nil {
+		t.Fatalf("couldn't create service, err %s\n", err)
+	}
+	return svc
+}