osm_mon/tests/unit/collector/vnf_collectors/vmware/mock_http.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright 2019 VMware
   3 # *************************************************************
   4
   5 # This file is part of OSM Monitoring module
   6 # All Rights Reserved to VMware
   7
   8 # Licensed under the Apache License, Version 2.0 (the "License"); you may
   9 # not use this file except in compliance with the License. You may obtain
  10 # a copy of the License at
  11
  12 #         http://www.apache.org/licenses/LICENSE-2.0
  13
  14 # Unless required by applicable law or agreed to in writing, software
  15 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  16 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  17 # License for the specific language governing permissions and limitations
  18 # under the License.
  19 # For those usages not covered by the Apache License, Version 2.0 please
  20 # contact: mbeierl@vmware.com
  21 # #
  22
  23 import os
  24 import re
  25
  26
  27 def mock_http_response(mocker, method='GET', site='https://vrops', url_pattern='', response_file='OK',
  28                        status_code=200, exception=None):
  29     '''Helper function to load a canned response from a file.'''
  30     with open(os.path.join(os.path.dirname(__file__), 'vmware_mocks',
  31                            '%s' % response_file), 'r') as f:
  32         response = f.read()
  33
  34     matcher = re.compile(site + url_pattern)
  35     if exception is None:
  36         mocker.register_uri(method, matcher, text=response, status_code=status_code)
  37     else:
  38         mocker.register_uri(method, matcher, exc=exception)