@@ -13439,76 +13439,6 @@ If a workspace is needed, use list_templates before create_workspace and follow
1343913439 require.Empty(t, mAudit.AuditLogs())
1344013440 })
1344113441
13442- // A baseline read failure rejects the write and records the failed attempt.
13443- t.Run("AuditOldCaptureFailureRejectsWrite", func(t *testing.T) {
13444- ctx := testutil.Context(t, testutil.WaitLong)
13445-
13446- rawDB, pubsub := dbtestutil.NewDB(t)
13447- store := newFailNextChatSystemPromptStore(rawDB)
13448- mAudit := audit.NewMock()
13449- rawClient, _, api := coderdtest.NewWithAPI(t, &coderdtest.Options{
13450- Database: store,
13451- Pubsub: pubsub,
13452- DeploymentValues: coderdtest.DeploymentValues(t),
13453- Auditor: mAudit,
13454- })
13455- aibridgedtest.StartTestAIBridgeDaemon(t.Context(), t, api, nil)
13456- client := codersdk.NewExperimentalClient(rawClient)
13457- _ = coderdtest.CreateFirstUser(t, client.Client)
13458- // Discard the login entry emitted by user creation.
13459- mAudit.ResetLogs()
13460-
13461- store.failNextGetChatSystemPromptConfig.Store(true)
13462- err := client.UpdateChatSystemPrompt(ctx, codersdk.UpdateChatSystemPromptRequest{
13463- SystemPrompt: "Must not be stored.",
13464- })
13465- requireSDKError(t, err, http.StatusInternalServerError)
13466-
13467- resp, err := client.GetChatSystemPrompt(ctx)
13468- require.NoError(t, err)
13469- require.Empty(t, resp.SystemPrompt)
13470-
13471- logs := mAudit.AuditLogs()
13472- require.Len(t, logs, 1)
13473- require.EqualValues(t, http.StatusInternalServerError, logs[0].StatusCode)
13474- require.JSONEq(t, "{}", string(logs[0].Diff))
13475- })
13476-
13477- // A transaction setup failure rejects the write and records the failed attempt.
13478- t.Run("AuditTransactionFailureRejectsWrite", func(t *testing.T) {
13479- ctx := testutil.Context(t, testutil.WaitLong)
13480-
13481- rawDB, pubsub := dbtestutil.NewDB(t)
13482- store := newFailNextChatInstructionTransactionStore(rawDB)
13483- mAudit := audit.NewMock()
13484- rawClient, _, api := coderdtest.NewWithAPI(t, &coderdtest.Options{
13485- Database: store,
13486- Pubsub: pubsub,
13487- DeploymentValues: coderdtest.DeploymentValues(t),
13488- Auditor: mAudit,
13489- })
13490- aibridgedtest.StartTestAIBridgeDaemon(t.Context(), t, api, nil)
13491- client := codersdk.NewExperimentalClient(rawClient)
13492- _ = coderdtest.CreateFirstUser(t, client.Client)
13493- // Discard the login entry emitted by user creation.
13494- mAudit.ResetLogs()
13495-
13496- store.failNextTransaction.Store(true)
13497- err := client.UpdateChatSystemPrompt(ctx, codersdk.UpdateChatSystemPromptRequest{
13498- SystemPrompt: "Must not be stored.",
13499- })
13500- requireSDKError(t, err, http.StatusInternalServerError)
13501-
13502- resp, err := client.GetChatSystemPrompt(ctx)
13503- require.NoError(t, err)
13504- require.Empty(t, resp.SystemPrompt)
13505-
13506- logs := mAudit.AuditLogs()
13507- require.Len(t, logs, 1)
13508- require.EqualValues(t, http.StatusInternalServerError, logs[0].StatusCode)
13509- require.JSONEq(t, "{}", string(logs[0].Diff))
13510- })
13511-
1351213442 // A lock failure rejects the write and records the failed attempt.
1351313443 t.Run("AuditLockFailureRejectsWrite", func(t *testing.T) {
1351413444 ctx := testutil.Context(t, testutil.WaitLong)
0 commit comments