Updating python dependencies
[osm/N2VC.git] / n2vc / tests / unit / test_utils.py
1 # Copyright 2020 Canonical Ltd.
2 #
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 # http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14
15 from unittest import TestCase
16
17 from n2vc.utils import (
18 Dict,
19 EntityType,
20 JujuStatusToOSM,
21 N2VCDeploymentStatus,
22 get_ee_id_components,
23 )
24 from juju.machine import Machine
25 from juju.application import Application
26 from juju.action import Action
27 from juju.unit import Unit
28
29
30 class UtilsTest(TestCase):
31 def test_dict(self):
32 example = Dict({"key": "value"})
33 self.assertEqual(example["key"], example.key)
34
35 def test_entity_type(self):
36 self.assertFalse(EntityType.has_value("machine2"))
37 values = [Machine, Application, Action, Unit]
38 for value in values:
39 self.assertTrue(EntityType.has_value(value))
40
41 self.assertEqual(EntityType.MACHINE, EntityType.get_entity(Machine))
42 self.assertEqual(EntityType.APPLICATION, EntityType.get_entity(Application))
43 self.assertEqual(EntityType.UNIT, EntityType.get_entity(Unit))
44 self.assertEqual(EntityType.ACTION, EntityType.get_entity(Action))
45
46 def test_juju_status_to_osm(self):
47 tests = [
48 {
49 "entity_type": "machine",
50 "status": [
51 {"juju": "pending", "osm": N2VCDeploymentStatus.PENDING},
52 {"juju": "started", "osm": N2VCDeploymentStatus.COMPLETED},
53 ],
54 },
55 {
56 "entity_type": "application",
57 "status": [
58 {"juju": "waiting", "osm": N2VCDeploymentStatus.RUNNING},
59 {"juju": "maintenance", "osm": N2VCDeploymentStatus.RUNNING},
60 {"juju": "blocked", "osm": N2VCDeploymentStatus.RUNNING},
61 {"juju": "error", "osm": N2VCDeploymentStatus.FAILED},
62 {"juju": "active", "osm": N2VCDeploymentStatus.COMPLETED},
63 ],
64 },
65 {
66 "entity_type": "unit",
67 "status": [
68 {"juju": "waiting", "osm": N2VCDeploymentStatus.RUNNING},
69 {"juju": "maintenance", "osm": N2VCDeploymentStatus.RUNNING},
70 {"juju": "blocked", "osm": N2VCDeploymentStatus.RUNNING},
71 {"juju": "error", "osm": N2VCDeploymentStatus.FAILED},
72 {"juju": "active", "osm": N2VCDeploymentStatus.COMPLETED},
73 ],
74 },
75 {
76 "entity_type": "action",
77 "status": [
78 {"juju": "running", "osm": N2VCDeploymentStatus.RUNNING},
79 {"juju": "completed", "osm": N2VCDeploymentStatus.COMPLETED},
80 ],
81 },
82 ]
83
84 for test in tests:
85 entity_type = test["entity_type"]
86 self.assertTrue(entity_type in JujuStatusToOSM)
87
88 for status in test["status"]:
89 juju_status = status["juju"]
90 osm_status = status["osm"]
91 self.assertTrue(juju_status in JujuStatusToOSM[entity_type])
92 self.assertEqual(osm_status, JujuStatusToOSM[entity_type][juju_status])
93
94
95 class GetEEComponentTest(TestCase):
96 def test_valid(self):
97 model, application, machine = get_ee_id_components("model.application.machine")
98 self.assertEqual(model, "model")
99 self.assertEqual(application, "application")
100 self.assertEqual(machine, "machine")
101
102 def test_invalid(self):
103 with self.assertRaises(Exception):
104 get_ee_id_components("model.application.machine.1")
105 with self.assertRaises(Exception):
106 get_ee_id_components("model.application")