async def test_vnf_package_ids_token_valid(): v = VnfDetector() mocked_get_auth_token = CoroutineMock() mocked_get_auth_token.return_value = "dummy" v._token = mocked_get_auth_token.return_value v._get_and_set_authentication_token = mocked_get_auth_token assert "dummy" == await v._get_and_set_authentication_token(None) resp = CoroutineMock() first_get_response = MagicMock() first_get_response.__aenter__.return_value = resp g = asyncio.Future() resp.json.return_value = g g.set_result([{"_id": 1}, {"_id": 2}]) context_manager = MagicMock(side_effect=[ first_get_response, ]) mocked_session = MagicMock() mocked_session.get = context_manager resp.status = 200 pkg_ids = await v.get_vnf_package_ids(mocked_session) assert pkg_ids == [1, 2] assert mocked_get_auth_token.await_count == 1
def session(): sess = MagicMock() mock_resp = MagicMock() mock_resp.json = CoroutineMock() mock_resp_gen = CoroutineMock(return_value=mock_resp) request_ctx_mgr = MagicMock() request_ctx_mgr.__aenter__ = mock_resp_gen request_ctx_mgr.__aexit__ = CoroutineMock() sess.get = Mock(return_value=request_ctx_mgr) sess.post = Mock(return_value=request_ctx_mgr) return sess
def redis_connection(): conn = MagicMock() conn.__enter__ = MagicMock(return_value=conn) conn.__exit__ = MagicMock() conn.get = CoroutineMock() conn.mget = CoroutineMock() conn.set = CoroutineMock() conn.setex = CoroutineMock() conn.mset = CoroutineMock() conn.incrby = CoroutineMock() conn.exists = CoroutineMock() conn.persist = CoroutineMock() conn.expire = CoroutineMock() conn.delete = CoroutineMock() conn.flushdb = CoroutineMock() conn.eval = CoroutineMock() conn.keys = CoroutineMock() conn.multi_exec = MagicMock(return_value=conn) conn.execute = CoroutineMock() return conn