def test_update_invalid(self):
data = pd.DriverPrivateData(storage=self.fake_storage)
details = ["invalid"]
self.mock_object(self.fake_storage, 'update',
mock.Mock(return_value=True))
self.assertRaises(ValueError, data.update, self.entity_id, details)
self.assertFalse(self.fake_storage.update.called)
def test_delete(self):
data = pd.DriverPrivateData(storage=self.fake_storage)
key = "fake_key"
self.mock_object(self.fake_storage, 'get',
mock.Mock(return_value=True))
actual_result = data.delete(self.entity_id, key)
self.assertTrue(actual_result)
self.fake_storage.delete.assert_called_once_with(self.entity_id, key)
def test_get(self):
data = pd.DriverPrivateData(storage=self.fake_storage)
key = "fake_key"
value = "fake_value"
default_value = "def"
self.mock_object(self.fake_storage, 'get',
mock.Mock(return_value=value))
actual_result = data.get(self.entity_id, key, default_value)
self.assertEqual(value, actual_result)
self.fake_storage.get.assert_called_once_with(self.entity_id, key,
default_value)
def test_update(self):
data = pd.DriverPrivateData(storage=self.fake_storage)
details = {"foo": "bar"}
self.mock_object(self.fake_storage, 'update',
mock.Mock(return_value=True))
actual_result = data.update(self.entity_id,
details,
delete_existing=True)
self.assertTrue(actual_result)
self.fake_storage.update.assert_called_once_with(
self.entity_id, details, True)
def test_validate_entity_id_invalid(self, entity_id):
data = pd.DriverPrivateData(storage="fake")
self.assertRaises(ValueError, data._validate_entity_id, entity_id)
def test_custom_storage_driver(self):
private_data = pd.DriverPrivateData(storage=self.fake_storage)
self.assertEqual(private_data._storage, self.fake_storage)
def test_default_storage_driver(self):
private_data = pd.DriverPrivateData(storage=None,
context="fake",
backend_host="fake")
self.assertIsInstance(private_data._storage, pd.SqlStorageDriver)
