1
0
Fork 0
tuned/tests/unit/storage/test_storage.py
Tomas Korbar 140d982354 Replace python flexmock module with builtin mock
Also replace dependency in specfile and alter dockerfile

Signed-off-by: Tomas Korbar <tkorbar@redhat.com>
2019-11-28 07:54:15 +01:00

31 lines
925 B
Python

import unittest
from unittest.mock import Mock
from unittest.mock import call
import tuned.storage
class StorageStorageTestCase(unittest.TestCase):
def test_set(self):
mock_provider = Mock()
factory = tuned.storage.Factory(mock_provider)
storage = factory.create("foo")
storage.set("optname", "optval")
mock_provider.set.assert_called_once_with("foo", "optname", "optval")
def test_get(self):
mock_provider = Mock()
mock_provider.get.side_effect = [ None, "defval", "somevalue" ]
factory = tuned.storage.Factory(mock_provider)
storage = factory.create("foo")
self.assertEqual(None, storage.get("optname"))
self.assertEqual("defval", storage.get("optname", "defval"))
self.assertEqual("somevalue", storage.get("existing"))
calls = [
call("foo", "optname", None),
call("foo", "optname", "defval"),
call("foo", "existing", None)
]
mock_provider.get.assert_has_calls(calls)