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


import requests
import json
import unittest


__author__ = 'James Iter'
__date__ = '2017/3/31'
__contact__ = 'james.iter.cn@gmail.com'
__copyright__ = '(c) 2017 by James Iter.'


class TestOSInitWrite(unittest.TestCase):

    base_url = 'http://127.0.0.1:8008/api'
    os_init_id = 0
    os_init_write_id = 0

    def setUp(self):
        pass

    def tearDown(self):
        pass

    # 创建系统初始化组
    def test_11_create(self):
        payload = {
            "name": "CentOS-Systemd",
            "remark": u"用作红帽 Systemd 系列的系统初始化。初始化操作依据 CentOS 7 来实现。"
        }

        url = TestOSInitWrite.base_url + '/os_init'
        headers = {'content-type': 'application/json'}
        r = requests.post(url, data=json.dumps(payload), headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        TestOSInitWrite.os_init_id = j_r['data']['id']
        self.assertEqual('200', j_r['state']['code'])

    # 添加系统初始化操作
    def test_21_create(self):
        payload = {
            "os_init_id": TestOSInitWrite.os_init_id,
            "path": "/etc/resolv.conf",
            "content": "\n".join([
                "nameserver {DNS1}",
                "nameserver {DNS2}"
            ])
        }

        url = TestOSInitWrite.base_url + '/os_init_write'
        headers = {'content-type': 'application/json'}
        r = requests.post(url, data=json.dumps(payload), headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        self.assertEqual('200', j_r['state']['code'])

    # 添加系统初始化操作
    def test_22_create(self):
        payload = {
            "os_init_id": TestOSInitWrite.os_init_id,
            "path": "/etc/sysconfig/network-scripts/ifcfg-eth0",
            "content": "\n".join([
                "DEVICE=eth0",
                "TYPE=Ethernet",
                "ONBOOT=yes",
                "BOOTPROTO=\"static\"",
                "IPADDR={IP}",
                "NETMASK={NETMASK}",
                "GATEWAY={GATEWAY}",
                "DNS1={DNS1}",
                "DNS2={DNS2}",
                "IPV6INIT=no",
                "NAME=eth0"
            ])
        }

        url = TestOSInitWrite.base_url + '/os_init_write'
        headers = {'content-type': 'application/json'}
        r = requests.post(url, data=json.dumps(payload), headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        self.assertEqual('200', j_r['state']['code'])

    # 添加系统初始化操作
    def test_23_create(self):
        payload = {
            "os_init_id": TestOSInitWrite.os_init_id,
            "path": "/etc/hostname",
            "content": "hostname"
        }

        url = TestOSInitWrite.base_url + '/os_init_write'
        headers = {'content-type': 'application/json'}
        r = requests.post(url, data=json.dumps(payload), headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        TestOSInitWrite.os_init_write_id = j_r['data']['id']
        self.assertEqual('200', j_r['state']['code'])

    # 添加系统初始化操作
    def test_24_create(self):
        payload = {
            "os_init_id": TestOSInitWrite.os_init_id,
            "path": "/etc/hostname",
            "content": "{HOSTNAME}"
        }

        url = TestOSInitWrite.base_url + '/os_init_write'
        headers = {'content-type': 'application/json'}
        r = requests.post(url, data=json.dumps(payload), headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        self.assertEqual('40901', j_r['state']['sub']['code'])

    def test_25_get(self):
        url = TestOSInitWrite.base_url + '/os_init_writes'
        headers = {'content-type': 'application/json'}
        r = requests.get(url, headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        self.assertEqual('200', j_r['state']['code'])

    def test_26_update(self):
        payload = {
            "os_init_id": TestOSInitWrite.os_init_id,
            "path": "/etc/hostname",
            "content": "{HOSTNAME}"
        }

        url = TestOSInitWrite.base_url + '/os_init_write/' + TestOSInitWrite.os_init_write_id.__str__()
        headers = {'content-type': 'application/json'}
        r = requests.patch(url, data=json.dumps(payload), headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        self.assertEqual('200', j_r['state']['code'])

    @unittest.skip('skip delete os init write!')
    def test_27_delete(self):
        url = TestOSInitWrite.base_url + '/os_init_write/' + TestOSInitWrite.os_init_write_id.__str__()
        headers = {'content-type': 'application/json'}
        r = requests.delete(url, headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        self.assertEqual('200', j_r['state']['code'])

    @unittest.skip('skip delete os init!')
    # 删除系统初始化组列表更新结果
    def test_31_delete(self):
        url = TestOSInitWrite.base_url + '/os_init/' + TestOSInitWrite.os_init_id.__str__()
        headers = {'content-type': 'application/json'}
        r = requests.delete(url, headers=headers)
        j_r = json.loads(r.content)
        print json.dumps(j_r, ensure_ascii=False)
        self.assertEqual('200', j_r['state']['code'])


if __name__ == '__main__':
    unittest.main()