OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	02/18/2024 07:58:21 PM	rwxr-xr-x
📄 __init__.py	0 bytes	07/27/2016 11:12:20 AM	rw-r--r--
📁 __pycache__	-	02/18/2024 07:58:21 PM	rwxr-xr-x
📁 data	-	02/18/2024 07:58:21 PM	rwxr-xr-x
📄 fakes.py	213 bytes	07/27/2016 11:12:20 AM	rw-r--r--
📄 test_network.py	3.93 KB	07/27/2016 11:12:20 AM	rw-r--r--
📄 test_prober.py	1.92 KB	07/27/2016 11:12:20 AM	rw-r--r--
📄 test_storage.py	4.21 KB	07/27/2016 11:12:20 AM	rw-r--r--
📄 test_utils.py	1.2 KB	11/21/2016 01:30:02 AM	rw-r--r--

Editing: test_prober.py

import testtools
import json
import argparse

from mock import patch

from probert.prober import Prober
from probert.storage import Storage
from probert.network import Network
from probert.tests.fakes import FAKE_PROBE_ALL_JSON

class ProbertTestProber(testtools.TestCase):
    def setUp(self):
        super(ProbertTestProber, self).setUp()
        self.results = json.load(open(FAKE_PROBE_ALL_JSON))
        self.options = argparse.Namespace(probe_all=True,
                                     probe_network=False,
                                     probe_storage=False)

def test_prober_init(self):
        p = Prober(options=self.options, results=self.results)
        self.assertNotEqual(p, None)

@patch.object(Prober, 'probe_all')
    def test_prober_probe_all(self, _probe_all):
        p = Prober(options=self.options, results=self.results)
        p.probe()
        self.assertTrue(_probe_all.called)

@patch.object(Prober, 'probe_network')
    @patch.object(Prober, 'probe_storage')
    def test_prober_probe_all_invoke_others(self, _storage, _network):
        p = Prober(options=self.options, results=self.results)
        p.probe()
        self.assertTrue(_storage.called)
        self.assertTrue(_network.called)

def test_prober_get_results(self):
        p = Prober(options=self.options, results=self.results)
        self.assertEqual(self.results, p.get_results())

@patch.object(Network, 'probe')
    @patch.object(Storage, 'probe')
    def test_prober_probe_all_check_results(self, _storage, _network):
        p = Prober(options=self.options, results=self.results)
        results = {
            'storage': {'lambic': 99},
            'network': {'saison': 99},
        }
        _storage.return_value = results['storage']
        _network.return_value = results['network']
        p.probe()
        self.assertTrue(_storage.called)
        self.assertTrue(_network.called)
        self.assertEqual(results, p.get_results())