( ctx context.Context, t *testing.T, db database.Store, baseURL string, )
| 185 | } |
| 186 | |
| 187 | func setOpenAIProviderBaseURL( |
| 188 | ctx context.Context, |
| 189 | t *testing.T, |
| 190 | db database.Store, |
| 191 | baseURL string, |
| 192 | ) { |
| 193 | t.Helper() |
| 194 | |
| 195 | providers, err := db.GetAIProviders(ctx, database.GetAIProvidersParams{IncludeDisabled: true}) |
| 196 | require.NoError(t, err) |
| 197 | for _, provider := range providers { |
| 198 | if provider.Type != database.AiProviderTypeOpenai { |
| 199 | continue |
| 200 | } |
| 201 | _, err = db.UpdateAIProvider(ctx, database.UpdateAIProviderParams{ |
| 202 | ID: provider.ID, |
| 203 | Type: provider.Type, |
| 204 | DisplayName: provider.DisplayName, |
| 205 | Enabled: provider.Enabled, |
| 206 | BaseUrl: baseURL, |
| 207 | Settings: provider.Settings, |
| 208 | SettingsKeyID: provider.SettingsKeyID, |
| 209 | }) |
| 210 | require.NoError(t, err) |
| 211 | return |
| 212 | } |
| 213 | require.Fail(t, "openai provider not found") |
| 214 | } |
| 215 | |
| 216 | func TestSubscribeRelayReconnectsOnDrop(t *testing.T) { |
| 217 | t.Parallel() |
no test coverage detected