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