__init__.py 4.67 KB
Newer Older
André Anjos's avatar
André Anjos committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :

###############################################################################
#                                                                             #
# Copyright (c) 2016 Idiap Research Institute, http://www.idiap.ch/           #
# Contact: beat.support@idiap.ch                                              #
#                                                                             #
# This file is part of the beat.core module of the BEAT platform.             #
#                                                                             #
# Commercial License Usage                                                    #
# Licensees holding valid commercial BEAT licenses may use this file in       #
# accordance with the terms contained in a written agreement between you      #
# and Idiap. For further information contact tto@idiap.ch                     #
#                                                                             #
# Alternatively, this file may be used under the terms of the GNU Affero      #
# Public License version 3 as published by the Free Software and appearing    #
# in the file LICENSE.AGPL included in the packaging of this file.            #
# The BEAT platform is distributed in the hope that it will be useful, but    #
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY  #
# or FITNESS FOR A PARTICULAR PURPOSE.                                        #
#                                                                             #
# You should have received a copy of the GNU Affero Public License along      #
# with the BEAT platform. If not, see http://www.gnu.org/licenses/.           #
#                                                                             #
###############################################################################


# Basic setup for slow tests

import os
Philip ABBET's avatar
Philip ABBET committed
32
import sys
33
import subprocess
André Anjos's avatar
André Anjos committed
34
35
36
import shutil
import tempfile
import pkg_resources
37
import logging
André Anjos's avatar
André Anjos committed
38

Philip ABBET's avatar
Philip ABBET committed
39
if sys.platform == 'darwin':
40
41
42
43
44
45
    tmp_prefix = tempfile.mkdtemp(prefix=__name__,
                                  suffix='.tmpdir',
                                  dir='/tmp')
    prefix_folder = tempfile.mkdtemp(prefix=__name__,
                                     suffix='.prefix',
                                     dir='/tmp')
Philip ABBET's avatar
Philip ABBET committed
46
else:
47
48
49
50
    tmp_prefix = tempfile.mkdtemp(prefix=__name__,
                                  suffix='.tmpdir')
    prefix_folder = tempfile.mkdtemp(prefix=__name__,
                                     suffix='.prefix')
André Anjos's avatar
André Anjos committed
51
52


53
prefix = os.path.join(prefix_folder, 'prefix')
54
55

DOCKER_NETWORK_TEST_ENABLED = os.environ.get('DOCKER_NETWORK_TEST_ENABLED', False) == 'True'
56
57
network_name = os.environ.get('DOCKER_TEST_NETWORK', 'beat_core_test_network')
network = None
André Anjos's avatar
André Anjos committed
58

59
# Setup the logging system
60
61
62
VERBOSE_TEST_LOGGING = os.environ.get('VERBOSE_TEST_LOGGING', False) == 'True'

if VERBOSE_TEST_LOGGING:
63
    formatter = logging.Formatter(fmt="[%(asctime)s - TESTS - "
64
65
66
67
68
69
70
                                      "%(name)s] %(levelname)s: %(message)s",
                                  datefmt="%d/%b/%Y %H:%M:%S")

    handler = logging.StreamHandler()
    handler.setFormatter(formatter)

    beat_core_logger = logging.getLogger('beat.core')
71
    beat_core_logger.setLevel(logging.DEBUG)
72
73
74
    beat_core_logger.addHandler(handler)

    beat_backend_logger = logging.getLogger('beat.backend.python')
75
    beat_backend_logger.setLevel(logging.DEBUG)
76
77
78
    beat_backend_logger.addHandler(handler)


79
80
81
82
83
84
85
86
87
def setup_package():
    prefixes = [
        pkg_resources.resource_filename('beat.backend.python.test', 'prefix'),
        pkg_resources.resource_filename('beat.core.test', 'prefix')
    ]

    for path in prefixes:
        subprocess.check_call(['rsync', '-arz', path, prefix_folder])

88
89
90
91
92
93
94
95
96
97
    if DOCKER_NETWORK_TEST_ENABLED:
        import docker
        client = docker.from_env()
        try:
            network = client.networks.get(network_name)
        except docker.errors.NotFound:
            subnet = os.environ.get('DOCKER_TEST_SUBNET', '193.169.0.0/24')
            gateway = os.environ.get('DOCKER_TEST_GATEWAY', '193.169.0.254')
            ipam_pool = docker.types.IPAMPool(subnet=subnet,
                                          gateway=gateway)
98

99
            ipam_config = docker.types.IPAMConfig(pool_configs=[ipam_pool])
100

101
102
103
            network = client.networks.create(network_name,
                                             driver="bridge",
                                             ipam=ipam_config)
104
105

def teardown_package():
Philip ABBET's avatar
Philip ABBET committed
106
107
    if os.path.exists(tmp_prefix):
        shutil.rmtree(tmp_prefix)
André Anjos's avatar
André Anjos committed
108

109
    shutil.rmtree(prefix_folder)
110

111
112
113
    if DOCKER_NETWORK_TEST_ENABLED:
        if network:
            network.remove()