def main():
arg_parser = common.create_arg_parser()
grpc_server, file_server = common.services(arg_parser)
service = common.create_service(grpc_server)
p = MyPrompt(service, file_server)
p.cmdloop()
#!/usr/bin/env python3
import common
import witness.api.witness_pb2 as witness_pb2
import witness.api.witness_pb2_grpc as witness_pb2_grpc
DEFAULT_TIMEOUT = 5 # seconds
def open_webcam(service_stub, device=1):
open_request = witness_pb2.OpenWebcamRequest()
open_request.webcam_id = device
reply = service_stub.OpenWebcam(open_request, DEFAULT_TIMEOUT)
print(reply)
if __name__ == '__main__':
arg_parser = common.create_arg_parser()
grpc_server, _ = common.services(arg_parser)
service_stub = common.create_service(grpc_server)
open_webcam(service_stub)
def __call__(self, parser, args, values , option_string=None):
setattr(args, self.dest, values)
create_service(path=os.path.join(args.path, args.name), name=args.name)
def k8s_flannel__k8s_service(self):
# 创建flannel服务支持:环境变量，配置文件，local volume，日志文件，部署到指定机器
if self.haproxy:
test_flag = True
ret1 = get_haproxy(self.haproxy)
self.assert_successful(ret1)
k8sFlannelData = data.ServiceData(
self.k8s_flannel_service,
self.namespace,
settings.SERVICE_CLAAS_REGION[0],
lb_type='haproxy',
alb_name=self.haproxy,
lb_id=ret1['haproxy_id'],
node_tag=self.node_tag.split(":")[1],
mipn_enabled=True)
#创建服务支持 环境变量 local volume，配置文件，指定日志文件，部署在指定的机器上 block后面操作
ret2 = create_service(self.k8s_flannel_service,
k8sFlannelData.k8s_flannel_service(),
settings.SERVICE_CLAAS_REGION[0])
self.assert_successful(ret2)
#验证服务是否可以访问 失败可以继续后面测试
ret3 = access_service(self.k8s_flannel_service, self.haproxy)
if not ret3['success']:
test_flag = False
#验证环境变量是否添加进去 失败可以继续后面测试
ret4 = exec_feature(self.k8s_flannel_service,
self.namespace,
command="env",
commands_string="k8s_key=k8s_value")
if not ret4['success']:
test_flag = False
#验证配置文件是否添加进去 失败可以继续后面测试
ret5 = exec_feature(self.k8s_flannel_service,
self.namespace,
command="'cat /home/abc'",
commands_string="config")
if not ret5['success']:
test_flag = False
#判断存储卷类型 不阻塞后面的测试
ret6 = verify_volumes(self.k8s_flannel_service, "host_path")
if not ret6['success']:
test_flag = False
#判断是否有Metrics 不阻塞后面的测试
ret7 = get_metrics(self.k8s_flannel_service)
if not ret7['success']:
test_flag = False
# 停止服务 如果失败block后面操作
ret8 = stop_app(self.k8s_flannel_service)
self.assert_successful(ret8)
#启动服务 如果失败block 后面操作
ret9 = start_app(self.k8s_flannel_service, num=1)
self.assert_successful(ret9)
#scale up 服务，更新服务的数量和size，失败block后面操作
ret10 = update_service(self.k8s_flannel_service, num=2, size="XS")
self.assert_successful(ret10)
#check 所有的容器都部署在指定的机器上 不阻塞后面的测试
ret11 = check_node(self.k8s_flannel_service, self.node_tag)
if not ret11['success']:
test_flag = False
# scale down 服务更新服务的数量和size 失败block后面操作
ret12 = update_service(self.k8s_flannel_service, num=1, size="XXS")
self.assert_successful(ret12)
# check 所有的容器都部署在指定的机器上 不阻塞后面的测试
ret13 = check_node(self.k8s_flannel_service, self.node_tag)
if not ret13['success']:
test_flag = False
# check 服务的日志
ret14 = get_logs(self.k8s_flannel_service)
if not ret14['success']:
test_flag = False
#check 日志文件
ret15 = get_logfile(self.k8s_flannel_service)
if not ret15['success']:
test_flag = False
#删除服务
ret16 = delete_app(self.k8s_flannel_service,
settings.SERVICE_CLAAS_REGION[0])
if not ret16['success']:
test_flag = False
result = {
'success': test_flag,
"get haproxy id ": ret1,
"create k8s haproxy service": ret2,
"access sercie ": ret3,
"get service env": ret4,
"get servie config": ret5,
"get service volume type": ret6,
"get service metrics": ret7,
"stop service": ret8,
"start service": ret9,
"scale up service": ret10,
"check instance in node": ret11,
"scale down service": ret12,
"check instance in node": ret13,
"get service log": ret14,
"get service logfile": ret15,
"delete service": ret16
}
self.assert_successful(result)
return result
def inner_elb_service_test__k8s_service(self):
test_flag = True
# 创建一个内网的ELB 失败返回
ret1 = create_alb(self.k8s_internal_elb,
create_type='manual',
type='elb',
alb_region=settings.SERVICE_CLAAS_REGION[0],
address_type="internal")
self.assert_successful(ret1)
elb_data = data.ServiceData(self.k8s_innerelb_service,
settings.CLAAS_NAMESPACE,
settings.SERVICE_CLAAS_REGION[0],
lb_type='elb',
alb_name=self.k8s_internal_elb,
lb_id=ret1['id'])
# 创建一个使用ELB网络模式的服务 失败返回
ret2 = create_service(self.k8s_innerelb_service,
elb_data.k8s_innerelb_service(),
settings.SERVICE_CLAAS_REGION[0])
self.assert_successful(ret2)
# 验证服务是否可以访问 失败继续测试
ret3 = access_service(self.k8s_innerelb_service, self.k8s_internal_elb)
if not ret3['success']:
test_flag = False
# 验证环境变量是否添加进去 失败可以继续后面测试
ret4 = exec_feature(self.k8s_innerelb_service,
self.namespace,
command="'/bin/ls /'")
if not ret4['success']:
test_flag = False
# 判断是否有Metrics 不阻塞后面的测试
ret5 = get_metrics(self.k8s_innerelb_service)
if not ret5['success']:
test_flag = False
# 停止服务 如果失败block后面操作
ret6 = stop_app(self.k8s_innerelb_service)
self.assert_successful(ret6)
# 启动服务 如果失败block 后面操作
ret7 = start_app(self.k8s_innerelb_service, num=1)
self.assert_successful(ret7)
# scale up 服务，更新服务的数量和size，失败block后面操作
ret8 = update_service(self.k8s_innerelb_service, num=2, size="XS")
self.assert_successful(ret8)
# scale down 服务更新服务的数量和size 失败block后面操作
ret9 = update_service(self.k8s_innerelb_service, num=1, size="XXS")
self.assert_successful(ret9)
# check 服务的日志
# ret10 = get_logs(self.k8s_innerelb_service)
# if not ret10['success']:
# test_flag = False
ret11 = delete_alb(self.k8s_innerelb_service)
self.assert_successful(ret11)
ret12 = delete_app(self.k8s_innerelb_service)
self.assert_successful(ret12)
result = {
'success': test_flag,
"create elb": ret1,
"create elb service": ret2,
"access service": ret3,
"service exec ": ret4,
"get metrics": ret5,
"stop service": ret6,
"start service": ret7,
"scale up service": ret8,
"scale down service": ret9,
# "get logs": ret10,
"delete elb": ret11,
"delete service": ret12
}
self.assert_successful(result)
return result
def k8s_healthycheck__k8s_service(self):
if self.haproxy:
# 设置flag
test_flag = True
# 创建部署服务需要的ebs存储卷，创建失败直接返回
ret1 = create_volume(self.ebs_name, 'ebs', 'gp2')
self.assert_successful(ret1)
# 获取当前所在集群的Haproxy信息，获取失败直接返回
ret2 = get_haproxy(self.haproxy)
self.assert_successful(ret2)
# 获取创建服务需要的数据
k8sFlannelData = data.ServiceData(
self.k8s_healthycheck_service,
self.namespace,
settings.SERVICE_CLAAS_REGION[0],
lb_type='haproxy',
alb_name=self.haproxy,
lb_id=ret2['haproxy_id'],
volume_id=ret1['volume_id'],
volume_name=self.ebs_name)
# 创建挂载ebs存储卷和添加健康检查的服务
ret3 = create_service(
self.k8s_healthycheck_service,
k8sFlannelData.k8s_healthycheck_service(),
settings.SERVICE_CLAAS_REGION[0])
if not ret3["success"]:
delete_volume(ret1['volume_id'], self.ebs_name)
self.assert_successful(ret3)
# 访问服务，及时失败也可以继续执行测试
ret4 = access_service(self.k8s_healthycheck_service,
self.haproxy)
if not ret4['success']:
test_flag = False
# 验证volume id是否一致
ret5 = verify_volumes(self.k8s_healthycheck_service,
ret1['volume_id'])
if not ret5['success']:
test_flag = False
ret6 = delete_app(self.k8s_healthycheck_service,
settings.SERVICE_CLAAS_REGION[0])
if not ret6['success']:
test_flag = False
#sleep 30s 保证volume变成可以删除的状态
sleep(30)
# 删除volume
ret7 = delete_volume(ret1['volume_id'], self.ebs_name)
result = {
'success': test_flag,
"create ebs volume": ret1,
"get haproxy id ": ret2,
"create service": ret3,
"access service": ret4,
"check volume": ret5,
"delete service": ret6,
"delete volume": ret7
}
self.assert_successful(result)
return result
def k8s_host__k8s_service(self):
if self.haproxy:
#设置flag
test_flag = True
#创建部署服务需要的环境变量文件 创建失败就直接返回
ret1 = create_envs(self.k8s_envfile)
self.assert_successful(ret1)
#创建部署服务需要的glustfs存储卷，创建失败直接返回
ret2 = create_volume(self.gluster_name, 'glusterfs', None)
self.assert_successful(ret2)
#获取当前所在集群的Haproxy信息，获取失败直接返回
ret3 = get_haproxy(self.haproxy)
self.assert_successful(ret3)
#获取创建服务需要的数据
k8sFlannelData = data.ServiceData(self.k8s_host_service,
self.namespace,
settings.SERVICE_CLAAS_REGION[0],
lb_type='haproxy',
alb_name=self.haproxy,
lb_id=ret3['haproxy_id'],
volume_id=ret2['volume_id'],
volume_name=self.gluster_name,
envfile=self.k8s_envfile)
#创建Host模式挂载环境变量文件和glustfs存储卷的服务
ret4 = create_service(self.k8s_host_service,
k8sFlannelData.k8s_host_service(),
settings.SERVICE_CLAAS_REGION[0])
if not ret4["success"]:
delete_volume(ret2['volume_id'], self.gluster_name)
self.assert_successful(ret4)
#访问服务，及时失败也可以继续执行测试
ret5 = access_service(self.k8s_host_service, self.haproxy)
if not ret5['success']:
test_flag = False
# 验证环境变量是否添加进去 失败可以继续后面测试
ret6 = exec_feature(self.k8s_host_service,
self.namespace,
command="env",
commands_string="key=value")
if not ret6['success']:
test_flag = False
# 验证volume id是否一致
ret7 = verify_volumes(self.k8s_host_service, ret2['volume_id'])
if not ret7['success']:
test_flag = False
# 判断是否有Metrics 不阻塞后面的测试
ret8 = get_metrics(self.k8s_host_service)
if not ret8['success']:
test_flag = False
# 停止服务 如果失败block后面操作
ret9 = stop_app(self.k8s_host_service)
self.assert_successful(ret9)
# 启动服务 如果失败block 后面操作
ret10 = start_app(self.k8s_host_service, num=1)
self.assert_successful(ret10)
# scale up 服务，更新服务的数量和size，失败block后面操作
ret11 = update_service(self.k8s_host_service, num=2, size="XS")
self.assert_successful(ret11)
# scale down 服务更新服务的数量和size 失败block后面操作
ret12 = update_service(self.k8s_host_service, num=1, size="XXS")
self.assert_successful(ret12)
# check 服务的日志
ret13 = get_logs(self.k8s_host_service)
if not ret13['success']:
test_flag = False
ret14 = delete_app(self.k8s_host_service,
settings.SERVICE_CLAAS_REGION[0])
if not ret14['success']:
test_flag = False
#删除volume
sleep(30)
delete_volume(ret2['volume_id'], self.gluster_name)
result = {
'success': test_flag,
"create envfile ": ret1,
"create glustfs volume": ret2,
"get haproxy id ": ret3,
"create service": ret4,
"access service": ret5,
"check envfile": ret6,
"check volume": ret7,
"check metrics": ret8,
"stop service": ret9,
"start service": ret10,
"scale up": ret11,
"scale down": ret12,
"check log": ret13,
"delete service": ret14
}
self.assert_successful(result)
return result
def alarm_state__alarms(self):
''' test alarm state: INSUFFICIENT_DATA, OK, ALARM, OUT_OF_OPERATION '''
# step1: create one service
# step2: create one alarm to monitor the service, set "metric_name" & "threshold"... to get 'OK' state of alarm
# step3: update service to get 'ALARM' state of alarm
# step4: stop service to get 'INSUFFICIENT_DATA' state of alarm
# step5: delete service to get 'OUT_OF_OPERATION' state of alarm
ret = create_service(self.create_service_name,
self.serviceData.ha_heathy_check_service(), region=self.app_region)
self.assert_successful(ret)
ret = get_service_info(self.create_service_name)
self.assert_successful(ret)
content = json.loads(ret["total"].content)
payload = [{
"alarm_actions": {"notifications": [settings.SERVICE_CLAAS_REGION[0]]},
"resource_name": self.create_service_name,
"metric_name": "custom_health_check_healthy_instances",
"evaluation_periods": 1,
"description": "",
"period": 60,
"actions_enabled": True,
"dimensions": [],
"statistic": "Average",
"threshold": 2,
"threshold_display": 2,
"space_name": settings.SPACE_NAME,
"comparison_operator": "GreaterThanOrEqualToThreshold",
"resource_type": "SERVICE",
"resource_uuid": content["uuid"],
"name": self.create_alarm_state_name,
}]
print('payload: {}'.format(payload))
ret = create_alarms(self.create_alarm_state_name, payload)
self.assert_successful(ret)
ret = get_alarms(self.create_alarm_state_name, self.create_service_name)
self.assert_successful(ret)
ret = query_alarm(self.create_alarm_state_name)
self.assert_successful(ret)
content = json.loads(ret['total'].content)
if content["state_value"] == 'OK':
self.assert_successful({"success": True, "total": '{} state is OK'.format(self.create_alarm_state_name)})
else:
self.assert_successful({"success": False, "total": '{} state is {}, should be OK'
.format(self.create_alarm_state_name, content["state_value"])})
ret = update_service(self.create_service_name, 2, 'XXS')
self.assert_successful(ret)
sleep(300)
ret = query_alarm(self.create_alarm_state_name)
self.assert_successful(ret)
content = json.loads(ret['total'].content)
if content["state_value"] == 'ALARM':
self.assert_successful({"success": True, "total": '{} state is ALARM'.format(self.create_alarm_state_name)})
else:
self.assert_successful({"success": False, "total": '{} state is {}, should be ALARM'
.format(self.create_alarm_state_name, content["state_value"])})
ret = stop_app(self.create_service_name)
self.assert_successful(ret)
sleep(300)
ret = query_alarm(self.create_alarm_state_name)
self.assert_successful(ret)
content = json.loads(ret['total'].content)
if content["state_value"] == 'INSUFFICIENT_DATA':
self.assert_successful({"success": True, "total": '{} state is INSUFFICIENT_DATA'.format(self.create_alarm_state_name)})
else:
self.assert_successful({"success": False, "total": '{} state is {}, should be INSUFFICIENT_DATA'
.format(self.create_alarm_state_name, content["state_value"])})
ret = delete_app(self.create_service_name, self.app_region)
self.assert_successful(ret)
sleep(120)
ret = query_alarm(self.create_alarm_state_name)
self.assert_successful(ret)
content = json.loads(ret['total'].content)
if content["state_value"] == 'OUT_OF_OPERATION':
self.assert_successful({"success": True, "total": '{} state is OUT_OF_OPERATION'.format(self.create_alarm_state_name)})
else:
self.assert_successful({"success": False, "total": '{} state is {}, should be OUT_OF_OPERATION'
.format(self.create_alarm_state_name, content["state_value"])})
ret = delete_alarms(self.create_alarm_state_name)
self.assert_successful(ret)
