test_config.py 3.63 KB
Newer Older
André Anjos's avatar
André Anjos committed
1
2
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :
André Anjos's avatar
André Anjos committed
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98

###############################################################################
#                                                                             #
# Copyright (c) 2016 Idiap Research Institute, http://www.idiap.ch/           #
# Contact: beat.support@idiap.ch                                              #
#                                                                             #
# This file is part of the beat.cmdline 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 tests for the command line beat program: config

import os

import nose.tools
import simplejson

from . import tmp_prefix
from ..scripts.beat import main
from beat.core.test.utils import cleanup
from .. import config


def call(*args, **kwargs):
  '''A central mechanism to call the main routine with the right parameters'''

  use_prefix = kwargs.get('prefix', tmp_prefix)

  return main((
      '--prefix=%s' % use_prefix,
      '--test-mode',
      ) + args)


def test_config_list():
  nose.tools.eq_(call('config', 'list'), 0)


def test_config_cache():
  cache_dir = 'cache'
  c = config.Configuration({'--cache': cache_dir})
  nose.tools.eq_(c.cache, os.path.join(c.path, cache_dir))

  cache_dir = '/an/absolute/cache/dir'
  c = config.Configuration({'--cache': cache_dir})
  nose.tools.eq_(c.cache, cache_dir)


@nose.tools.with_setup(teardown=cleanup)
def test_config_save():
  nose.tools.eq_(call('config', 'save'), 0)
  config = os.path.join(tmp_prefix, '.beat', 'config.json')
  assert os.path.exists(config)
  with open(config, 'rt') as f: contents = simplejson.load(f)
  assert contents


@nose.tools.with_setup(teardown=cleanup)
def test_set_token():
  token_value = '123456abcdef'
  nose.tools.eq_(call('config', 'set', 'token', token_value), 0)
  config = os.path.join(tmp_prefix, '.beat', 'config.json')
  assert os.path.exists(config)
  with open(config, 'rt') as f: contents = simplejson.load(f)
  assert contents['token'] == token_value


@nose.tools.with_setup(teardown=cleanup)
def test_get_token():
  nose.tools.eq_(call('config', 'get', 'token'), 0)


@nose.tools.with_setup(teardown=cleanup)
def test_color_list():
  nose.tools.eq_(call('config', 'color', '--list'), 0)


@nose.tools.with_setup(teardown=cleanup)
def test_color_set():
  nose.tools.eq_(call('config', 'color', 'dark'), 0)