[FIX][URGENT] Fix /custom endpoint (#854)

* Fix /custom endpoint

Signed-off-by: Tarun Arora <tarun.arora.030402@gmail.com>

* Fix unittests for /custom endpoint

Signed-off-by: Tarun Arora <tarun.arora.030402@gmail.com>

---------

Signed-off-by: Tarun Arora <tarun.arora.030402@gmail.com>

Author: Tarun-Arora

mod_customized/controllers.py

@@ -52,12 +52,11 @@ def index():
         repository = gh.get_repo(f"{username}/{g.github['repository']}")
         # Only commits since last month
         last_month = datetime.now() - timedelta(days=30)
-        commit_since = last_month.isoformat() + 'Z'
-        commits = repository.get_commits(since=commit_since)
+        commits = repository.get_commits(since=last_month)
         commit_arr = []
         for commit in commits:
-            commit_url = commit['html_url']
-            commit_sha = commit['sha']
+            commit_url = commit.html_url
+            commit_sha = commit.sha
             commit_option = f'<a href="{commit_url}">{commit_sha}</a>'
             commit_arr.append((commit_sha, commit_option))
         # If there are commits present, display it on webpage

tests/test_customized/test_controllers.py

@@ -118,6 +118,7 @@ def test_customize_test_creates_with_multiple_platforms(self, mock_user, mock_re
     def test_customize_test_creates_with_select_arr(self, mock_user, mock_repo):
         """Test customize test creation with commits list."""
         from flask import g
+        from github.Commit import Commit
 
         import mod_customized.controllers
         reload(mod_customized.controllers)
@@ -128,14 +129,17 @@ def test_customize_test_creates_with_select_arr(self, mock_user, mock_repo):
         for i in range(num_commits):
             commit_hash = self.create_random_string()
             url = f"https://github.com/{return_git_user()}/{g.github['repository']}/commit/{commit_hash}"
-            commits.append({'html_url': url, 'sha': commit_hash})
+            new_commit = mock.MagicMock(Commit)
+            new_commit.sha = commit_hash
+            new_commit.html_url = url
+            commits.append(new_commit)
         with self.app.test_client() as c:
             c.post('/account/login', data=self.create_login_form_data(self.user.email, self.user.password))
 
             mock_repo.return_value.get_commits.return_value = commits
             response = c.get('/custom/')
             for commit in commits:
-                self.assertIn(commit['sha'], str(response.data))
+                self.assertIn(commit.sha, str(response.data))
 
     def test_customize_regression_tests_load(self, mock_user, mock_repo):
         """Test loading of the regression tests."""