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
