conftest.py

#
#  Copyright 2022 Il`ya (Marshal) <https://github.com/MarshalX>. All rights reserved.
#
#  Use of this source code is governed by a BSD-style license
#  that can be found in the LICENSE.md file in the root of the project.
#

import pytest

import webrtc


@pytest.fixture
def rtc_peer_connection(request):
    pc = webrtc.RTCPeerConnection()

    def close_pc():
        pc.close()

    request.addfinalizer(close_pc)

    return pc


# aliases
pc = caller = callee = rtc_peer_connection


def get_stream(constraints, request):
    # TODO pass constraints when cpp part will be ready
    stream = webrtc.get_user_media()

    def stop_tracks():
        for track in stream.get_tracks():
            track.stop()

    request.addfinalizer(stop_tracks)

    return stream


@pytest.fixture
def audio_stream(request):
    return get_stream(None, request)


# alias
audio_stream2 = audio_stream


@pytest.fixture
def video_stream(request):
    return get_stream(None, request)