fix: handle if ORGANIZATION not provided

cleanowners.py

@@ -85,10 +85,11 @@ def main():  # pragma: no cover
         usernames = get_usernames_from_codeowners(codeowners_file_contents)
 
         for username in usernames:
+            org = organization if organization else repo.owner.login
             # Check to see if the username is a member of the organization
-            if not github_connection.organization(organization).is_member(username):
+            if not github_connection.organization(org).is_member(username):
                 print(
-                    f"\t{username} is not a member of {organization}. Suggest removing them from {repo.full_name}"
+                    f"\t{username} is not a member of {org}. Suggest removing them from {repo.full_name}"
                 )
                 users_count += 1
                 if not dry_run:
@@ -143,10 +144,10 @@ def get_repos_iterator(organization, repository_list, github_connection):
         repos = github_connection.organization(organization).repositories()
     else:
         # Get the repositories from the repository_list
-        for repo in repository_list:
-            repos.append(
-                github_connection.repository(repo.split("/")[0], repo.split("/")[1])
-            )
+        for full_repo_path in repository_list:
+            org = full_repo_path.split("/")[0]
+            repo = full_repo_path.split("/")[1]
+            repos.append(github_connection.repository(org, repo))
 
     return repos
 

test_auth.py

@@ -1,4 +1,5 @@
 """Test cases for the auth module."""
+
 import unittest
 from unittest.mock import patch
 

test_cleanowners.py

@@ -111,7 +111,7 @@ def test_get_repos_iterator_with_organization(self, mock_github):
     def test_get_repos_iterator_with_repository_list(self, mock_github):
         """Test the get_repos_iterator function with a repository list"""
         organization = None
-        repository_list = ["org/repo1", "org/repo2"]
+        repository_list = ["org/repo1", "org2/repo2"]
         github_connection = mock_github.return_value
 
         mock_repository = MagicMock()
@@ -123,7 +123,7 @@ def test_get_repos_iterator_with_repository_list(self, mock_github):
         # Assert that the repository method was called with the correct arguments for each repository in the list
         expected_calls = [
             unittest.mock.call("org", "repo1"),
-            unittest.mock.call("org", "repo2"),
+            unittest.mock.call("org2", "repo2"),
         ]
         github_connection.repository.assert_has_calls(expected_calls)