1 # Copyright 2020 Canonical Ltd.
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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.
17 from n2vc
.utils
import Dict
, N2VCDeploymentStatus
18 from n2vc
.n2vc_conn
import N2VCConnector
19 from unittest
.mock
import MagicMock
22 kubeconfig
= """apiVersion: v1
25 certificate-authority-data: LS0tLS1CRUdJTiBDRVJUSUZJQ0FURS0tLS0tCk1\
26 JSURBVENDQWVtZ0F3SUJBZ0lKQUxjMk9xVUpwcnVCTUEwR0NTcUdTSWIzRFFFQk\
27 N3VUFNQmN4RlRBVEJnTlYKQkFNTURERXdMakUxTWk0eE9ETXVNVEFlRncweU1EQ\
28 TVNVEV4TkRJeU16VmFGdzB6TURBNU1Ea3hOREl5TXpWYQpNQmN4RlRBVEJnTlZC\
29 QU1NRERFd0xqRTFNaTR4T0RNdU1UQ0NBU0l3RFFZSktvWklodmNOQVFFQkJRQUR\
30 nZ0VQCkFEQ0NBUW9DZ2dFQkFNV0tyQkdxWlJRT0VONDExR2RESmY2ckZWRDcvMU\
31 xHNlZMWjNhd1BRdHBhRTRxdVdyNisKWjExTWwra2kwVEU1cGZFV3dKenVUZXlCU\
32 XVkUEpnYm1QTjF1VWROdGRiNlpocHEzeC9oT0hCMVJLNC9iSlNFUgpiZ0dITmN6\
33 MzR6SHRaZ1dwb2NPTXpPOW9oRUdhMTZUaDhmQWVxYU1CQTJRaklmeUFlaVp3VHJ\
34 nZ3BrY2dBMUlOCjBvQkdqSURnSGVoSU5tbGZOOURkQ3hNN1FNTmtSbzRXdE13bF\
35 JSRWZ4QnFiVkNpZGFjbVhhb1VPUjJPeFVmQWEKN1orSUU1TmN5ZFQ1TGovazdwd\
36 XZCVkdIa0JQWnE0TmlBa3R4aXd5NVB5R29GTk9mT0NrV2I2VnBzVzNhTlNJeAo4\
37 aXBITkc3enV3elc1TGQ5TkhQYWpRckZwdFZBSHpJNWNhRUNBd0VBQWFOUU1FNHd\
38 IUVlEVlIwT0JCWUVGQ1dVCkFaTXNaeE13L1k1OGlXMGZJWVAzcDdTYk1COEdBMV\
39 VkSXdRWU1CYUFGQ1dVQVpNc1p4TXcvWTU4aVcwZklZUDMKcDdTYk1Bd0dBMVVkR\
40 XdRRk1BTUJBZjh3RFFZSktvWklodmNOQVFFTEJRQURnZ0VCQUJaMlYxMWowRzhh\
41 Z1Z6Twp2YWtKTGt4UGZ0UE1NMFFOaVRzZmV6RzlicnBkdEVLSjFyalFCblNXYTN\
42 WbThWRGZTYkhLQUNXaGh0OEhzcXhtCmNzdVQyOWUyaGZBNHVIOUxMdy9MVG5EdE\
43 tJSjZ6aWFzaTM5RGh3UGwwaExuamJRMjk4VVo5TGovVlpnZGlqemIKWnVPdHlpT\
44 nVOS0E2Nmd0dGxXcWZRQ2hkbnJ5MlZUbjBjblR5dU9UalByYWdOdXJMdlVwL3Nl\
45 eURhZmsxNXJ4egozcmlYZldiQnRhUUk1dnM0ekFKU2xneUg2RnpiZStoTUhlUzF\
46 mM2ppb3dJV0lRR2NNbHpGT1RpMm1xWFRybEJYCnh1WmpLZlpOcndjQVNGbk9qYV\
47 BWeFQ1ODJ4WWhtTm8wR3J2MlZEck51bDlSYkgvK3lNS2J5NEhkOFRvVThMU2kKY\
48 3Uxajh3cz0KLS0tLS1FTkQgQ0VSVElGSUNBVEUtLS0tLQo=
49 server: https://192.168.0.22:16443
50 name: microk8s-cluster
53 cluster: microk8s-cluster
56 current-context: microk8s
62 token: clhkRExRem5Xd1dCdnFEVXdvRGtDRGE5b1F3WnNrZk5qeHFCOU10bHBZRT0K
66 async def AsyncMockFunc():
67 await asyncio
.sleep(1)
70 class AsyncMock(MagicMock
):
71 async def __call__(self
, *args
, **kwargs
):
72 return super(AsyncMock
, self
).__call
__(*args
, **kwargs
)
75 class FakeN2VC(MagicMock
):
76 last_written_values
= None
78 async def write_app_status_to_db(
81 status
: N2VCDeploymentStatus
,
88 Write application status to database
90 :param: db_dict: DB dictionary
91 :param: status: Status of the application
92 :param: detailed_status: Detailed status
93 :param: vca_status: VCA status
94 :param: entity_type: Entity type ("application", "machine, and "action")
95 :param: vca_id: Id of the VCA. If None, the default VCA will be used.
97 self
.last_written_values
= Dict(
99 "n2vc_status": status
,
100 "message": detailed_status
,
101 "vca_status": vca_status
,
102 "entity": entity_type
,
106 osm_status
= N2VCConnector
.osm_status
109 class FakeMachine(MagicMock
):
111 dns_name
= "FAKE ENDPOINT"
112 model_name
= "FAKE MODEL"
113 entity_type
= "machine"
114 safe_data
= {"instance-id": "myid"}
116 async def destroy(self
, force
):
120 class FakeManualMachine(MagicMock
):
122 dns_name
= "FAKE ENDPOINT"
123 model_name
= "FAKE MODEL"
124 entity_type
= "machine"
125 safe_data
= {"instance-id": "manual:myid"}
127 async def destroy(self
, force
):
131 class FakeWatcher(AsyncMock
):
133 delta_to_return
= None
135 async def Next(self
):
136 return Dict({"deltas": self
.delta_to_return
})
139 class FakeConnection(MagicMock
):
144 class FakeAction(MagicMock
):
150 def __init__(self
, applications
: dict = {}):
151 self
._applications
= applications
154 def applications(self
):
155 return self
._applications
158 class FakeUnit(MagicMock
):
159 async def is_leader_from_status(self
):
162 async def run_action(self
, action_name
):
166 class FakeApplication(AsyncMock
):
167 async def set_config(self
, config
):
170 async def add_unit(self
, to
):
173 async def get_actions(self
):
174 return ["existing_action"]
176 async def get_config(self
):
177 return ["app_config"]
179 units
= [FakeUnit(), FakeUnit()]
183 def __init__(self
, content
: str = ""):
184 self
.content
= content
186 def read(self
, size
: int = -1):
190 class FakeFileWrapper
:
191 def __init__(self
, content
: str = ""):
192 self
.file = FakeFile(content
=content
)
197 def __exit__(self
, type, value
, traceback
):
201 FAKE_DELTA_MACHINE_PENDING
= Dict(
203 "deltas": ["machine", "change", {}],
208 "instance-id": "juju-1b5808-2",
209 "agent-status": {"current": "pending", "message": "", "version": ""},
210 "instance-status": {"current": "running", "message": "Running"},
214 FAKE_DELTA_MACHINE_STARTED
= Dict(
216 "deltas": ["machine", "change", {}],
221 "instance-id": "juju-1b5808-2",
222 "agent-status": {"current": "started", "message": "", "version": ""},
223 "instance-status": {"current": "running", "message": "Running"},
228 FAKE_DELTA_UNIT_PENDING
= Dict(
230 "deltas": ["unit", "change", {}],
235 "application": "git",
237 "workload-status": {"current": "waiting", "message": ""},
238 "agent-status": {"current": "idle", "message": ""},
243 FAKE_DELTA_UNIT_STARTED
= Dict(
245 "deltas": ["unit", "change", {}],
250 "application": "git",
252 "workload-status": {"current": "active", "message": ""},
253 "agent-status": {"current": "idle", "message": ""},
258 FAKE_DELTA_APPLICATION_MAINTENANCE
= Dict(
260 "deltas": ["application", "change", {}],
261 "entity": "application",
266 "current": "maintenance",
267 "message": "installing charm software",
273 FAKE_DELTA_APPLICATION_ACTIVE
= Dict(
275 "deltas": ["application", "change", {}],
276 "entity": "application",
278 "data": {"name": "git", "status": {"current": "active", "message": "Ready!"}},
282 FAKE_DELTA_ACTION_COMPLETED
= Dict(
284 "deltas": ["action", "change", {}],
288 "model-uuid": "af19cdd4-374a-4d9f-86b1-bfed7b1b5808",
292 "status": "completed",
301 "entity": Dict({"id": "2", "type": "machine"}),
302 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
303 "delta": FAKE_DELTA_MACHINE_PENDING
,
304 "entity_status": Dict(
305 {"status": "pending", "message": "Running", "vca_status": "running"}
312 "message": "Running",
314 "vca_status": "running",
315 "n2vc_status": N2VCDeploymentStatus
.PENDING
,
324 "entity": Dict({"id": "2", "type": "machine"}),
325 "filter": Dict({"entity_id": "1", "entity_type": "machine"}),
326 "delta": FAKE_DELTA_MACHINE_PENDING
,
327 "entity_status": Dict(
328 {"status": "pending", "message": "Running", "vca_status": "running"}
330 "db": Dict({"written": False, "data": None}),
335 "entity": Dict({"id": "2", "type": "machine"}),
336 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
337 "delta": FAKE_DELTA_MACHINE_STARTED
,
338 "entity_status": Dict(
339 {"status": "started", "message": "Running", "vca_status": "running"}
346 "message": "Running",
348 "vca_status": "running",
349 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
358 "entity": Dict({"id": "2", "type": "machine"}),
359 "filter": Dict({"entity_id": "1", "entity_type": "machine"}),
360 "delta": FAKE_DELTA_MACHINE_STARTED
,
361 "entity_status": Dict(
362 {"status": "started", "message": "Running", "vca_status": "running"}
364 "db": Dict({"written": False, "data": None}),
369 "entity": Dict({"id": "git/0", "type": "unit"}),
370 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
371 "delta": FAKE_DELTA_UNIT_PENDING
,
372 "entity_status": Dict(
373 {"status": "waiting", "message": "", "vca_status": "waiting"}
382 "vca_status": "waiting",
383 "n2vc_status": N2VCDeploymentStatus
.RUNNING
,
392 "entity": Dict({"id": "git/0", "type": "unit"}),
393 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
394 "delta": FAKE_DELTA_UNIT_PENDING
,
395 "entity_status": Dict(
396 {"status": "waiting", "message": "", "vca_status": "waiting"}
398 "db": Dict({"written": False, "data": None}),
403 "entity": Dict({"id": "git/0", "type": "unit"}),
404 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
405 "delta": FAKE_DELTA_UNIT_STARTED
,
406 "entity_status": Dict(
407 {"status": "active", "message": "", "vca_status": "active"}
416 "vca_status": "active",
417 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
426 "entity": Dict({"id": "git/0", "type": "unit"}),
427 "filter": Dict({"entity_id": "1", "entity_type": "action"}),
428 "delta": FAKE_DELTA_UNIT_STARTED
,
429 "entity_status": Dict(
430 {"status": "active", "message": "", "vca_status": "active"}
432 "db": Dict({"written": False, "data": None}),
437 "entity": Dict({"id": "git", "type": "application"}),
438 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
439 "delta": FAKE_DELTA_APPLICATION_MAINTENANCE
,
440 "entity_status": Dict(
442 "status": "maintenance",
443 "message": "installing charm software",
444 "vca_status": "maintenance",
452 "message": "installing charm software",
453 "entity": "application",
454 "vca_status": "maintenance",
455 "n2vc_status": N2VCDeploymentStatus
.RUNNING
,
464 "entity": Dict({"id": "git", "type": "application"}),
465 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
466 "delta": FAKE_DELTA_APPLICATION_MAINTENANCE
,
467 "entity_status": Dict(
469 "status": "maintenance",
470 "message": "installing charm software",
471 "vca_status": "maintenance",
474 "db": Dict({"written": False, "data": None}),
479 "entity": Dict({"id": "git", "type": "application"}),
480 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
481 "delta": FAKE_DELTA_APPLICATION_ACTIVE
,
482 "entity_status": Dict(
483 {"status": "active", "message": "Ready!", "vca_status": "active"}
491 "entity": "application",
492 "vca_status": "active",
493 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
502 "entity": Dict({"id": "git", "type": "application"}),
503 "filter": Dict({"entity_id": "1", "entity_type": "action"}),
504 "delta": FAKE_DELTA_APPLICATION_ACTIVE
,
505 "entity_status": Dict(
506 {"status": "active", "message": "Ready!", "vca_status": "active"}
508 "db": Dict({"written": False, "data": None}),
513 "entity": Dict({"id": "1", "type": "action"}),
514 "filter": Dict({"entity_id": "1", "entity_type": "action"}),
515 "delta": FAKE_DELTA_ACTION_COMPLETED
,
516 "entity_status": Dict(
518 "status": "completed",
519 "message": "completed",
520 "vca_status": "completed",
528 "message": "completed",
530 "vca_status": "completed",
531 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
540 "entity": Dict({"id": "git", "type": "action"}),
541 "filter": Dict({"entity_id": "1", "entity_type": "machine"}),
542 "delta": FAKE_DELTA_ACTION_COMPLETED
,
543 "entity_status": Dict(
545 "status": "completed",
546 "message": "completed",
547 "vca_status": "completed",
550 "db": Dict({"written": False, "data": None}),