@elisra-devops/docgen-data-provider 1.63.11 → 1.63.13

package/src/modules/test/ticketsDataProvider.test.ts

@@ -30,9 +30,7 @@ describe('TicketsDataProvider', () => {
       const result = await ticketsDataProvider.FetchImageAsBase64(mockUrl);
 
       // Assert
-      expect(TFSServices.fetchAzureDevOpsImageAsBase64).toHaveBeenCalledWith(
-        mockUrl, mockToken, 'get', null
-      );
+      expect(TFSServices.fetchAzureDevOpsImageAsBase64).toHaveBeenCalledWith(mockUrl, mockToken, 'get', null);
       expect(result).toBe(mockBase64);
     });
   });
@@ -62,7 +60,7 @@ describe('TicketsDataProvider', () => {
       const mockIds = [1, 2];
       const mockWorkItems = [
         { id: 1, fields: { 'System.Title': 'Item 1' } },
-        { id: 2, fields: { 'System.Title': 'Item 2' } }
+        { id: 2, fields: { 'System.Title': 'Item 2' } },
       ];
       const mockLinksMap = new Map();
       mockLinksMap.set('1', { id: '1', rels: ['3'] });
@@ -73,7 +71,8 @@ describe('TicketsDataProvider', () => {
 
       jest.spyOn(ticketsDataProvider, 'PopulateWorkItemsByIds').mockResolvedValueOnce(mockWorkItems);
       jest.spyOn(ticketsDataProvider, 'GetRelationsIds').mockResolvedValueOnce(mockLinksMap);
-      jest.spyOn(ticketsDataProvider, 'GetParentLink')
+      jest
+        .spyOn(ticketsDataProvider, 'GetParentLink')
         .mockResolvedValueOnce(mockTraceItem)
         .mockResolvedValueOnce({ id: '2', title: 'Item 2', url: 'url', customerId: 'customer', links: [] });
       jest.spyOn(ticketsDataProvider, 'PopulateWorkItemsByIds').mockResolvedValueOnce(mockRelatedItems);
@@ -94,11 +93,21 @@ describe('TicketsDataProvider', () => {
       // Arrange
       const mockPath = '';
       const mockDocType = 'STD';
-      const mockQueries = { name: 'Query 1' };
-      const mockResponse = { reqTestTree: {}, testReqTree: {} };
+      const mockQueries = { name: 'Query 1' } as any;
+      const mockBranchesResponse = {
+        reqToTest: { result: { reqTestTree: {} }, usedFolder: {} },
+        testToReq: { result: { testReqTree: {} }, usedFolder: {} },
+        mom: { result: { linkedMomTree: {} }, usedFolder: {} },
+      } as any;
 
       (TFSServices.getItemContent as jest.Mock).mockResolvedValueOnce(mockQueries);
-      jest.spyOn(ticketsDataProvider as any, 'fetchLinkedQueries').mockResolvedValueOnce(mockResponse);
+      jest.spyOn(ticketsDataProvider as any, 'ensureQueryChildren').mockResolvedValueOnce(mockQueries);
+      jest
+        .spyOn(ticketsDataProvider as any, 'getDocTypeRoot')
+        .mockResolvedValueOnce({ root: mockQueries, found: true });
+      jest
+        .spyOn(ticketsDataProvider as any, 'fetchDocTypeBranches')
+        .mockResolvedValueOnce(mockBranchesResponse);
 
       // Act
       const result = await ticketsDataProvider.GetSharedQueries(mockProject, mockPath, mockDocType);
@@ -108,18 +117,30 @@ describe('TicketsDataProvider', () => {
         `${mockOrgUrl}${mockProject}/_apis/wit/queries/Shared%20Queries?$depth=2&$expand=all`,
         mockToken
       );
-      expect(result).toEqual(mockResponse);
+      expect(result).toEqual({
+        reqTestQueries: { reqTestTree: {}, testReqTree: {} },
+        linkedMomQueries: { linkedMomTree: {} },
+      });
     });
 
     it('should fetch SVD shared queries and call fetchAnyQueries', async () => {
       // Arrange
       const mockPath = 'Custom Path';
       const mockDocType = 'SVD';
-      const mockQueries = { name: 'Query 1' };
-      const mockResponse = { systemOverviewQueryTree: {}, knownBugsQueryTree: {} };
+      const mockQueries = { name: 'Query 1' } as any;
+      const mockBranchesResponse = {
+        systemOverview: { result: {}, usedFolder: {} },
+        knownBugs: { result: {}, usedFolder: {} },
+      } as any;
 
       (TFSServices.getItemContent as jest.Mock).mockResolvedValueOnce(mockQueries);
-      jest.spyOn(ticketsDataProvider as any, 'fetchAnyQueries').mockResolvedValueOnce(mockResponse);
+      jest.spyOn(ticketsDataProvider as any, 'ensureQueryChildren').mockResolvedValueOnce(mockQueries);
+      jest
+        .spyOn(ticketsDataProvider as any, 'getDocTypeRoot')
+        .mockResolvedValueOnce({ root: mockQueries, found: true });
+      jest
+        .spyOn(ticketsDataProvider as any, 'fetchDocTypeBranches')
+        .mockResolvedValueOnce(mockBranchesResponse);
 
       // Act
       const result = await ticketsDataProvider.GetSharedQueries(mockProject, mockPath, mockDocType);
@@ -129,7 +150,10 @@ describe('TicketsDataProvider', () => {
         `${mockOrgUrl}${mockProject}/_apis/wit/queries/${mockPath}?$depth=2&$expand=all`,
         mockToken
       );
-      expect(result).toEqual(mockResponse);
+      expect(result).toEqual({
+        systemOverviewQueryTree: {},
+        knownBugsQueryTree: {},
+      });
     });
 
     it('should handle errors', async () => {
@@ -142,8 +166,7 @@ describe('TicketsDataProvider', () => {
       });
 
       // Act & Assert
-      await expect(ticketsDataProvider.GetSharedQueries(mockProject, mockPath))
-        .rejects.toThrow('API error');
+      await expect(ticketsDataProvider.GetSharedQueries(mockProject, mockPath)).rejects.toThrow('API error');
       expect(logger.error).toHaveBeenCalled();
     });
   });
@@ -156,22 +179,21 @@ describe('TicketsDataProvider', () => {
       const mockQueryResult = {
         queryType: QueryType.OneHop,
         columns: [],
-        workItemRelations: [{ source: null, target: { id: 1, url: 'url' } }]
+        workItemRelations: [{ source: null, target: { id: 1, url: 'url' } }],
       };
       const mockTableResult = {
         sourceTargetsMap: new Map(),
         sortingSourceColumnsMap: new Map(),
-        sortingTargetsColumnsMap: new Map()
+        sortingTargetsColumnsMap: new Map(),
       };
 
       (TFSServices.getItemContent as jest.Mock).mockResolvedValueOnce(mockQueryResult);
-      jest.spyOn(ticketsDataProvider as any, 'parseDirectLinkedQueryResultForTableFormat')
+      jest
+        .spyOn(ticketsDataProvider as any, 'parseDirectLinkedQueryResultForTableFormat')
         .mockResolvedValueOnce(mockTableResult);
 
       // Act
-      const result = await ticketsDataProvider.GetQueryResultsFromWiql(
-        mockWiqlHref, true, mockTestCaseMap
-      );
+      const result = await ticketsDataProvider.GetQueryResultsFromWiql(mockWiqlHref, true, mockTestCaseMap);
 
       // Assert
       expect(TFSServices.getItemContent).toHaveBeenCalledWith(mockWiqlHref, mockToken);
@@ -196,13 +218,13 @@ describe('TicketsDataProvider', () => {
         {
           name: 'Query 1',
           _links: { wiql: 'http://example.com/wiql1' },
-          id: 'q1'
+          id: 'q1',
         },
         {
           name: 'Query 2',
           _links: { wiql: null },
-          id: 'q2'
-        }
+          id: 'q2',
+        },
       ];
 
       // Act
@@ -211,7 +233,7 @@ describe('TicketsDataProvider', () => {
       // Assert
       expect(result).toEqual([
         { queryName: 'Query 1', wiql: 'http://example.com/wiql1', id: 'q1' },
-        { queryName: 'Query 2', wiql: null, id: 'q2' }
+        { queryName: 'Query 2', wiql: null, id: 'q2' },
       ]);
     });
   });
@@ -220,8 +242,8 @@ describe('TicketsDataProvider', () => {
     it('should fetch work items in batches of 200', async () => {
       // Arrange
       const mockIds = Array.from({ length: 250 }, (_, i) => i + 1);
-      const mockResponse1 = { value: mockIds.slice(0, 200).map(id => ({ id })) };
-      const mockResponse2 = { value: mockIds.slice(200).map(id => ({ id })) };
+      const mockResponse1 = { value: mockIds.slice(0, 200).map((id) => ({ id })) };
+      const mockResponse2 = { value: mockIds.slice(200).map((id) => ({ id })) };
 
       (TFSServices.getItemContent as jest.Mock)
         .mockResolvedValueOnce(mockResponse1)
@@ -303,7 +325,10 @@ describe('TicketsDataProvider', () => {
 
       // Act
       const result = await ticketsDataProvider.CreateNewWorkItem(
-        mockProject, mockWiBody, mockWiType, mockByPass
+        mockProject,
+        mockWiBody,
+        mockWiType,
+        mockByPass
       );
 
       // Assert
@@ -313,7 +338,7 @@ describe('TicketsDataProvider', () => {
         'POST',
         mockWiBody,
         {
-          'Content-Type': 'application/json-patch+json'
+          'Content-Type': 'application/json-patch+json',
         }
       );
       expect(result).toEqual(mockResponse);
@@ -332,7 +357,10 @@ describe('TicketsDataProvider', () => {
 
       // Act
       const result = await ticketsDataProvider.UpdateWorkItem(
-        mockProject, mockWiBody, mockWorkItemId, mockByPass
+        mockProject,
+        mockWiBody,
+        mockWorkItemId,
+        mockByPass
       );
 
       // Assert
@@ -342,7 +370,7 @@ describe('TicketsDataProvider', () => {
         'patch',
         mockWiBody,
         {
-          'Content-Type': 'application/json-patch+json'
+          'Content-Type': 'application/json-patch+json',
         }
       );
       expect(result).toEqual(mockResponse);
@@ -358,9 +386,9 @@ describe('TicketsDataProvider', () => {
           {
             rel: 'AttachedFile',
             url: 'https://example.com/attachment/1',
-            attributes: { name: 'file.txt' }
-          }
-        ]
+            attributes: { name: 'file.txt' },
+          },
+        ],
       };
 
       // Mock the TFSServices.getItemContent directly to return our mock data
@@ -404,14 +432,14 @@ describe('TicketsDataProvider', () => {
           {
             rel: 'Parent',
             url: 'https://example.com/parent/1',
-            attributes: { name: 'parent' }
+            attributes: { name: 'parent' },
           },
           {
             rel: 'AttachedFile',
             url: 'https://example.com/attachment/1',
-            attributes: { name: 'file.txt' }
-          }
-        ]
+            attributes: { name: 'file.txt' },
+          },
+        ],
       };
 
       // Mock TFSServices.getItemContent directly instead of GetWorkItem
@@ -428,7 +456,7 @@ describe('TicketsDataProvider', () => {
       expect(result.length).toBe(1);
       expect(result[0].rel).toBe('AttachedFile');
       // Only the AttachedFile relation should be in the result
-      expect(result.some(item => item.rel === 'Parent')).toBe(false);
+      expect(result.some((item) => item.rel === 'Parent')).toBe(false);
     });
   });
-});
+});