s010_registration_logic.py

Go to the documentation of this file.

# -*- coding: utf-8 -*-
# @COPYRIGHT_begin
#
# Copyright [2010-2014] Institute of Nuclear Physics PAN, Krakow, Poland
#
# 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.
#
# @COPYRIGHT_end

from datetime import datetime
import hashlib
from random import random
from predefined.macros import *
from predefined.actions import *
from predefined.users import *

actions = [list_clusters]

# Register users
for u_new in test_users:
    actions.append({'fname': 'user.guest.register', 'args': [{'login': u_new['l'],
                                                        'password': hashlib.sha1(u_new['p']).hexdigest(),
                                                        'act_key': u_new['act_key'],
                                                        'organization': u_new['organization'],
                                                        'email': u_new['email'],
                                                        'first': u_new['first'],
                                                        'last': u_new['last'],
                                                        }]})

    # List vm images. Inactive user shouldn't be allowed to list private images
    actions.append({'u': u_new,
                    'fname': 'image.user.list',
                    'args':[{'access': 0, 'type': 2}], 'estatus': 'not_activated'})

    # Activate user
    actions.append({'fname': 'user.guest.activate', 'args': ['fake_key'], 'estatus': 'user_get'})
    actions.append({'fname': 'user.guest.activate', 'args': [u_new['act_key']], 'estatus': 'ok'})

    # Now user should be able to list private images
    actions.append({'u': u_new,
                    'fname': 'image.user.list',
                    'args': [{'access': 1, 'type': 2}],
                    'store_as': 'public_images'})

    # Check user's login
    #TODO: change to check_password
    #actions.append({'fname': 'user.guest.get_by_login', 'args': [u_new['l']], 'store_as': 'get_by_login'})

    # Check user's password hash
    actions.append({'fname': 'user.guest.check_password',
                    'args': [u_new['l'], hashlib.sha1(u_new['p']).hexdigest()]})

    # Get message list
    actions.append({'u': u_new, 'fname': 'message.user.list', 'args': []})

    #TODO: Change password
    #TODO: recovery password
    #TODO: Show quota
    #TODO: Check admin_cm password cmadmin.admin_cm.check_password
    #TODO: Check cm_user password

# Try to list VMs with various logins
actions.append({'u': u_new, 'fname': 'vm.user.list', 'args': [], 'estatus': 'ok'})
actions.append({'u': u_nexists, 'fname': 'vm.user.list', 'args': [], 'estatus': 'user_get'})
actions.append({'u': u_pass, 'fname': 'vm.user.list', 'args': [], 'estatus': 'user_get'})
actions.append({'u': u_cm, 'fname': 'vm.user.list', 'args': [], 'estatus': 'cluster_get'})
actions.append({'u': u_cm_pass, 'fname': 'vm.user.list', 'args': [], 'estatus': 'user_get'})

# Store user list
actions.append(list_users)

# Set user_clm_admin as CLM administrator
actions.append({'u': u_admin,
                'fname': 'user.admin_clm.set_admin',
                'args': [MACRO_GET_USER_ID(u_admin_clm['l']),
                         True]})
actions.append({'u': u_admin,
                'fname': 'user.admin_clm.set_admin',
                'args': [MACRO_GET_USER_ID(u_admin_clm['l']),
                         False]})
actions.append({'u': u_admin,
                'fname': 'user.admin_clm.set_admin',
                'args': [MACRO_GET_USER_ID(u_admin_clm['l']),
                         True]})

# Set user admin_cm as CM administrator
actions.append({'u': u_admin,
                'as_cmadmin': True,
                'fname': 'cmadmin.admin_cm.add',
                'args': [MACRO_GET_USER_ID(u_admin_cm['l']),
                         {'password': hashlib.sha1(u_admin_cm['cm_p']).hexdigest()}]
                })