Skip to content

Commit 14a8974

Browse files
committed
test(core): keep existing non-null assertions
1 parent d2f0931 commit 14a8974

1 file changed

Lines changed: 20 additions & 26 deletions

File tree

src/core/task/__tests__/Task.spec.ts

Lines changed: 20 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -477,7 +477,7 @@ describe("Cline", () => {
477477
content: [{ type: "text", text: "test message" }],
478478
},
479479
])
480-
expect(Object.keys(requireDefined(cleanConversationHistory[0]))).toEqual(["role", "content"])
480+
expect(Object.keys(cleanConversationHistory[0]!)).toEqual(["role", "content"])
481481
})
482482

483483
it("should shape image blocks for API compatibility before request construction", async () => {
@@ -1837,9 +1837,7 @@ describe("Cline", () => {
18371837

18381838
expect(summarizeConversation).toHaveBeenCalled()
18391839
const [options] = vi.mocked(summarizeConversation).mock.calls.at(-1)!
1840-
expect(options.metadata?.abortSignal).toBe(
1841-
requireDefined(task.currentRequestAbortController).signal,
1842-
)
1840+
expect(options.metadata?.abortSignal).toBe(task.currentRequestAbortController!.signal)
18431841
})
18441842

18451843
it("should omit abortSignal from condenseContext metadata when no current request exists", async () => {
@@ -1872,7 +1870,7 @@ describe("Cline", () => {
18721870
vi.spyOn(getTaskTestAccess(task), "getSystemPrompt").mockResolvedValue("mock system prompt")
18731871

18741872
vi.spyOn(task.api, "getModel").mockReturnValue({
1875-
id: requireDefined(mockApiConfig.apiModelId),
1873+
id: mockApiConfig.apiModelId!,
18761874
info: {
18771875
supportsImages: false,
18781876
supportsPromptCache: true,
@@ -1926,9 +1924,7 @@ describe("Cline", () => {
19261924
const [, , metadata] = createMessageSpy.mock.calls[0]!
19271925

19281926
expect(metadata).toBeDefined()
1929-
expect(requireDefined(metadata).abortSignal).toBe(
1930-
requireDefined(task.currentRequestAbortController).signal,
1931-
)
1927+
expect(metadata!.abortSignal).toBe(task.currentRequestAbortController!.signal)
19321928
})
19331929

19341930
it("configures tool restrictions for Gemini requests", async () => {
@@ -2010,7 +2006,7 @@ describe("Cline", () => {
20102006

20112007
vi.spyOn(getTaskTestAccess(task), "getSystemPrompt").mockResolvedValue("mock system prompt")
20122008
vi.spyOn(task.api, "getModel").mockReturnValue({
2013-
id: requireDefined(mockApiConfig.apiModelId),
2009+
id: mockApiConfig.apiModelId!,
20142010
info: {
20152011
supportsImages: false,
20162012
supportsPromptCache: true,
@@ -2095,7 +2091,7 @@ describe("Cline", () => {
20952091
vi.spyOn(getTaskTestAccess(task), "getSystemPrompt").mockResolvedValue("mock system prompt")
20962092

20972093
vi.spyOn(task.api, "getModel").mockReturnValue({
2098-
id: requireDefined(mockApiConfig.apiModelId),
2094+
id: mockApiConfig.apiModelId!,
20992095
info: {
21002096
supportsImages: false,
21012097
supportsPromptCache: true,
@@ -2146,10 +2142,10 @@ describe("Cline", () => {
21462142

21472143
// Get the signal from metadata
21482144
const [, , metadata] = createMessageSpy.mock.calls[0]!
2149-
const metadataSignal = requireDefined(metadata).abortSignal
2145+
const metadataSignal = metadata!.abortSignal
21502146

21512147
// The signal in metadata should be the same as the one from currentRequestAbortController
2152-
expect(metadataSignal).toBe(requireDefined(task.currentRequestAbortController).signal)
2148+
expect(metadataSignal).toBe(task.currentRequestAbortController!.signal)
21532149
})
21542150

21552151
it("should omit createMessage abortSignal metadata when no current request exists before condense metadata checks", async () => {
@@ -2162,7 +2158,7 @@ describe("Cline", () => {
21622158

21632159
vi.spyOn(getTaskTestAccess(task), "getSystemPrompt").mockResolvedValue("mock system prompt")
21642160
vi.spyOn(task.api, "getModel").mockReturnValue({
2165-
id: requireDefined(mockApiConfig.apiModelId),
2161+
id: mockApiConfig.apiModelId!,
21662162
info: {
21672163
supportsImages: false,
21682164
supportsPromptCache: true,
@@ -2213,10 +2209,8 @@ describe("Cline", () => {
22132209

22142210
const [, , metadata] = createMessageSpy.mock.calls[0]!
22152211
expect(metadata).toBeDefined()
2216-
expect("abortSignal" in requireDefined(metadata)).toBe(true)
2217-
expect(requireDefined(metadata).abortSignal).toBe(
2218-
requireDefined(task.currentRequestAbortController).signal,
2219-
)
2212+
expect("abortSignal" in metadata!).toBe(true)
2213+
expect(metadata!.abortSignal).toBe(task.currentRequestAbortController!.signal)
22202214
})
22212215

22222216
it("should keep createMessage abortSignal metadata unaborted before cancellation", async () => {
@@ -2229,7 +2223,7 @@ describe("Cline", () => {
22292223

22302224
vi.spyOn(getTaskTestAccess(task), "getSystemPrompt").mockResolvedValue("mock system prompt")
22312225
vi.spyOn(task.api, "getModel").mockReturnValue({
2232-
id: requireDefined(mockApiConfig.apiModelId),
2226+
id: mockApiConfig.apiModelId!,
22332227
info: {
22342228
supportsImages: false,
22352229
supportsPromptCache: true,
@@ -2277,7 +2271,7 @@ describe("Cline", () => {
22772271
await iterator.next()
22782272

22792273
const [, , metadata] = createMessageSpy.mock.calls[0]!
2280-
expect(metadata?.abortSignal).toBe(requireDefined(task.currentRequestAbortController).signal)
2274+
expect(metadata?.abortSignal).toBe(task.currentRequestAbortController!.signal)
22812275
expect(metadata?.abortSignal?.aborted).toBe(false)
22822276
})
22832277
})
@@ -2292,7 +2286,7 @@ describe("Cline", () => {
22922286

22932287
vi.spyOn(getTaskTestAccess(task), "getSystemPrompt").mockResolvedValue("mock system prompt")
22942288
vi.spyOn(task.api, "getModel").mockReturnValue({
2295-
id: requireDefined(mockApiConfig.apiModelId),
2289+
id: mockApiConfig.apiModelId!,
22962290
info: {
22972291
supportsImages: false,
22982292
supportsPromptCache: true,
@@ -2334,9 +2328,9 @@ describe("Cline", () => {
23342328

23352329
expect(createMessageSpy).toHaveBeenCalledTimes(1)
23362330
const [, , metadata1] = createMessageSpy.mock.calls[0]!
2337-
const signal1 = requireDefined(metadata1).abortSignal
2331+
const signal1 = metadata1!.abortSignal
23382332
expect(signal1).toBeDefined()
2339-
expect(requireDefined(signal1).aborted).toBe(false)
2333+
expect(signal1!.aborted).toBe(false)
23402334

23412335
// Simulate request completion and cancellation to clear the controller
23422336
task.cancelCurrentRequest()
@@ -2348,12 +2342,12 @@ describe("Cline", () => {
23482342

23492343
expect(createMessageSpy).toHaveBeenCalledTimes(2)
23502344
const [, , metadata2] = createMessageSpy.mock.calls[1]!
2351-
const signal2 = requireDefined(metadata2).abortSignal
2345+
const signal2 = metadata2!.abortSignal
23522346

23532347
// Signals should be different instances (fresh controller per request)
23542348
expect(signal2).not.toBe(signal1)
2355-
expect(signal2).toBe(requireDefined(task.currentRequestAbortController).signal)
2356-
expect(requireDefined(signal2).aborted).toBe(false)
2349+
expect(signal2).toBe(task.currentRequestAbortController!.signal)
2350+
expect(signal2!.aborted).toBe(false)
23572351
})
23582352

23592353
it("should propagate AbortController signal through attemptApiRequest context-window retry path", async () => {
@@ -2372,7 +2366,7 @@ describe("Cline", () => {
23722366
contextTokens: 120000,
23732367
})
23742368
vi.spyOn(task.api, "getModel").mockReturnValue({
2375-
id: requireDefined(mockApiConfig.apiModelId),
2369+
id: mockApiConfig.apiModelId!,
23762370
info: {
23772371
supportsImages: false,
23782372
supportsPromptCache: true,

0 commit comments

Comments
 (0)