Add support for the beta feature endpoint

beta_features.go

@@ -62,3 +62,26 @@ func (bs *BetaFeaturesService) List(ctx context.Context, userId uint) ([]BetaFea
 
 	return getBetaFeaturesResponse.BetaFeatures, resp, err
 }
+
+// Update updates a user's beta_feature
+//
+// Travis CI API docs: https://developer.travis-ci.com/resource/beta_feature#update
+func (bs *BetaFeaturesService) Update(ctx context.Context, userId uint, id uint, enabled bool) (*BetaFeature, *http.Response, error) {
+	u, err := urlWithOptions(fmt.Sprintf("/user/%d/beta_feature/%d", userId, id), nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	req, err := bs.client.NewRequest(http.MethodPatch, u, map[string]bool{"enabled": enabled}, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	var feature BetaFeature
+	resp, err := bs.client.Do(ctx, req, &feature)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return &feature, resp, err
+}

beta_features_integration_test.go

@@ -11,8 +11,11 @@ import (
 	"context"
 	"net/http"
 	"testing"
+	"time"
 )
 
+const integrationBetaFeatureId = 1
+
 func TestBetaFeaturesService_Integration_List(t *testing.T) {
 	_, res, err := integrationClient.BetaFeatures.List(context.TODO(), integrationUserId)
 
@@ -24,3 +27,35 @@ func TestBetaFeaturesService_Integration_List(t *testing.T) {
 		t.Fatalf("invalid http status: %s", res.Status)
 	}
 }
+
+func TestBetaFeaturesService_Integration_Update(t *testing.T) {
+	feature, res, err := integrationClient.BetaFeatures.Update(context.TODO(), integrationUserId, integrationBetaFeatureId, true)
+
+	if err != nil {
+		t.Fatalf("unexpected error occured: %s", err)
+	}
+
+	if res.StatusCode != http.StatusOK {
+		t.Fatalf("invalid http status: %s", res.Status)
+	}
+
+	if feature.Id != integrationBetaFeatureId || feature.Enabled != true {
+		t.Fatalf("unexpected beta feature has returned: %v", feature)
+	}
+
+	time.Sleep(2 * time.Second)
+
+	feature, res, err = integrationClient.BetaFeatures.Update(context.TODO(), integrationUserId, integrationBetaFeatureId, false)
+
+	if err != nil {
+		t.Fatalf("unexpected error occured: %s", err)
+	}
+
+	if res.StatusCode != http.StatusOK {
+		t.Fatalf("invalid http status: %s", res.Status)
+	}
+
+	if feature.Id != integrationBetaFeatureId || feature.Enabled != false {
+		t.Fatalf("unexpected beta feature has returned: %v", feature)
+	}
+}

beta_features_test.go

@@ -33,3 +33,25 @@ func TestBetaFeaturesService_List(t *testing.T) {
 		t.Errorf("BetaFeatures.List returned %+v, want %+v", features, want)
 	}
 }
+
+func TestBetaFeaturesService_Update(t *testing.T) {
+	client, mux, _, teardown := setup()
+	defer teardown()
+
+	mux.HandleFunc(fmt.Sprintf("/user/%d/beta_feature/1", testUserId), func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, http.MethodPatch)
+		testBody(t, r, `{"enabled":true}`+"\n")
+		fmt.Fprint(w, `{"id":1,"name":"dashboard","description":"Try the new personal Dashboard layout","enabled":true,"feedback_url":"https://github.com/travis-ci/beta-features/issues/5"}`)
+	})
+
+	features, _, err := client.BetaFeatures.Update(context.Background(), testUserId, 1, true)
+
+	if err != nil {
+		t.Errorf("BetaFeatures.Update returned error: %v", err)
+	}
+
+	want := &BetaFeature{Id: 1, Name: "dashboard", Description: "Try the new personal Dashboard layout", Enabled: true, FeedbackUrl: "https://github.com/travis-ci/beta-features/issues/5"}
+	if !reflect.DeepEqual(features, want) {
+		t.Errorf("BetaFeatures.Update returned %+v, want %+v", features, want)
+	}
+}