1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# -*- coding: utf-8 -*-
# SPDX-FileCopyrightText: 2021-2023 Greenbone AG
#
# SPDX-License-Identifier: AGPL-3.0-or-later
from pathlib import Path
from unittest import TestCase
from unittest.mock import Mock, patch
from typing import Dict, Optional
from ospd_openvas.gpg_sha_verifier import (
ReloadConfiguration,
create_verify,
gpg_sha256sums,
reload_sha256sums,
)
class GpgTest(TestCase):
@patch("gnupg.GPG")
@patch("pathlib.Path")
def test_reload(self, gmock, pathmock: Path):
def on_failure(_: Optional[Dict[str, str]]) -> Dict[str, str]:
raise Exception( # pylint: disable=broad-exception-raised
"verification_failed"
)
omock = Mock()
emock = Mock()
omock.__enter__ = Mock(return_value=emock)
omock.__exit__ = Mock()
pathmock.open.return_value = omock
emock.readlines.side_effect = [["h hi\n"], ["g gude\n"]]
emock.read.side_effect = [b"hi", b"", b"hi", b"", b"ih", b""]
load = reload_sha256sums(
ReloadConfiguration(
hash_file=pathmock,
on_verification_failure=on_failure,
gpg=gmock,
)
)
self.assertDictEqual(load(), {"h": "hi"})
self.assertDictEqual(load(), {"h": "hi"})
self.assertDictEqual(load(), {"g": "gude"})
gmock.verify_file.side_effect = [False]
with self.assertRaises(Exception):
load()
@patch("gnupg.GPG")
@patch("pathlib.Path")
def test_verifying(self, gmock, pathmock: Path):
omock = Mock()
emock = Mock()
omock.__enter__ = Mock(return_value=emock)
omock.__exit__ = Mock()
pathmock.open.return_value = omock
emock.readlines.side_effect = [["h hi\n", "g gude\n"]]
success_result = gpg_sha256sums(pathmock, gmock)
self.assertIsNotNone(success_result)
self.assertDictEqual(success_result, {"h": "hi", "g": "gude"})
gmock.verify_file.side_effect = [False]
self.assertIsNone(gpg_sha256sums(pathmock, gmock))
@patch("pathlib.Path")
def test_verify_closure(self, pathmock):
shas = (
"98ea6e4f216f2fb4b69fff9b3a44842c38686ca685f3f55dc48c5d3fb1107be4"
)
vsuccess = create_verify(lambda: {shas: "hi.txt"})
omock = Mock()
emock = Mock()
omock.__enter__ = Mock(return_value=emock)
omock.__exit__ = Mock()
pathmock.open.return_value = omock
emock.read.side_effect = [bytes("hi\n", "utf-8"), ""]
pathmock.name = "hi.txt"
self.assertTrue(vsuccess(pathmock))
emock.read.side_effect = [bytes("hi\n", "utf-8"), ""]
pathmock.name = "false.txt"
self.assertFalse(vsuccess(pathmock))
emock.read.side_effect = [bytes("hin", "utf-8"), ""]
pathmock.name = "hi.txt"
self.assertFalse(vsuccess(pathmock))
|
