1
# Copyright 2012 Canonical Ltd. This software is licensed under the
2
# GNU Affero General Public License version 3 (see the file LICENSE).
4
"""Tests for `provisioningserver.cluster_config`."""
6
from __future__ import (
15
from fixtures import EnvironmentVariableFixture
16
from maastesting.factory import factory
17
from maastesting.testcase import TestCase
18
from provisioningserver.cluster_config import (
25
class TestClusterConfig(TestCase):
27
def test_get_cluster_variable_reads_env(self):
28
var = factory.make_name('variable')
29
value = factory.make_name('value')
30
self.useFixture(EnvironmentVariableFixture(var, value))
31
self.assertEqual(value, get_cluster_variable(var))
33
def test_get_cluster_variable_fails_if_not_set(self):
36
get_cluster_variable, factory.make_name('nonexistent-variable'))
38
def test_get_cluster_uuid_reads_CLUSTER_UUID(self):
39
uuid = factory.make_name('uuid')
40
self.useFixture(EnvironmentVariableFixture('CLUSTER_UUID', uuid))
41
self.assertEqual(uuid, get_cluster_uuid())
43
def test_get_maas_url_reads_MAAS_URL(self):
44
maas_url = factory.make_name('maas_url')
45
self.useFixture(EnvironmentVariableFixture('MAAS_URL', maas_url))
46
self.assertEqual(maas_url, get_maas_url())