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 } = useDataStore((s) => s)
 
   const selectedNode = useSelectedNode()
 
@@ -163,6 +164,20 @@ export const NodeControls = memo(() => {
     setAnchorEl(null)
   }
 
+  const handleAnalyzeTestCoverage = async (githubName: string) => {
+    try {
+      const res = await analyzeGitHubRepository(githubName)
+
+      if (res) {
+        setSelectedNode(null)
+
+        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 +263,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()
@@ -345,7 +367,7 @@
     setHideNodeDetails: (hideNodeDetails) => set({ hideNodeDetails }),
     setSeedQuestions: (questions) => set({ seedQuestions: questions }),
     updateNode: (updatedNode) => {
       console.info(updatedNode)
     },
     addNewNode: (data) => {
       const { dataInitial: existingData, filters } = get()
@@ -407,7 +429,7 @@
       })
     },
     removeNode: (id) => {
       console.log(id)
     },
 
     setRunningProjectId: (runningProjectId) => set({ runningProjectId, runningProjectMessages: [] }),