refactor paths for windows test fail

Author: tomcat323

packages/core/src/codewhisperer/nextEditPrediction/PredictionTracker.ts

@@ -10,8 +10,6 @@ import * as codewhispererClient from '../client/codewhisperer'
 import { predictionTrackerDefaultConfig } from '../models/constants'
 import globals from '../../shared/extensionGlobals'
 
-const logger = getLogger('nextEditPrediction')
-
 // defaul values are stored in codewhisperer/model/constants
 export interface FileTrackerConfig {
     maxFiles: number
@@ -33,6 +31,7 @@ export interface FileSnapshot {
 
 export class PredictionTracker {
     private snapshots: Map<string, FileSnapshot[]> = new Map()
+    private logger = getLogger('nextEditPrediction')
     readonly config: FileTrackerConfig
     private storageSize: number = 0
 
@@ -81,14 +80,14 @@ export class PredictionTracker {
             fileSnapshots.push(snapshot)
             this.snapshots.set(filePath, fileSnapshots)
             this.storageSize += size
-            logger.debug(
+            this.logger.debug(
                 `Snapshot taken for file: ${filePath}, total snapshots: ${this.getTotalSnapshotCount()}, total size: ${Math.round(this.storageSize / 1024)} KB`
             )
 
             await this.enforceMemoryLimits()
             this.enforceTimeLimits(snapshot)
         } catch (err) {
-            logger.error(`Failed to save snapshot: ${err}`)
+            this.logger.error(`Failed to save snapshot: ${err}`)
         }
     }
 
@@ -110,7 +109,7 @@ export class PredictionTracker {
                 if (fileSnapshots.length === 0) {
                     this.snapshots.delete(snapshot.filePath)
                 }
-                logger.debug(
+                this.logger.debug(
                     `Snapshot deleted (aged out) for file: ${snapshot.filePath}, remaining snapshots: ${this.getTotalSnapshotCount()}, new size: ${Math.round(this.storageSize / 1024)} KB`
                 )
             }
@@ -136,7 +135,7 @@ export class PredictionTracker {
             const removedSnapshot = fileSnapshots.shift()
             if (removedSnapshot) {
                 this.storageSize -= removedSnapshot.size
-                logger.debug(
+                this.logger.debug(
                     `Snapshot deleted (memory limit) for file: ${removedSnapshot.filePath}, remaining snapshots: ${this.getTotalSnapshotCount()}, new size: ${Math.round(this.storageSize / 1024)} KB`
                 )
             }

packages/core/src/test/codewhisperer/nextEditPrediction/PredictionTracker.test.ts

@@ -6,6 +6,7 @@
 import * as vscode from 'vscode'
 import * as sinon from 'sinon'
 import assert from 'assert'
+import * as path from 'path'
 import {
     FileSnapshot,
     FileTrackerConfig,
@@ -51,7 +52,7 @@ describe('PredictionTracker', function () {
         let mockDocument: vscode.TextDocument
 
         beforeEach(function () {
-            filePath = '/path/to/file.js'
+            filePath = path.join('path', 'to', 'file.js')
             previousContent = 'previous content'
             tracker = new PredictionTracker(mockExtensionContext)
 
@@ -126,8 +127,8 @@ describe('PredictionTracker', function () {
             }
             tracker = new PredictionTracker(mockExtensionContext, customConfig)
 
-            const file1 = '/path/to/file1.js'
-            const file2 = '/path/to/file2.js'
+            const file1 = path.join('path', 'to', 'file1.js')
+            const file2 = path.join('path', 'to', 'file2.js')
 
             const initialTime = globals.clock.Date.now()
 
@@ -160,12 +161,12 @@ describe('PredictionTracker', function () {
         })
 
         it('should return empty array for non-existent file', function () {
-            const result = tracker.getFileSnapshots('/non-existent/file.js')
+            const result = tracker.getFileSnapshots(path.join('non-existent', 'file.js'))
             assert.deepStrictEqual(result, [])
         })
 
         it('should return snapshots for existing file', async function () {
-            const file = '/path/to/file.js'
+            const file = path.join('path', 'to', 'file.js')
             const content = 'file content'
             const mockDocument = createMockDocument(content, file)
             await tracker.processEdit(mockDocument, content)
@@ -184,7 +185,7 @@ describe('PredictionTracker', function () {
 
         beforeEach(async function () {
             tracker = new PredictionTracker(mockExtensionContext)
-            file = '/path/to/file.js'
+            file = path.join('path', 'to', 'file.js')
             snapshotContent = 'snapshot content'
             const mockDocument = createMockDocument(snapshotContent, file)
             await tracker.processEdit(mockDocument, snapshotContent)
@@ -207,7 +208,7 @@ describe('PredictionTracker', function () {
 
             // Mock active editor, we only care about document
             mockEditor = {
-                document: createMockDocument('current content', '/path/to/active.js'),
+                document: createMockDocument('current content', path.join('path', 'to', 'active.js')),
                 selection: new vscode.Selection(0, 0, 0, 0),
                 selections: [new vscode.Selection(0, 0, 0, 0)],
                 options: {},
@@ -233,7 +234,7 @@ describe('PredictionTracker', function () {
         })
 
         it('should generate and return supplemental contexts', async function () {
-            const filePath = '/path/to/active.js'
+            const filePath = path.join('path', 'to', 'active.js')
             const initialTime = globals.clock.Date.now()
 
             const mockDoc = createMockDocument('old content 1', filePath)