karbor/karbor/tests/unit/fake_restore.py

44 lines
1.6 KiB
Python

# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
from oslo_versionedobjects import fields
from karbor import objects
def fake_db_restore(**updates):
db_restore = {
"id": "36ea41b2-c358-48a7-9117-70cb7617410a",
"project_id": "586cc6ce-e286-40bd-b2b5-dd32694d9944",
"provider_id": "2220f8b1-975d-4621-a872-fa9afb43cb6c",
"checkpoint_id": "09edcbdc-d1c2-49c1-a212-122627b20968",
"restore_target": "192.168.1.2/identity/",
"parameters": '{}',
"restore_auth": '{"type": "password", "username": "admin",'
'"password": "test" }',
"status": "SUCCESS"
}
for name, field in objects.Restore.fields.items():
if name in db_restore:
continue
if field.nullable:
db_restore[name] = None
elif field.default != fields.UnspecifiedDefault:
db_restore[name] = field.default
else:
raise Exception('fake_db_restore needs help with %s.' % name)
if updates:
db_restore.update(updates)
return db_restore