fuel-ccp/fuel_ccp/tests/test_templates.py

import base64
import six

from fuel_ccp import templates
from fuel_ccp.tests import base


class TestDeploy(base.TestCase):

    def test_serialize_daemon_container_spec(self):
        container = {
            "name": "name_foo",
            "image": "image_foo",
            "command": "command_foo",
            "cm_version": 1,
            "env": [{
                "name": "env_foo",
                "valueFrom": {
                    "valueField": {
                        "valuePath": "metadata.name"
                    }
                }
            }],
            "probes": {
                "liveness": {
                    "command": "true",
                    "type": "exec"
                }
            },
            "daemon": {
                "command": "run.sh"
            }
        }
        container_spec = templates.serialize_daemon_container_spec(container)
        expected = {
            "name": "name_foo",
            "image": "ccp/image_foo:latest",
            "imagePullPolicy": None,
            "command": [
                "dumb-init",
                "/usr/bin/python",
                "/opt/ccp_start_script/bin/start_script.py",
                "provision",
                "name_foo"
            ],
            "volumeMounts": [
                {'mountPath': '/etc/ccp/globals', 'name': 'globals'},
                {'mountPath': '/etc/ccp/global-secrets',
                 'name': 'global-secrets'},
                {'mountPath': '/etc/ccp/role', 'name': 'role'},
                {'mountPath': '/etc/ccp/meta', 'name': 'meta'},
                {'mountPath': '/opt/ccp_start_script/bin',
                 'name': 'start-script'},
                {'mountPath': '/etc/ccp/exports', 'name': 'exports'},
                {'mountPath': '/etc/ccp/files', 'name': 'files'},
                {'mountPath': '/etc/ccp/nodes-config', 'name': 'nodes-config'},
                {'mountPath': '/etc/ccp/service-config',
                 'name': 'service-config'}
            ],
            "readinessProbe": {
                "exec": {
                    "command": [
                        "/usr/bin/python",
                        "/opt/ccp_start_script/bin/start_script.py",
                        "status",
                        "name_foo"
                    ]
                },
                "timeoutSeconds": 1
            },
            "livenessProbe": {
                "exec": {
                    "command": ['true']
                },
                "timeoutSeconds": 1,
                "initialDelaySeconds": 10
            },
            "env": [
                {
                    "name": "CCP_NODE_NAME",
                    'valueFrom': {
                        'fieldRef': {
                            'fieldPath': 'spec.nodeName'
                        }
                    }
                },
                {
                    "name": "CCP_POD_NAME",
                    "valueFrom": {
                        "fieldRef": {
                            "fieldPath": "metadata.name"
                        }
                    }
                },
                {
                    'name': 'CPU_LIMIT',
                    'valueFrom': {
                        'resourceFieldRef': {
                            'resource': 'limits.cpu'
                        }
                    }
                },
                {
                    'name': 'MEMORY_LIMIT',
                    'valueFrom': {
                        'resourceFieldRef': {
                            'resource': 'limits.memory'
                        }
                    }
                },
                {
                    "name": "env_foo",
                    "valueFrom": {
                        "valueField": {
                            "valuePath": "metadata.name"
                        }
                    }
                },
                {
                    "name": "CM_VERSION",
                    "value": 1
                }
            ],
            "securityContext": {
                "privileged": False
            }
        }
        self.assertDictEqual(expected, container_spec)

    def test_serialize_liveness_probe_exec(self):
        probe_definition = {"type": "exec", "command": "true"}
        expected = {
            "livenessProbe": {
                "exec": {
                    "command": ["true"]
                },
                "timeoutSeconds": 1,
                "initialDelaySeconds": 10
            }
        }
        probe_spec = templates.serialize_liveness_probe(probe_definition)
        self.assertDictEqual(expected, probe_spec)

    def test_serialize_liveness_probe_http(self):
        probe_definition = {
            "type": "httpGet",
            "path": "_status",
            "port": 8080,
            "initialDelay": 7
        }
        expected = {
            "livenessProbe": {
                "httpGet": {
                    "path": "_status",
                    "port": 8080,
                    "scheme": "HTTP"
                },
                "timeoutSeconds": 1,
                "initialDelaySeconds": 7
            }
        }
        probe_spec = templates.serialize_liveness_probe(probe_definition)
        self.assertDictEqual(expected, probe_spec)

    def test_serialize_secret(self):
        name = "the-most-secret"
        data = {"a": "value1", "b": " ./?+{}()[]|\\\'\""}
        expected = {
            "apiVersion": "v1",
            "data": data,
            "kind": "Secret",
            "metadata": {
                "name": name
            },
            "type": "Opaque"
        }
        serialized = templates.serialize_secret(name, data=data)
        serialized["data"] = dict(
            [(key, base64.b64decode(value).decode())
                for (key, value) in six.iteritems(serialized["data"])]
        )
        self.assertDictEqual(expected, serialized)