| Wei-Yu Chen | ad55cb8 | 2022-02-15 20:07:01 +0800 | [diff] [blame] | 1 | # SPDX-FileCopyrightText: 2020 The Magma Authors. |
| 2 | # SPDX-FileCopyrightText: 2022 Open Networking Foundation <support@opennetworking.org> |
| 3 | # |
| 4 | # SPDX-License-Identifier: BSD-3-Clause |
| Wei-Yu Chen | 49950b9 | 2021-11-08 19:19:18 +0800 | [diff] [blame] | 5 | |
| 6 | from unittest import mock |
| 7 | |
| 8 | from spyne.server.wsgi import WsgiMethodContext |
| 9 | |
| 10 | |
| 11 | def get_spyne_context_with_ip( |
| 12 | req_ip: str = "192.168.60.145", |
| 13 | ) -> WsgiMethodContext: |
| 14 | with mock.patch('spyne.server.wsgi.WsgiApplication') as MockTransport: |
| 15 | MockTransport.req_env = {"REMOTE_ADDR": req_ip} |
| 16 | with mock.patch('spyne.server.wsgi.WsgiMethodContext') as MockContext: |
| 17 | MockContext.transport = MockTransport |
| 18 | return MockContext |