1#@!#!123s

: / proc / self / root / snap / core18 / 2947 / usr / lib / python3 / dist-packages / probert / tests /
Filename : test_prober.py
back
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())