Skip to content

Commit 922e34f

Browse files
committed
fixed code style
Signed-off-by: Tushar912 <tushar.912u@gmail.com>
1 parent f809a89 commit 922e34f

3 files changed

Lines changed: 13 additions & 10 deletions

File tree

vulnerabilities/importers/elixir_security.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
from vulnerabilities.data_source import Reference
3838
from vulnerabilities.package_managers import HexVersionAPI
3939

40+
4041
class ElixirSecurityDataSource(GitDataSource):
4142
def __enter__(self):
4243
super(ElixirSecurityDataSource, self).__enter__()
@@ -79,8 +80,8 @@ def collect_packages(self):
7980
packages.add(data["package"])
8081

8182
return packages
82-
83-
def generate_all_version_list(self,pkg_name):
83+
84+
def generate_all_version_list(self, pkg_name):
8485
if not getattr(self, 'pkg_manager_api', None):
8586
self.pkg_manager_api = HexVersionAPI()
8687
version_list = self.pkg_manager_api.get(
@@ -105,14 +106,15 @@ def process_file(self, path):
105106
safe_pkg_versions = []
106107
if yaml_file.get("unaffected_versions"):
107108
safe_pkg_versions = self.get_versions_from_range(
108-
yaml_file["patched_versions"] + yaml_file["unaffected_versions"],
109+
yaml_file["patched_versions"] +
110+
yaml_file["unaffected_versions"],
109111
pkg_name,
110112
)
111113
else:
112114
safe_pkg_versions = self.get_versions_from_range(
113115
yaml_file["patched_versions"], pkg_name
114116
)
115-
117+
116118
cve_id = "CVE-"+str(yaml_file["cve"])
117119
safe_purls = []
118120
if safe_pkg_versions is not None:

vulnerabilities/package_managers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -322,6 +322,7 @@ async def fetch(self, owner_repo: str, session) -> None:
322322
print(resp)
323323
self.cache[owner_repo] = [release["ref"].split("/")[-1] for release in resp]
324324

325+
325326
class HexVersionAPI(VersionAPI):
326327
async def load_api(self, pkg_set):
327328
async with ClientSession(raise_for_status=True) as session:
@@ -341,4 +342,3 @@ async def fetch(self, pkg, session):
341342
pass
342343

343344
self.cache[pkg] = versions
344-

vulnerabilities/tests/test_elixir_security.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,8 @@ def setUpClass(cls):
5858
"0.1.2",
5959
"0.1.1",
6060
"0.1.0",
61-
])
62-
def test_generate_all_version_list(self,mock_write):
61+
])
62+
def test_generate_all_version_list(self, mock_write):
6363
package = "coherence"
6464
actual_list = self.data_src.generate_all_version_list(package)
6565
expected_list = [
@@ -76,6 +76,7 @@ def test_generate_all_version_list(self,mock_write):
7676
"0.1.0",
7777
]
7878
assert actual_list == expected_list
79+
7980
@patch('vulnerabilities.package_managers.HexVersionAPI.get',
8081
return_value=[
8182
"0.5.2",
@@ -89,8 +90,8 @@ def test_generate_all_version_list(self,mock_write):
8990
"0.1.2",
9091
"0.1.1",
9192
"0.1.0",
92-
])
93-
def test_process_file(self,mock_write):
93+
])
94+
def test_process_file(self, mock_write):
9495

9596
path = os.path.join(BASE_DIR, "test_data/elixir_security/test_file.yml")
9697
expected_data = Advisory(
@@ -107,7 +108,7 @@ def test_process_file(self,mock_write):
107108
},
108109
vuln_references=[
109110
Reference(reference_id='2aae6e3a-24a3-4d5f-86ff-b964eaf7c6d1',
110-
url="https://github.com/smpallen99/coherence/issues/270")
111+
url="https://github.com/smpallen99/coherence/issues/270")
111112
],
112113
cve_id="CVE-2018-20301",
113114
)

0 commit comments

Comments
 (0)