#!/usr/bin/env python
#coding: utf-8

from splinter import Browser
from ConfigParser import ConfigParser
import time

cf = ConfigParser()
cf.read(confFile)

b = Browser("chrome")
b.visit(url)


def confcheck():
        conf = self.getConf()
        try:
            conf = {
                "url": self.cf.get("ser", "url"),
                "uname" : self.cf.get("ser", "username"),
                "passwd" : self.cf.get("ser", "password"),
                "tname" : self.cf.get("ser", "project")}

        except Exception as e:
                logging.critical("加载配置文件失败")
                logging.critical(e)
                sys.exit(1)


class base(object):
    def __init__(self):
        self.url = url

    def ClickCSS(t=1, *args):
        for css in args:
            link = b.find_by_css(css)[0]
            link.click()
            time.sleep(t)
            b.screenshot()

    def ClickID(ID, t=1):
        b.click_link_by_id(ID)
        b.screenshot()
        time.sleep(t)

    def ClickDropDown(name):
        css = "tr[data-display=%s] span[class='fa fa-caret-down']" % name
        b.find_by_css(css)[0].click()

    def submit():
        """提交"""
        css = "div[class='modal-footer'] input[type='submit']"
        b.find_by_css(css)[0].click()

    def instance(name, source, flavor, net):
        """创建instance"""

        #跳转到project>instance
        self.projectInstance()

        #点击创建虚拟机
        b.click_link_by_id("instances__action_launch-ng")


        #虚拟机名称
        b.fill("name", name)        

        #选择源
        b.find_by_css("a[href='#']")[3].click()
        b.find_by_css("button[tabindex='0']").click()

        #选择flover
        b.find_by_css("a[href='#']")[4].click()
        b.find_by_css("button[tabindex='0']")[1].click()

        #选择网络
        b.find_by_css("a[href='#']")[5].click()
        b.find_by_css("button[tabindex='0']")[-2].click()

        #点击创建实例
        submit()


    def instanceAction(name, action):
        """操作instance,重启,关闭,热迁移,resize"""

        if action == "resize":
            #点击下拉栏
            css = "tr[data-display=%s] span[class='fa fa-caret-down']" % name

            #选择调整云主机大小
            b.find_by_css("a[class='ajax-modal btn-resize']")[0].click()

            #选择flover
            b.select_by_text("flavor", "m1.tiny")

            #确认
            submit()

            #再次确认
            b.find_by_css("button[value^='instances__confirm']")[0].click()
        elif action == "reboot":
            pass
        elif action == "shutdown":
            pass
        elif action == "liveMigrate":
            self.panelgo(p="admin")

            #点击下拉栏
            ClickDropDown(name)

            #点击云主机热迁移
            b.find_by_css("a[href$='live_migrate']")   
        else:
            alertInfo("暂时不支持此操作 %s" % action)


    def snapshot(name):
        css = "tr[data-display=%s] span[class='fa fa-caret-down']" % name

        #创建快照
        b.fill("name", "snapshot1")

    def snapshotAction(action):
        #删除快照
        if action == "delete":
            b.find_by_css("button[value^='images_delete']")
            b.find_by_css("button[value^='images__delete__']")[0].click()
            b.find_by_css("div[class='modal-footer'] a[href='#']")[1].click()
        else:
            alertInfo("WTF")

    def net(name, subname, cidr):
        """创建网络"""

        #进入项目>网络
        self.panelgo(p="project", location="networks")

        #点击创建网络
        b.click_link_by_id("networks__action_create")

        #编辑网络tab
        b.fill("net_name","net1")

        #编辑子网
        b.fill("subnet_name", "net1-sub")
        b.fill("cidr", "192.168.0.1/24")

        #点击网络详情
        css = "div[class='modal-body'] a[href='#create_network__createsubnetdetailaction']"
        Click(1,css)

        #点击创建网络
        css = "div[class='modal-footer'] button[type='submit']"
        Click(1,css)


    def netExt(name, subname, cidr):
        """创建网络"""

    def netAction(name, action):
        """操作网络,删除"""
        pass

    def route(name):
        """新建路由"""
        pass

    def routeAction(name, action):
        pass

    def secgroup(name):
        pass

    def image(name):
        pass

    def imageAction(name, action):
        pass

    def terminal(stdin):
        pass


