CID-3144 Persist default branch name

leanix · Nov 8, 2024 · cd0a739 · cd0a739
1 parent 9bdc047
commit cd0a739
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 0 deletions.
diff --git a/src/main/kotlin/net/leanix/githubagent/dto/RepositoryDto.kt b/src/main/kotlin/net/leanix/githubagent/dto/RepositoryDto.kt
@@ -8,6 +8,7 @@ data class RepositoryDto(
     val organizationName: String,
     val description: String?,
     val url: String,
+    val defaultBranch: String?,
     val archived: Boolean,
     val visibility: RepositoryVisibility,
     val updatedAt: String,

diff --git a/src/main/kotlin/net/leanix/githubagent/services/GitHubGraphQLService.kt b/src/main/kotlin/net/leanix/githubagent/services/GitHubGraphQLService.kt
@@ -51,6 +51,7 @@ class GitHubGraphQLService(
                         organizationName = it.owner.login,
                         description = it.description,
                         url = it.url,
+                        defaultBranch = it.defaultBranchRef?.name,
                         archived = it.isArchived,
                         visibility = it.visibility,
                         updatedAt = it.updatedAt,

diff --git a/src/main/resources/graphql/GetRepositories.graphql b/src/main/resources/graphql/GetRepositories.graphql
@@ -10,6 +10,9 @@ query GetRepositories($pageCount: Int!, $cursor: String) {
                 name
                 description
                 url
+                defaultBranchRef{
+                    name
+                }
                 isArchived
                 visibility
                 updatedAt

diff --git a/src/test/kotlin/net/leanix/githubagent/services/GitHubScanningServiceTest.kt b/src/test/kotlin/net/leanix/githubagent/services/GitHubScanningServiceTest.kt
@@ -103,6 +103,7 @@ class GitHubScanningServiceTest {
                     organizationName = "testOrg",
                     description = "A test repository",
                     url = "https://github.com/testRepo",
+                    defaultBranch = "main",
                     archived = false,
                     visibility = RepositoryVisibility.PUBLIC,
                     updatedAt = "2024-01-01T00:00:00Z",
@@ -130,6 +131,7 @@ class GitHubScanningServiceTest {
                     organizationName = "testOrg",
                     description = "A test repository",
                     url = "https://github.com/testRepo",
+                    defaultBranch = "main",
                     archived = false,
                     visibility = RepositoryVisibility.PUBLIC,
                     updatedAt = "2024-01-01T00:00:00Z",
@@ -180,6 +182,7 @@ class GitHubScanningServiceTest {
                     organizationName = "testOrg",
                     description = "A test repository",
                     url = "https://github.com/testRepo",
+                    defaultBranch = "main",
                     archived = false,
                     visibility = RepositoryVisibility.PUBLIC,
                     updatedAt = "2024-01-01T00:00:00Z",