# Copyright 2018 Red Hat, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import json


testinfra_hosts = ['gitea-lb02.opendev.org']


def test_gitea_listening(host):
    gitea_https = host.socket("tcp://0.0.0.0:443")
    assert gitea_https.is_listening
    gitea_http = host.socket("tcp://0.0.0.0:80")
    assert gitea_http.is_listening

def test_haproxy_statsd_running(host):
    cmd = host.run("docker inspect haproxy-docker_haproxy-statsd_1")
    out = json.loads(cmd.stdout)
    assert out[0]["State"]["Status"] == "running"
    assert out[0]["RestartCount"] == 0

def test_haproxy_gitea_connection(host):
    cmd = host.run('curl --resolve opendev.org:443:127.0.0.1 '
                   'https://opendev.org')
    assert 'OpenDev: Free Software Needs Free Tools' in cmd.stdout

def test_more_than_haproxy_maxconn_conns(host):
    '''Test we can handle more than haproxy maxconn limit for connections'''
    # Perform more than 4000 (maxconn limit) requests.
    cmd = host.run('bash -c "seq 1 4096 | xargs -IUNUSED -n 1 -P 0 '
                   'curl --resolve opendev.org:443:127.0.0.1 '
                   'https://opendev.org > /dev/null 2>&1"')
    # Now make a request and ensure we get expected content back
    cmd = host.run('curl --resolve opendev.org:443:127.0.0.1 '
                   'https://opendev.org')
    assert 'OpenDev: Free Software Needs Free Tools' in cmd.stdout

def test_haproxy_stats(host):
    cmd = host.run('echo "show servers state" | socat /var/haproxy/run/stats stdio | '
                   'tail +3 | awk \'{print $2,$4,$6}\'')

    assert 'balance_git_http gitea99.opendev.org 2' in cmd.stdout
    assert 'balance_git_https gitea99.opendev.org 2' in cmd.stdout