class poctest(base):
    def __init__(self):
        self.url = url

    def login(self, uname="admin", passwd="admin"):
        """登陆"""
        b.fill("username", "admin")
        b.fill("password", "admin")
        b.click_link_by_id("loginBtn")

    def panelgo(self, p="project", location="instances", v=9):
        """默认进入侧边栏project>instance"""
        visitURL = "%s/horizon/%s/%s" % (self.url, p, location)
        b.visit(visitURL)

    def alertInfo(info):
        """提示测试内容"""
        b.execute_script("alert(%s)" % info)
        a = b.get_alert()
        sleep(2)
        a.accept()

    def test1(self, restart=True):
        """创建,关闭,重启虚拟机"""

        self.projectInstance()
        
        #点击创建虚拟机
        b.click_link_by_id("instances__action_launch-ng")
        
        #虚拟机名称
        b.fill("name","vm1")        

        #选择源
        b.find_by_css("a[href='#']")[3].click()
        b.find_by_css("button[tabindex='0']").click()

        #选择flover
        b.find_by_css("a[href='#']")[4].click()
        b.find_by_css("button[tabindex='0']")[1].click()
        
        #选择网络
        b.find_by_css("a[href='#']")[5].click()
        b.find_by_css("button[tabindex='0']")[-2].click()

        #点击创建实例
        b.find_by_css("span[ng-bind='::viewModel.btnText.finish']")[0].click()

        #关闭实例

        #重启实例

    def test2(self):
        """创建,删除快照"""
        b.find_by_css("a[id$='snapshot']")[0].click()

        #创建快照
        b.fill("name","snapshot1")
        b.find_by_css("input[type='submit']")[0].click()

        #删除快照
        b.find_by_css("button[value^='images_delete']")
        b.find_by_css("button[value^='images__delete__']")[0].click()
        b.find_by_css("div[class='modal-footer'] a[href='#']")[1].click()        


    def test3(self):
        """调整云主机大小"""

        #选择调整云主机大小
        b.find_by_css("a[class='ajax-modal btn-resize']")[0].click()
        
        #选择flover
        b.select_by_text("flavor","m1.tiny")
        
        #确认
        b.find_by_css("div[class='modal-footer'] input[type='submit']")[0].click()

        #再次确认
        b.find_by_css("button[value^='instances__confirm']")[0].click()


    def test4():
        """虚拟机热迁移"""
        self.projectInstance(p="admin")
        
        #点击下拉栏
        b.find_by_css("tr[class='ajax-update status_up'] span[class='fa fa-caret-down']")[0].click()

        #点击云主机热迁移
        b.find_by_css("a[href$='live_migrate']")        

           
    def test5():
        """创建用户内部私有网络, net1, net2"""
       
        #进入项目>网络
        self.panelgo(p="project", location="networks")

        #点击创建网络
        b.click_link_by_id("networks__action_create")

        #编辑网络tab
        b.fill("net_name","net1")

        #编辑子网
        b.fill("subnet_name", "net1-sub")
        b.fill("cidr", "192.168.0.1/24")

        #点击网络详情
        css = "div[class='modal-body'] a[href='#create_network__createsubnetdetailaction']"
        Click(1,css)

        #点击创建网络
        css = "div[class='modal-footer'] button[type='submit']"
        Click(1,css)
            
        
    def test6():
        """创建用户外部EXT网络"""
        js = "var chk = document.getElementById('id_external');chk.checked = true"

        #跳转admin>networks 
        b.visit()        

        #点击创建网络
        ID = "networks__action_create"
       
        #填入信息
        b.fill("name","pub")
        b.select_by_text("tenant_id","admin")
        b.select("id_network_type","flat")
         

        #选择外部网络
        b.execute_script(js)
       
        #提交
        submit()

           
    def test7():
        """创建路由器"""
        #跳转project>route
        b.visit()

        #点击创建路由
        ID = "routers__action_create"
        b.click_link_by_id()
        
        #填入信息
        b.fill("name", "route1")

        #选择外部网络
        b.select_by_text("external_network")

        #提交
        submit()

        #将新建的子网net1,net2加入此路由器
        #点击刚创建的路由器
        b.find_by_css("tr[data-display='test'] a")[0].click()

        #跳转到接口
        b.click_link_by_href("?tab=router_details__interfaces")
        
        #增加接口
        ID = "interfaces__action_create"

        #填入信息
        css = ""
        ID = [i.value for i in b.find_by_css("select option") if i.text.startswith("net1")][0]
        
        #提交
        submit()
        
           
    def test8():
        """分配及绑定floating ip"""
        #跳转project>instance
        b.visit()

        #点击绑定浮动ip
        

        #获取

        #提交

        #ping

    def test9():
        """不同租户下网络隔离"""
        #跳转

        #登陆terminal

        #执行命令

        #xxx
           
    def test10():
        """安全组访问控制策略"""
       #跳转project>secgroup
 
       #点击新建secgroup

       #配置

       #提交
           
    def test11():
        """创建,挂载云硬盘"""
        #跳转project>volume
        b.visit("http://172.16.0.3/horizon/project/volumes/")

        #点击创建云硬盘
        b.click_link_by_id("volumes__action_create")

        #配置

        #提交
        self.submit()

        #挂载

    def test12():
        """扩展云硬盘"""
        #跳转
        b.visit("http://172.16.0.3/horizon/project/volumes/")

        #点击下拉栏
        css = "tr[data-display='vol11'] span[class='fa fa-caret-down']"

        #点击扩展
        css = "a[id$='action_extend']"
        
        #配置
        b.fill("new_size","2")

        #提交
        submit()
           
    def test13():
        """上传,下载,删除镜像"""
        #跳转
        b.visit("http://172.16.0.3/horizon/project/images/")        

        #点击新建
        ID = "images__action_create"        

        #配置
        b.fill("name","img1")
        alertInfo("请在10秒内选择相应的镜像文件")
        b.select("disk_format","qcow2")       
        while something:
            submit()

        #提交

        #删除
        css = "tr[data-display='img1'] span[class='fa fa-caret-down']"

        css = "tr[data-display='img1'] button[id$='__action_delete']"
        b.find_by_css("div[class='modal-footer'] a")[-1].click()

    def test14():
        """日志数据收集"""
        pass

    def test15():
        """云平台告警"""
        pass

    def test16():
        """openstack服务监控"""
        pass