[RIC-621] - Fix Automation
[ric-plt/e2mgr.git] / Automation / Tests / Scripts / cleanup_db.py
1 ##############################################################################
2 #
3 #   Copyright (c) 2019 AT&T Intellectual Property.
4 #
5 #   Licensed under the Apache License, Version 2.0 (the "License");
6 #   you may not use this file except in compliance with the License.
7 #   You may obtain a copy of the License at
8 #
9 #       http://www.apache.org/licenses/LICENSE-2.0
10 #
11 #   Unless required by applicable law or agreed to in writing, software
12 #   distributed under the License is distributed on an "AS IS" BASIS,
13 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 #   See the License for the specific language governing permissions and
15 #   limitations under the License.
16 #
17 ##############################################################################
18 #
19 #   This source code is part of the near-RT RIC (RAN Intelligent Controller)
20 #   platform project (RICP).
21 #
22 import config
23 import redis
24 import time
25 import k8s_helper
26
27
28 def flush(set_new_timestamp):
29     c = config.redis_ip_address
30
31     p = config.redis_ip_port
32
33     r = redis.Redis(host=c, port=p, db=0, )
34
35     e2t_ip = k8s_helper.extract_service_ip("e2term-rmr-alpha")
36     et2_address = e2t_ip + ":38000"
37
38     r.flushall()
39     r.set("{e2Manager},GENERAL", "{\"enableRic\":true}")
40     r.set("{e2Manager},E2TAddresses", "[\"{}\"]".format(et2_address))
41
42     timestamp = str(int((time.time() + 2) * 1000000000)) if set_new_timestamp else str(
43         int((time.time() - 300) * 1000000000))
44     r.set("{{e2Manager}},E2TInstance:{}".format(et2_address),
45           "{{\"address\":\"{}\",\"associatedRanList\":[],\"keepAliveTimestamp\":".format(et2_address) + timestamp +
46           ",\"state\":\"ACTIVE\",\"deletionTimeStamp\":0}")
47
48     return True