Replace "Generate Tests" with "Analyze Test Coverage" in Repository Node

src/components/Universe/Graph/UI/NodeControls/index.tsx

@@ -24,6 +24,7 @@ import { useUserStore } from '~/stores/useUserStore'
 import { NodeExtended } from '~/types'
 import { colors } from '~/utils/colors'
 import { buttonColors } from './constants'
+import { analyzeGitHubRepository } from '~/network/fetchSourcesData'
 
 const reuseableVector3 = new Vector3()
 
@@ -38,7 +39,7 @@ export const NodeControls = memo(() => {
   const { open: createBountyModal } = useModal('createBounty')
 
   const [isAdmin] = useUserStore((s) => [s.isAdmin])
-  const [addNewNode] = useDataStore((s) => [s.addNewNode])
+  const { addNewNode, setGraph, resetData } = useDataStore((s) => s)
 
   const selectedNode = useSelectedNode()
 
@@ -163,6 +164,21 @@ export const NodeControls = memo(() => {
     setAnchorEl(null)
   }
 
+  const handleAnalyzeTestCoverage = async (githubName: string) => {
+    try {
+      const res = await analyzeGitHubRepository(githubName)
+
+      if (res) {
+        resetData()
+        setSelectedNode(null)
+        addNewNode({ nodes: res.functions, edges: [] })
+        setGraph({ nodes: res.functions })
+      }
+    } catch (error) {
+      console.error('error during test coverage analysis:', error)
+    }
+  }
+
   const open = Boolean(anchorEl)
 
   const id = open ? 'simple-popover' : undefined
@@ -248,13 +264,17 @@ export const NodeControls = memo(() => {
               <PopoverOption
                 data-testid="generate_tests"
                 onClick={() => {
+                  if (selectedNode?.name) {
+                    handleAnalyzeTestCoverage(selectedNode.name)
+                  }
+
                   handleClose()
                 }}
               >
                 <IconWrapper>
                   <AddCircleIcon data-testid="AddCircleIcon" />
                 </IconWrapper>
-                Generate Tests
+                Analyze Test Coverage
               </PopoverOption>
               <PopoverOption
                 data-testid="add_comments"

src/network/fetchSourcesData/index.ts

@@ -180,6 +180,18 @@ export interface UpdateSchemaParams {
   }
 }
 
+type GithubRepositoryResponse = {
+  functions: NodeExtended[]
+}
+
+export const analyzeGitHubRepository = async (github_repository: string): Promise<GithubRepositoryResponse> => {
+  const url = `/github/analyze?github_repository=${github_repository}&analysis=["coverage"]`
+
+  const res = await api.get<GithubRepositoryResponse>(url)
+
+  return res
+}
+
 export const getNodes = async (): Promise<FetchDataResponse> => {
   const url = `/prediction/graph/search?node_type=['Episode']&include_properties=true&includeContent=true&sort_by=date`
 

src/stores/useDataStore/index.ts

@@ -86,6 +86,7 @@
   abortFetchData: () => void
   resetGraph: () => void
   resetData: () => void
+  setGraph: (graph: { nodes: NodeExtended[] }) => void
 }
 
 const defaultData: Omit<
@@ -117,6 +118,7 @@
   | 'abortFetchData'
   | 'resetGraph'
   | 'resetData'
+  | 'setGraph'
 > = {
   categoryFilter: null,
   dataInitial: null,
@@ -318,6 +320,26 @@
       })
     },
 
+    setGraph: (data: { nodes: NodeExtended[] }) => {
+      const uniqueNodes = deduplicateByRefId(data.nodes)
+
+      const nodeTypes = [...new Set(uniqueNodes.map((node) => node.node_type))]
+      const sidebarFilters = ['all', ...nodeTypes.map((type) => type.toLowerCase())]
+
+      const sidebarFilterCounts = sidebarFilters.map((filter) => ({
+        name: filter,
+        count: uniqueNodes.filter((node) => filter === 'all' || node.node_type?.toLowerCase() === filter).length,
+      }))
+
+      set({
+        dataInitial: { nodes: uniqueNodes, links: [] },
+        dataNew: { nodes: uniqueNodes, links: [] },
+        nodeTypes,
+        sidebarFilters,
+        sidebarFilterCounts,
+      })
+    },
+
     nextPage: () => {
       const { filters, fetchData, setAbortRequests } = get()
       const { setBudget } = useUserStore.getState()
@@ -336,20 +358,20 @@
     setTrendingTopics: (trendingTopics) => set({ trendingTopics }),
     setStats: (stats) => set({ stats }),
     setIsFetching: (isFetching) => set({ isFetching }),
 
     setCategoryFilter: (categoryFilter) => set({ categoryFilter }),
     setQueuedSources: (queuedSources) => set({ queuedSources }),
     setSidebarFilter: (sidebarFilter: string) => set({ sidebarFilter }),
     setSelectedTimestamp: (selectedTimestamp) => set({ selectedTimestamp }),
     setSources: (sources) => set({ sources }),
     setHideNodeDetails: (hideNodeDetails) => set({ hideNodeDetails }),
     setSeedQuestions: (questions) => set({ seedQuestions: questions }),
     updateNode: (updatedNode) => {
       console.info(updatedNode)
     },
     addNewNode: (data) => {
       const { dataInitial: existingData, filters } = get()
 
       if (!data?.nodes) {
         return
       }
@@ -392,7 +414,7 @@
       const sidebarFilters = ['all', ...nodeTypes.map((type) => type.toLowerCase())]
 
       // Step 7: Calculate sidebar filter counts
       const sidebarFilterCounts = sidebarFilters.map((filter) => ({
         name: filter,
         count: updatedNodes.filter((node) => filter === 'all' || node.node_type?.toLowerCase() === filter).length,
       }))