#    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