pgaas/tests/psycopg2.py

   1 # ============LICENSE_START====================================================
   2 # org.onap.dcaegen2
   3 # =============================================================================
   4 # Copyright (c) 2017-2020 AT&T Intellectual Property. All rights reserved.
   5 # =============================================================================
   6 # Licensed under the Apache License, Version 2.0 (the "License");
   7 # you may not use this file except in compliance with the License.
   8 # You may obtain a copy of the License at
   9 #
  10 #      http://www.apache.org/licenses/LICENSE-2.0
  11 #
  12 # Unless required by applicable law or agreed to in writing, software
  13 # distributed under the License is distributed on an "AS IS" BASIS,
  14 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15 # See the License for the specific language governing permissions and
  16 # limitations under the License.
  17 # ============LICENSE_END======================================================
  18
  19 """
  20
  21 This is a mock psycopg2 module.
  22
  23 """
  24
  25 class MockCursor(object):
  26   """
  27   mocked cursor
  28   """
  29   def __init__(self, **kwargs):
  30     pass
  31
  32   def execute(self, cmd, exc=None):
  33     """
  34     mock SQL execution
  35     """
  36     pass
  37
  38   def close(self):
  39     """
  40     mock SQL close
  41     """
  42     pass
  43
  44   def __iter__(self):
  45     return iter([])
  46
  47 class MockConn(object): # pylint: disable=too-few-public-methods
  48   """
  49   mock SQL connection
  50   """
  51   def __init__(self, **kwargs):
  52     pass
  53
  54   def __enter__(self):
  55     return self
  56
  57   def __exit__(self, exc_type, exc_value, traceback):
  58     pass
  59
  60   def cursor(self): # pylint: disable=no-self-use
  61     """
  62     mock return a cursor
  63     """
  64     return MockCursor()
  65
  66 def connect(**kwargs): # pylint: disable=unused-argument
  67   """
  68   mock get-a-connection
  69   """
  70   return MockConn()