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
, **kwargs
):
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 async def scale(self
, scale
):
182 units
= [FakeUnit(), FakeUnit()]
186 def __init__(self
, content
: str = ""):
187 self
.content
= content
189 def read(self
, size
: int = -1):
193 class FakeFileWrapper
:
194 def __init__(self
, content
: str = ""):
195 self
.file = FakeFile(content
=content
)
200 def __exit__(self
, type, value
, traceback
):
204 FAKE_DELTA_MACHINE_PENDING
= Dict(
206 "deltas": ["machine", "change", {}],
211 "instance-id": "juju-1b5808-2",
212 "agent-status": {"current": "pending", "message": "", "version": ""},
213 "instance-status": {"current": "running", "message": "Running"},
217 FAKE_DELTA_MACHINE_STARTED
= Dict(
219 "deltas": ["machine", "change", {}],
224 "instance-id": "juju-1b5808-2",
225 "agent-status": {"current": "started", "message": "", "version": ""},
226 "instance-status": {"current": "running", "message": "Running"},
231 FAKE_DELTA_UNIT_PENDING
= Dict(
233 "deltas": ["unit", "change", {}],
238 "application": "git",
240 "workload-status": {"current": "waiting", "message": ""},
241 "agent-status": {"current": "idle", "message": ""},
246 FAKE_DELTA_UNIT_STARTED
= Dict(
248 "deltas": ["unit", "change", {}],
253 "application": "git",
255 "workload-status": {"current": "active", "message": ""},
256 "agent-status": {"current": "idle", "message": ""},
261 FAKE_DELTA_APPLICATION_MAINTENANCE
= Dict(
263 "deltas": ["application", "change", {}],
264 "entity": "application",
269 "current": "maintenance",
270 "message": "installing charm software",
276 FAKE_DELTA_APPLICATION_ACTIVE
= Dict(
278 "deltas": ["application", "change", {}],
279 "entity": "application",
281 "data": {"name": "git", "status": {"current": "active", "message": "Ready!"}},
285 FAKE_DELTA_ACTION_COMPLETED
= Dict(
287 "deltas": ["action", "change", {}],
291 "model-uuid": "af19cdd4-374a-4d9f-86b1-bfed7b1b5808",
295 "status": "completed",
304 "entity": Dict({"id": "2", "type": "machine"}),
305 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
306 "delta": FAKE_DELTA_MACHINE_PENDING
,
307 "entity_status": Dict(
308 {"status": "pending", "message": "Running", "vca_status": "running"}
315 "message": "Running",
317 "vca_status": "running",
318 "n2vc_status": N2VCDeploymentStatus
.PENDING
,
327 "entity": Dict({"id": "2", "type": "machine"}),
328 "filter": Dict({"entity_id": "1", "entity_type": "machine"}),
329 "delta": FAKE_DELTA_MACHINE_PENDING
,
330 "entity_status": Dict(
331 {"status": "pending", "message": "Running", "vca_status": "running"}
333 "db": Dict({"written": False, "data": None}),
338 "entity": Dict({"id": "2", "type": "machine"}),
339 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
340 "delta": FAKE_DELTA_MACHINE_STARTED
,
341 "entity_status": Dict(
342 {"status": "started", "message": "Running", "vca_status": "running"}
349 "message": "Running",
351 "vca_status": "running",
352 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
361 "entity": Dict({"id": "2", "type": "machine"}),
362 "filter": Dict({"entity_id": "1", "entity_type": "machine"}),
363 "delta": FAKE_DELTA_MACHINE_STARTED
,
364 "entity_status": Dict(
365 {"status": "started", "message": "Running", "vca_status": "running"}
367 "db": Dict({"written": False, "data": None}),
372 "entity": Dict({"id": "git/0", "type": "unit"}),
373 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
374 "delta": FAKE_DELTA_UNIT_PENDING
,
375 "entity_status": Dict(
376 {"status": "waiting", "message": "", "vca_status": "waiting"}
385 "vca_status": "waiting",
386 "n2vc_status": N2VCDeploymentStatus
.RUNNING
,
395 "entity": Dict({"id": "git/0", "type": "unit"}),
396 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
397 "delta": FAKE_DELTA_UNIT_PENDING
,
398 "entity_status": Dict(
399 {"status": "waiting", "message": "", "vca_status": "waiting"}
401 "db": Dict({"written": False, "data": None}),
406 "entity": Dict({"id": "git/0", "type": "unit"}),
407 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
408 "delta": FAKE_DELTA_UNIT_STARTED
,
409 "entity_status": Dict(
410 {"status": "active", "message": "", "vca_status": "active"}
419 "vca_status": "active",
420 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
429 "entity": Dict({"id": "git/0", "type": "unit"}),
430 "filter": Dict({"entity_id": "1", "entity_type": "action"}),
431 "delta": FAKE_DELTA_UNIT_STARTED
,
432 "entity_status": Dict(
433 {"status": "active", "message": "", "vca_status": "active"}
435 "db": Dict({"written": False, "data": None}),
440 "entity": Dict({"id": "git", "type": "application"}),
441 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
442 "delta": FAKE_DELTA_APPLICATION_MAINTENANCE
,
443 "entity_status": Dict(
445 "status": "maintenance",
446 "message": "installing charm software",
447 "vca_status": "maintenance",
455 "message": "installing charm software",
456 "entity": "application",
457 "vca_status": "maintenance",
458 "n2vc_status": N2VCDeploymentStatus
.RUNNING
,
467 "entity": Dict({"id": "git", "type": "application"}),
468 "filter": Dict({"entity_id": "2", "entity_type": "machine"}),
469 "delta": FAKE_DELTA_APPLICATION_MAINTENANCE
,
470 "entity_status": Dict(
472 "status": "maintenance",
473 "message": "installing charm software",
474 "vca_status": "maintenance",
477 "db": Dict({"written": False, "data": None}),
482 "entity": Dict({"id": "git", "type": "application"}),
483 "filter": Dict({"entity_id": "git", "entity_type": "application"}),
484 "delta": FAKE_DELTA_APPLICATION_ACTIVE
,
485 "entity_status": Dict(
486 {"status": "active", "message": "Ready!", "vca_status": "active"}
494 "entity": "application",
495 "vca_status": "active",
496 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
505 "entity": Dict({"id": "git", "type": "application"}),
506 "filter": Dict({"entity_id": "1", "entity_type": "action"}),
507 "delta": FAKE_DELTA_APPLICATION_ACTIVE
,
508 "entity_status": Dict(
509 {"status": "active", "message": "Ready!", "vca_status": "active"}
511 "db": Dict({"written": False, "data": None}),
516 "entity": Dict({"id": "1", "type": "action"}),
517 "filter": Dict({"entity_id": "1", "entity_type": "action"}),
518 "delta": FAKE_DELTA_ACTION_COMPLETED
,
519 "entity_status": Dict(
521 "status": "completed",
522 "message": "completed",
523 "vca_status": "completed",
531 "message": "completed",
533 "vca_status": "completed",
534 "n2vc_status": N2VCDeploymentStatus
.COMPLETED
,
543 "entity": Dict({"id": "git", "type": "action"}),
544 "filter": Dict({"entity_id": "1", "entity_type": "machine"}),
545 "delta": FAKE_DELTA_ACTION_COMPLETED
,
546 "entity_status": Dict(
548 "status": "completed",
549 "message": "completed",
550 "vca_status": "completed",
553 "db": Dict({"written": False, "data": None}),