Fix context

kylecarbs · kylecarbs · commit c23a42c70549 · 2022-09-22T14:44:52.000Z
diff --git a/enterprise/coderd/licenses_test.go b/enterprise/coderd/licenses_test.go
@@ -78,16 +78,18 @@ func TestGetLicense(t *testing.T) {
 		defer cancel()
 
 		coderdenttest.AddLicense(t, client, coderdenttest.LicenseOptions{
-			AccountID: "testing",
-			AuditLog:  true,
-			SCIM:      true,
+			AccountID:   "testing",
+			AuditLog:    true,
+			SCIM:        true,
+			BrowserOnly: true,
 		})
 
 		coderdenttest.AddLicense(t, client, coderdenttest.LicenseOptions{
-			AccountID: "testing2",
-			AuditLog:  true,
-			SCIM:      true,
-			UserLimit: 200,
+			AccountID:   "testing2",
+			AuditLog:    true,
+			SCIM:        true,
+			BrowserOnly: true,
+			UserLimit:   200,
 		})
 
 		licenses, err := client.Licenses(ctx)
@@ -96,16 +98,18 @@ func TestGetLicense(t *testing.T) {
 		assert.Equal(t, int32(1), licenses[0].ID)
 		assert.Equal(t, "testing", licenses[0].Claims["account_id"])
 		assert.Equal(t, map[string]interface{}{
-			codersdk.FeatureUserLimit: json.Number("0"),
-			codersdk.FeatureAuditLog:  json.Number("1"),
-			codersdk.FeatureSCIM:      json.Number("1"),
+			codersdk.FeatureUserLimit:   json.Number("0"),
+			codersdk.FeatureAuditLog:    json.Number("1"),
+			codersdk.FeatureSCIM:        json.Number("1"),
+			codersdk.FeatureBrowserOnly: json.Number("1"),
 		}, licenses[0].Claims["features"])
 		assert.Equal(t, int32(2), licenses[1].ID)
 		assert.Equal(t, "testing2", licenses[1].Claims["account_id"])
 		assert.Equal(t, map[string]interface{}{
-			codersdk.FeatureUserLimit: json.Number("200"),
-			codersdk.FeatureAuditLog:  json.Number("1"),
-			codersdk.FeatureSCIM:      json.Number("1"),
+			codersdk.FeatureUserLimit:   json.Number("200"),
+			codersdk.FeatureAuditLog:    json.Number("1"),
+			codersdk.FeatureSCIM:        json.Number("1"),
+			codersdk.FeatureBrowserOnly: json.Number("1"),
 		}, licenses[1].Claims["features"])
 	})
 }
diff --git a/enterprise/coderd/workspaceagents.go b/enterprise/coderd/workspaceagents.go
@@ -1,6 +1,7 @@
 package coderd
 
 import (
+	"context"
 	"net/http"
 
 	"github.com/coder/coder/coderd/httpapi"
@@ -12,7 +13,7 @@ func (api *API) shouldBlockNonBrowserConnections(rw http.ResponseWriter) bool {
 	browserOnly := api.entitlements.browserOnly
 	api.entitlementsMu.Unlock()
 	if api.BrowserOnly && browserOnly != codersdk.EntitlementNotEntitled {
-		httpapi.Write(rw, http.StatusConflict, codersdk.Response{
+		httpapi.Write(context.Background(), rw, http.StatusConflict, codersdk.Response{
 			Message: "Non-browser connections are disabled for your deployment.",
 		})
 		return true
