|
|
|
|
#coding:utf-8
|
|
|
|
|
import django
|
|
|
|
|
import os
|
|
|
|
|
import sys
|
|
|
|
|
|
|
|
|
|
sys.path.append('../')
|
|
|
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = 'jumpserver.settings'
|
|
|
|
|
django.setup()
|
|
|
|
|
|
|
|
|
|
from juser.views import db_add_user, md5_crypt, CRYPTOR
|
|
|
|
|
from jasset.models import Asset, IDC, BisGroup
|
|
|
|
|
from juser.models import UserGroup
|
|
|
|
|
from jasset.views import jasset_group_add
|
|
|
|
|
from jperm.models import CmdGroup
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_add_idc():
|
|
|
|
|
for i in range(1, 20):
|
|
|
|
|
name = 'IDC' + str(i)
|
|
|
|
|
IDC.objects.create(name=name, comment='')
|
|
|
|
|
print 'Add: %s' % name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_add_asset_group():
|
|
|
|
|
BisGroup.objects.create(name='ALL', type='A', comment='ALL')
|
|
|
|
|
for i in range(1, 20):
|
|
|
|
|
name = 'AssetGroup' + str(i)
|
|
|
|
|
BisGroup.objects.create(name=name, type='A', comment=name)
|
|
|
|
|
print 'Add: %s' % name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_add_user_group():
|
|
|
|
|
for i in range(1, 20):
|
|
|
|
|
name = 'DepartGroup' + str(i)
|
|
|
|
|
UserGroup.objects.create(name=name, type='M', comment=name)
|
|
|
|
|
print 'Add: %s' % name
|
|
|
|
|
|
|
|
|
|
for i in range(1, 20):
|
|
|
|
|
name = 'UserGroup' + str(i)
|
|
|
|
|
UserGroup.objects.create(name=name, type='A', comment=name)
|
|
|
|
|
print 'Add: %s' % name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_add_cmd_group():
|
|
|
|
|
for i in range(1, 20):
|
|
|
|
|
name = 'CMD' + str(i)
|
|
|
|
|
cmd = '/sbin/ping%s, /sbin/ifconfig/' % str(i)
|
|
|
|
|
CmdGroup.objects.create(name=name, cmd=cmd, comment=name)
|
|
|
|
|
print 'Add: %s' % name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_add_user():
|
|
|
|
|
for i in range(1, 500):
|
|
|
|
|
username = "test" + str(i)
|
|
|
|
|
db_add_user(username=username,
|
|
|
|
|
password=md5_crypt(username),
|
|
|
|
|
name=username, email='%s@jumpserver.org' % username,
|
|
|
|
|
groups=[1,3], role='CU',
|
|
|
|
|
ssh_pwd=CRYPTOR.encrypt(username),
|
|
|
|
|
ssh_key_pwd=CRYPTOR.encrypt(username),
|
|
|
|
|
ldap_pwd=CRYPTOR.encrypt(username),
|
|
|
|
|
is_active=True,
|
|
|
|
|
date_joined=0)
|
|
|
|
|
print "Add: %s" % username
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_add_asset():
|
|
|
|
|
test_idc = IDC.objects.get(id=1)
|
|
|
|
|
for i in range(1, 500):
|
|
|
|
|
ip = '192.168.1.' + str(i)
|
|
|
|
|
Asset.objects.create(ip=ip, port=22, login_type='L', idc=test_idc, is_active=True, comment='test')
|
|
|
|
|
print "Add: %s" % ip
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
test_add_idc()
|
|
|
|
|
test_add_asset_group()
|
|
|
|
|
test_add_user_group()
|
|
|
|
|
test_add_cmd_group()
|
|
|
|
|
test_add_asset()
|
|
|
|
|
test_add_user()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
