|
1 | 1 | package httpmw_test |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "context" |
4 | 5 | "database/sql" |
| 6 | + "encoding/json" |
| 7 | + "io" |
5 | 8 | "net/http" |
6 | 9 | "net/http/httptest" |
7 | 10 | "testing" |
@@ -61,6 +64,38 @@ func TestWorkspaceAgent(t *testing.T) { |
61 | 64 | require.Equal(t, http.StatusOK, res.StatusCode) |
62 | 65 | }) |
63 | 66 |
|
| 67 | + t.Run("InactiveUser", func(t *testing.T) { |
| 68 | + t.Parallel() |
| 69 | + db, _ := dbtestutil.NewDB(t) |
| 70 | + authToken := uuid.New() |
| 71 | + req, rtr, workspace, _ := setup(t, db, authToken, httpmw.ExtractWorkspaceAgentAndLatestBuild( |
| 72 | + httpmw.ExtractWorkspaceAgentAndLatestBuildConfig{ |
| 73 | + DB: db, |
| 74 | + Optional: false, |
| 75 | + }), |
| 76 | + ) |
| 77 | + |
| 78 | + _, err := db.UpdateUserStatus(context.Background(), database.UpdateUserStatusParams{ |
| 79 | + ID: workspace.OwnerID, |
| 80 | + Status: database.UserStatusSuspended, |
| 81 | + UpdatedAt: dbtime.Now(), |
| 82 | + }) |
| 83 | + require.NoError(t, err) |
| 84 | + |
| 85 | + rw := httptest.NewRecorder() |
| 86 | + req.Header.Set(codersdk.SessionTokenHeader, authToken.String()) |
| 87 | + rtr.ServeHTTP(rw, req) |
| 88 | + |
| 89 | + res := rw.Result() |
| 90 | + defer res.Body.Close() |
| 91 | + require.Equal(t, http.StatusUnauthorized, res.StatusCode) |
| 92 | + body, err := io.ReadAll(res.Body) |
| 93 | + require.NoError(t, err) |
| 94 | + var response codersdk.Response |
| 95 | + require.NoError(t, json.Unmarshal(body, &response)) |
| 96 | + require.Contains(t, response.Message, `User is not active (status = "suspended")`) |
| 97 | + }) |
| 98 | + |
64 | 99 | t.Run("Latest", func(t *testing.T) { |
65 | 100 | t.Parallel() |
66 | 101 | db, _ := dbtestutil.NewDB(t) |
|
0 commit comments