Refine utest mock.
This commit is contained in:
winlin
parent
c9fe296342
commit
3f706f9c37
2
trunk/configure
vendored
2
trunk/configure
vendored
|
|
@ -385,7 +385,7 @@ if [[ $SRS_UTEST == YES ]]; then
|
||||||
"srs_utest_protocol3" "srs_utest_app" "srs_utest_app2" "srs_utest_app3" "srs_utest_app4"
|
"srs_utest_protocol3" "srs_utest_app" "srs_utest_app2" "srs_utest_app3" "srs_utest_app4"
|
||||||
"srs_utest_app5" "srs_utest_app6" "srs_utest_app7" "srs_utest_app8" "srs_utest_app9"
|
"srs_utest_app5" "srs_utest_app6" "srs_utest_app7" "srs_utest_app8" "srs_utest_app9"
|
||||||
"srs_utest_app10" "srs_utest_app11" "srs_utest_app15" "srs_utest_app16"
|
"srs_utest_app10" "srs_utest_app11" "srs_utest_app15" "srs_utest_app16"
|
||||||
"srs_utest_app17" "srs_utest_rtc_playstream")
|
"srs_utest_app17" "srs_utest_mock" "srs_utest_rtc_playstream")
|
||||||
# Always include SRT utest
|
# Always include SRT utest
|
||||||
MODULE_FILES+=("srs_utest_srt")
|
MODULE_FILES+=("srs_utest_srt")
|
||||||
if [[ $SRS_GB28181 == YES ]]; then
|
if [[ $SRS_GB28181 == YES ]]; then
|
||||||
|
|
|
||||||
|
|
@ -6,6 +6,8 @@
|
||||||
|
|
||||||
#include <srs_protocol_http_stack_llhttpadapter.hpp>
|
#include <srs_protocol_http_stack_llhttpadapter.hpp>
|
||||||
|
|
||||||
|
// LCOV_EXCL_START
|
||||||
|
|
||||||
// To avoid type mismatch, these functions are used to bridge the APIs.
|
// To avoid type mismatch, these functions are used to bridge the APIs.
|
||||||
// Bridge functions for internal callbacks that call the public API implementations
|
// Bridge functions for internal callbacks that call the public API implementations
|
||||||
// These bridge the internal callback signature to the public API signature
|
// These bridge the internal callback signature to the public API signature
|
||||||
|
|
@ -199,3 +201,5 @@ int llhttp__after_message_complete(llhttp__internal_t *s, const unsigned char *p
|
||||||
{
|
{
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// LCOV_EXCL_STOP
|
||||||
|
|
|
||||||
|
|
@ -17,6 +17,8 @@ using namespace std;
|
||||||
|
|
||||||
#include <srt/srt.h>
|
#include <srt/srt.h>
|
||||||
|
|
||||||
|
// LCOV_EXCL_START
|
||||||
|
|
||||||
// TODO: FIXME: protocol could no include app's header file, so define TAG_SRT in this file.
|
// TODO: FIXME: protocol could no include app's header file, so define TAG_SRT in this file.
|
||||||
#define TAG_SRT "SRT"
|
#define TAG_SRT "SRT"
|
||||||
|
|
||||||
|
|
@ -1029,3 +1031,5 @@ srs_error_t SrsSrtSocket::check_error()
|
||||||
|
|
||||||
return err;
|
return err;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// LCOV_EXCL_STOP
|
||||||
|
|
|
||||||
|
|
@ -32,6 +32,9 @@ using namespace std;
|
||||||
#include <srs_app_listener.hpp>
|
#include <srs_app_listener.hpp>
|
||||||
#include <srs_protocol_conn.hpp>
|
#include <srs_protocol_conn.hpp>
|
||||||
|
|
||||||
|
// Include headers for all the mocks.
|
||||||
|
#include <srs_utest_mock.hpp>
|
||||||
|
|
||||||
// we add an empty macro for upp to show the smart tips.
|
// we add an empty macro for upp to show the smart tips.
|
||||||
#define VOID
|
#define VOID
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -2364,678 +2364,6 @@ void MockContext::set_current_id(const SrsContextId &id)
|
||||||
get_id_result_ = id;
|
get_id_result_ = id;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Mock app config implementation
|
|
||||||
MockAppConfig::MockAppConfig()
|
|
||||||
{
|
|
||||||
http_hooks_enabled_ = true;
|
|
||||||
on_stop_directive_ = NULL;
|
|
||||||
on_unpublish_directive_ = NULL;
|
|
||||||
rtc_nack_enabled_ = true;
|
|
||||||
rtc_nack_no_copy_ = false;
|
|
||||||
rtc_drop_for_pt_ = 0;
|
|
||||||
rtc_twcc_enabled_ = true;
|
|
||||||
srt_enabled_ = false;
|
|
||||||
rtc_to_rtmp_ = false;
|
|
||||||
dash_dispose_ = 0;
|
|
||||||
dash_enabled_ = false;
|
|
||||||
api_as_candidates_ = true;
|
|
||||||
resolve_api_domain_ = true;
|
|
||||||
keep_api_domain_ = false;
|
|
||||||
mw_msgs_ = 8;
|
|
||||||
}
|
|
||||||
|
|
||||||
MockAppConfig::~MockAppConfig()
|
|
||||||
{
|
|
||||||
clear_on_stop_directive();
|
|
||||||
clear_on_unpublish_directive();
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_utime_t MockAppConfig::get_pithy_print()
|
|
||||||
{
|
|
||||||
return 10 * SRS_UTIME_SECONDS;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_default_app_name()
|
|
||||||
{
|
|
||||||
return "live";
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::subscribe(ISrsReloadHandler *handler)
|
|
||||||
{
|
|
||||||
// Do nothing in mock
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::unsubscribe(ISrsReloadHandler *handler)
|
|
||||||
{
|
|
||||||
// Do nothing in mock
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_vhost_http_hooks_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return http_hooks_enabled_;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsConfDirective *MockAppConfig::get_vhost_on_stop(std::string vhost)
|
|
||||||
{
|
|
||||||
return on_stop_directive_;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsConfDirective *MockAppConfig::get_vhost_on_unpublish(std::string vhost)
|
|
||||||
{
|
|
||||||
return on_unpublish_directive_;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsConfDirective *MockAppConfig::get_vhost_on_dvr(std::string vhost)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_rtc_nack_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return rtc_nack_enabled_;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_rtc_nack_no_copy(std::string vhost)
|
|
||||||
{
|
|
||||||
return rtc_nack_no_copy_;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_realtime_enabled(std::string vhost, bool is_rtc)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
int MockAppConfig::get_mw_msgs(std::string vhost, bool is_realtime, bool is_rtc)
|
|
||||||
{
|
|
||||||
return mw_msgs_;
|
|
||||||
}
|
|
||||||
|
|
||||||
int MockAppConfig::get_rtc_drop_for_pt(std::string vhost)
|
|
||||||
{
|
|
||||||
return rtc_drop_for_pt_;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_rtc_twcc_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return rtc_twcc_enabled_;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_srt_enabled()
|
|
||||||
{
|
|
||||||
return srt_enabled_;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_srt_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return srt_enabled_;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_srt_default_streamid()
|
|
||||||
{
|
|
||||||
return "#!::r=live/livestream,m=request";
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_srt_to_rtmp(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_rtc_to_rtmp(std::string vhost)
|
|
||||||
{
|
|
||||||
return rtc_to_rtmp_;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_utime_t MockAppConfig::get_rtc_stun_timeout(std::string vhost)
|
|
||||||
{
|
|
||||||
return 30 * SRS_UTIME_SECONDS; // Default 30 seconds timeout
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_rtc_stun_strict_check(std::string vhost)
|
|
||||||
{
|
|
||||||
return false; // Default to non-strict mode
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_rtc_dtls_role(std::string vhost)
|
|
||||||
{
|
|
||||||
return "passive"; // Default DTLS role
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_rtc_dtls_version(std::string vhost)
|
|
||||||
{
|
|
||||||
return "auto"; // Default DTLS version
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsConfDirective *MockAppConfig::get_vhost_on_hls(std::string vhost)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsConfDirective *MockAppConfig::get_vhost_on_hls_notify(std::string vhost)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_enabled(SrsConfDirective *vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_use_fmp4(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_entry_prefix(std::string vhost)
|
|
||||||
{
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_path(std::string vhost)
|
|
||||||
{
|
|
||||||
return "./objs/nginx/html";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_m3u8_file(std::string vhost)
|
|
||||||
{
|
|
||||||
return "[app]/[stream].m3u8";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_ts_file(std::string vhost)
|
|
||||||
{
|
|
||||||
return "[app]/[stream]-[seq].ts";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_fmp4_file(std::string vhost)
|
|
||||||
{
|
|
||||||
return "[app]/[stream]-[seq].m4s";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_init_file(std::string vhost)
|
|
||||||
{
|
|
||||||
return "[app]/[stream]/init.mp4";
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_ts_floor(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_utime_t MockAppConfig::get_hls_fragment(std::string vhost)
|
|
||||||
{
|
|
||||||
return 10 * SRS_UTIME_SECONDS;
|
|
||||||
}
|
|
||||||
|
|
||||||
double MockAppConfig::get_hls_td_ratio(std::string vhost)
|
|
||||||
{
|
|
||||||
return 1.5;
|
|
||||||
}
|
|
||||||
|
|
||||||
double MockAppConfig::get_hls_aof_ratio(std::string vhost)
|
|
||||||
{
|
|
||||||
return 2.0;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_utime_t MockAppConfig::get_hls_window(std::string vhost)
|
|
||||||
{
|
|
||||||
return 60 * SRS_UTIME_SECONDS;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_on_error(std::string vhost)
|
|
||||||
{
|
|
||||||
return "continue";
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_cleanup(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_utime_t MockAppConfig::get_hls_dispose(std::string vhost)
|
|
||||||
{
|
|
||||||
return 120 * SRS_UTIME_SECONDS;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_wait_keyframe(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_keys(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
int MockAppConfig::get_hls_fragments_per_key(std::string vhost)
|
|
||||||
{
|
|
||||||
return 5;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_key_file(std::string vhost)
|
|
||||||
{
|
|
||||||
return "[app]/[stream]-[seq].key";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_key_file_path(std::string vhost)
|
|
||||||
{
|
|
||||||
return "./objs/nginx/html";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockAppConfig::get_hls_key_url(std::string vhost)
|
|
||||||
{
|
|
||||||
return "";
|
|
||||||
}
|
|
||||||
|
|
||||||
int MockAppConfig::get_vhost_hls_nb_notify(std::string vhost)
|
|
||||||
{
|
|
||||||
return 64;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_vhost_hls_dts_directly(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_ctx_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_ts_ctx_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_hls_recover(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_forward_enabled(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsConfDirective *MockAppConfig::get_forwards(std::string vhost)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_utime_t MockAppConfig::get_queue_length(std::string vhost)
|
|
||||||
{
|
|
||||||
return 30 * SRS_UTIME_SECONDS;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsConfDirective *MockAppConfig::get_forward_backend(std::string vhost)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_atc(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
int MockAppConfig::get_time_jitter(std::string vhost)
|
|
||||||
{
|
|
||||||
return SrsRtmpJitterAlgorithmFULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_mix_correct(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::try_annexb_first(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_vhost_is_edge(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_atc_auto(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_reduce_sequence_header(std::string vhost)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool MockAppConfig::get_parse_sps(std::string vhost)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_http_hooks_enabled(bool enabled)
|
|
||||||
{
|
|
||||||
http_hooks_enabled_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_on_stop_urls(const std::vector<std::string> &urls)
|
|
||||||
{
|
|
||||||
clear_on_stop_directive();
|
|
||||||
if (!urls.empty()) {
|
|
||||||
on_stop_directive_ = new SrsConfDirective();
|
|
||||||
on_stop_directive_->name_ = "on_stop";
|
|
||||||
on_stop_directive_->args_ = urls;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::clear_on_stop_directive()
|
|
||||||
{
|
|
||||||
srs_freep(on_stop_directive_);
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_on_unpublish_urls(const std::vector<std::string> &urls)
|
|
||||||
{
|
|
||||||
clear_on_unpublish_directive();
|
|
||||||
if (!urls.empty()) {
|
|
||||||
on_unpublish_directive_ = new SrsConfDirective();
|
|
||||||
on_unpublish_directive_->name_ = "on_unpublish";
|
|
||||||
on_unpublish_directive_->args_ = urls;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::clear_on_unpublish_directive()
|
|
||||||
{
|
|
||||||
srs_freep(on_unpublish_directive_);
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_rtc_nack_enabled(bool enabled)
|
|
||||||
{
|
|
||||||
rtc_nack_enabled_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_rtc_nack_no_copy(bool no_copy)
|
|
||||||
{
|
|
||||||
rtc_nack_no_copy_ = no_copy;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_rtc_drop_for_pt(int pt)
|
|
||||||
{
|
|
||||||
rtc_drop_for_pt_ = pt;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_rtc_twcc_enabled(bool enabled)
|
|
||||||
{
|
|
||||||
rtc_twcc_enabled_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_srt_enabled(bool enabled)
|
|
||||||
{
|
|
||||||
srt_enabled_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_rtc_to_rtmp(bool enabled)
|
|
||||||
{
|
|
||||||
rtc_to_rtmp_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_api_as_candidates(bool enabled)
|
|
||||||
{
|
|
||||||
api_as_candidates_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_resolve_api_domain(bool enabled)
|
|
||||||
{
|
|
||||||
resolve_api_domain_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockAppConfig::set_keep_api_domain(bool enabled)
|
|
||||||
{
|
|
||||||
keep_api_domain_ = enabled;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Mock request implementation
|
|
||||||
MockRtcAsyncCallRequest::MockRtcAsyncCallRequest(std::string vhost, std::string app, std::string stream)
|
|
||||||
{
|
|
||||||
vhost_ = vhost;
|
|
||||||
app_ = app;
|
|
||||||
stream_ = stream;
|
|
||||||
}
|
|
||||||
|
|
||||||
MockRtcAsyncCallRequest::~MockRtcAsyncCallRequest()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
ISrsRequest *MockRtcAsyncCallRequest::copy()
|
|
||||||
{
|
|
||||||
MockRtcAsyncCallRequest *cp = new MockRtcAsyncCallRequest(vhost_, app_, stream_);
|
|
||||||
return cp;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockRtcAsyncCallRequest::get_stream_url()
|
|
||||||
{
|
|
||||||
return "rtmp://" + vhost_ + "/" + app_ + "/" + stream_;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcAsyncCallRequest::update_auth(ISrsRequest *req)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcAsyncCallRequest::strip()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
ISrsRequest *MockRtcAsyncCallRequest::as_http()
|
|
||||||
{
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Mock RTC source manager implementation
|
|
||||||
MockRtcSourceManager::MockRtcSourceManager()
|
|
||||||
{
|
|
||||||
initialize_error_ = srs_success;
|
|
||||||
fetch_or_create_error_ = srs_success;
|
|
||||||
initialize_count_ = 0;
|
|
||||||
fetch_or_create_count_ = 0;
|
|
||||||
mock_source_ = SrsSharedPtr<SrsRtcSource>(new SrsRtcSource());
|
|
||||||
}
|
|
||||||
|
|
||||||
MockRtcSourceManager::~MockRtcSourceManager()
|
|
||||||
{
|
|
||||||
srs_freep(initialize_error_);
|
|
||||||
srs_freep(fetch_or_create_error_);
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcSourceManager::initialize()
|
|
||||||
{
|
|
||||||
initialize_count_++;
|
|
||||||
return srs_error_copy(initialize_error_);
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcSourceManager::fetch_or_create(ISrsRequest *r, SrsSharedPtr<SrsRtcSource> &pps)
|
|
||||||
{
|
|
||||||
fetch_or_create_count_++;
|
|
||||||
if (fetch_or_create_error_ != srs_success) {
|
|
||||||
return srs_error_copy(fetch_or_create_error_);
|
|
||||||
}
|
|
||||||
pps = mock_source_;
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsSharedPtr<SrsRtcSource> MockRtcSourceManager::fetch(ISrsRequest *r)
|
|
||||||
{
|
|
||||||
return mock_source_;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcSourceManager::set_initialize_error(srs_error_t err)
|
|
||||||
{
|
|
||||||
srs_freep(initialize_error_);
|
|
||||||
initialize_error_ = srs_error_copy(err);
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcSourceManager::set_fetch_or_create_error(srs_error_t err)
|
|
||||||
{
|
|
||||||
srs_freep(fetch_or_create_error_);
|
|
||||||
fetch_or_create_error_ = srs_error_copy(err);
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcSourceManager::reset()
|
|
||||||
{
|
|
||||||
srs_freep(initialize_error_);
|
|
||||||
srs_freep(fetch_or_create_error_);
|
|
||||||
initialize_error_ = srs_success;
|
|
||||||
fetch_or_create_error_ = srs_success;
|
|
||||||
initialize_count_ = 0;
|
|
||||||
fetch_or_create_count_ = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Mock statistic implementation
|
|
||||||
MockRtcStatistic::MockRtcStatistic()
|
|
||||||
{
|
|
||||||
on_client_error_ = srs_success;
|
|
||||||
on_client_count_ = 0;
|
|
||||||
on_disconnect_count_ = 0;
|
|
||||||
last_client_id_ = "";
|
|
||||||
last_client_req_ = NULL;
|
|
||||||
last_client_conn_ = NULL;
|
|
||||||
last_client_type_ = SrsRtmpConnUnknown;
|
|
||||||
}
|
|
||||||
|
|
||||||
MockRtcStatistic::~MockRtcStatistic()
|
|
||||||
{
|
|
||||||
srs_freep(on_client_error_);
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcStatistic::on_disconnect(std::string id, srs_error_t err)
|
|
||||||
{
|
|
||||||
on_disconnect_count_++;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::on_client(std::string id, ISrsRequest *req, ISrsExpire *conn, SrsRtmpConnType type)
|
|
||||||
{
|
|
||||||
on_client_count_++;
|
|
||||||
last_client_id_ = id;
|
|
||||||
last_client_req_ = req;
|
|
||||||
last_client_conn_ = conn;
|
|
||||||
last_client_type_ = type;
|
|
||||||
return srs_error_copy(on_client_error_);
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcStatistic::set_on_client_error(srs_error_t err)
|
|
||||||
{
|
|
||||||
srs_freep(on_client_error_);
|
|
||||||
on_client_error_ = srs_error_copy(err);
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcStatistic::reset()
|
|
||||||
{
|
|
||||||
srs_freep(on_client_error_);
|
|
||||||
on_client_error_ = srs_success;
|
|
||||||
on_client_count_ = 0;
|
|
||||||
on_disconnect_count_ = 0;
|
|
||||||
last_client_id_ = "";
|
|
||||||
last_client_req_ = NULL;
|
|
||||||
last_client_conn_ = NULL;
|
|
||||||
last_client_type_ = SrsRtmpConnUnknown;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::on_video_info(ISrsRequest *req, SrsVideoCodecId vcodec, int avc_profile, int avc_level, int width, int height)
|
|
||||||
{
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::on_audio_info(ISrsRequest *req, SrsAudioCodecId acodec, SrsAudioSampleRate asample_rate, SrsAudioChannels asound_type, SrsAacObjectType aac_object)
|
|
||||||
{
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcStatistic::on_stream_publish(ISrsRequest *req, std::string publisher_id)
|
|
||||||
{
|
|
||||||
// Do nothing in mock
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcStatistic::on_stream_close(ISrsRequest *req)
|
|
||||||
{
|
|
||||||
// Do nothing in mock
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcStatistic::kbps_add_delta(std::string id, ISrsKbpsDelta *delta)
|
|
||||||
{
|
|
||||||
// Do nothing in mock
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcStatistic::kbps_sample()
|
|
||||||
{
|
|
||||||
// Do nothing in mock
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::on_video_frames(ISrsRequest *req, int nb_frames)
|
|
||||||
{
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockRtcStatistic::server_id()
|
|
||||||
{
|
|
||||||
return "mock_server_id";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockRtcStatistic::service_id()
|
|
||||||
{
|
|
||||||
return "mock_service_id";
|
|
||||||
}
|
|
||||||
|
|
||||||
std::string MockRtcStatistic::service_pid()
|
|
||||||
{
|
|
||||||
return "mock_pid";
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsStatisticVhost *MockRtcStatistic::find_vhost_by_id(std::string vid)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsStatisticStream *MockRtcStatistic::find_stream(std::string sid)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsStatisticStream *MockRtcStatistic::find_stream_by_url(std::string url)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsStatisticClient *MockRtcStatistic::find_client(std::string client_id)
|
|
||||||
{
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::dumps_vhosts(SrsJsonArray *arr)
|
|
||||||
{
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::dumps_streams(SrsJsonArray *arr, int start, int count)
|
|
||||||
{
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::dumps_clients(SrsJsonArray *arr, int start, int count)
|
|
||||||
{
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcStatistic::dumps_metrics(int64_t &send_bytes, int64_t &recv_bytes, int64_t &nstreams, int64_t &nclients, int64_t &total_nclients, int64_t &nerrs)
|
|
||||||
{
|
|
||||||
send_bytes = 0;
|
|
||||||
recv_bytes = 0;
|
|
||||||
nstreams = 0;
|
|
||||||
nclients = 0;
|
|
||||||
total_nclients = 0;
|
|
||||||
nerrs = 0;
|
|
||||||
return srs_success;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Unit tests for SrsRtcAsyncCallOnStop::call()
|
// Unit tests for SrsRtcAsyncCallOnStop::call()
|
||||||
VOID TEST(RtcAsyncCallOnStopTest, CallWithHttpHooksDisabled)
|
VOID TEST(RtcAsyncCallOnStopTest, CallWithHttpHooksDisabled)
|
||||||
{
|
{
|
||||||
|
|
@ -3418,68 +2746,6 @@ VOID TEST(RtcPlayStreamTest, OnStreamChangeSuccess)
|
||||||
srs_freep(source_desc);
|
srs_freep(source_desc);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Mock RTC async task executor implementation
|
|
||||||
MockRtcAsyncTaskExecutor::MockRtcAsyncTaskExecutor()
|
|
||||||
{
|
|
||||||
exec_error_ = srs_success;
|
|
||||||
exec_count_ = 0;
|
|
||||||
last_task_ = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
MockRtcAsyncTaskExecutor::~MockRtcAsyncTaskExecutor()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcAsyncTaskExecutor::exec_rtc_async_work(ISrsAsyncCallTask *t)
|
|
||||||
{
|
|
||||||
exec_count_++;
|
|
||||||
last_task_ = t;
|
|
||||||
return exec_error_;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcAsyncTaskExecutor::set_exec_error(srs_error_t err)
|
|
||||||
{
|
|
||||||
exec_error_ = err;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcAsyncTaskExecutor::reset()
|
|
||||||
{
|
|
||||||
exec_error_ = srs_success;
|
|
||||||
exec_count_ = 0;
|
|
||||||
last_task_ = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Mock RTC packet sender implementation
|
|
||||||
MockRtcPacketSender::MockRtcPacketSender()
|
|
||||||
{
|
|
||||||
send_packet_error_ = srs_success;
|
|
||||||
send_packet_count_ = 0;
|
|
||||||
last_sent_packet_ = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
MockRtcPacketSender::~MockRtcPacketSender()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
srs_error_t MockRtcPacketSender::do_send_packet(SrsRtpPacket *pkt)
|
|
||||||
{
|
|
||||||
send_packet_count_++;
|
|
||||||
last_sent_packet_ = pkt;
|
|
||||||
return send_packet_error_;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcPacketSender::set_send_packet_error(srs_error_t err)
|
|
||||||
{
|
|
||||||
send_packet_error_ = err;
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockRtcPacketSender::reset()
|
|
||||||
{
|
|
||||||
send_packet_error_ = srs_success;
|
|
||||||
send_packet_count_ = 0;
|
|
||||||
last_sent_packet_ = NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Mock RTC send track implementation
|
// Mock RTC send track implementation
|
||||||
MockRtcSendTrack::MockRtcSendTrack(ISrsRtcPacketSender *sender, SrsRtcTrackDescription *track_desc, bool is_audio)
|
MockRtcSendTrack::MockRtcSendTrack(ISrsRtcPacketSender *sender, SrsRtcTrackDescription *track_desc, bool is_audio)
|
||||||
: SrsRtcSendTrack(sender, track_desc, is_audio)
|
: SrsRtcSendTrack(sender, track_desc, is_audio)
|
||||||
|
|
|
||||||
|
|
@ -308,411 +308,6 @@ public:
|
||||||
void set_current_id(const SrsContextId &id);
|
void set_current_id(const SrsContextId &id);
|
||||||
};
|
};
|
||||||
|
|
||||||
// Mock app config for testing SrsRtcAsyncCallOnStop
|
|
||||||
class MockAppConfig : public ISrsAppConfig
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
bool http_hooks_enabled_;
|
|
||||||
SrsConfDirective *on_stop_directive_;
|
|
||||||
SrsConfDirective *on_unpublish_directive_;
|
|
||||||
bool rtc_nack_enabled_;
|
|
||||||
bool rtc_nack_no_copy_;
|
|
||||||
int rtc_drop_for_pt_;
|
|
||||||
bool rtc_twcc_enabled_;
|
|
||||||
bool srt_enabled_;
|
|
||||||
bool rtc_to_rtmp_;
|
|
||||||
srs_utime_t dash_dispose_;
|
|
||||||
bool dash_enabled_;
|
|
||||||
bool api_as_candidates_;
|
|
||||||
bool resolve_api_domain_;
|
|
||||||
bool keep_api_domain_;
|
|
||||||
int mw_msgs_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
MockAppConfig();
|
|
||||||
virtual ~MockAppConfig();
|
|
||||||
// ISrsConfig methods
|
|
||||||
virtual srs_utime_t get_pithy_print();
|
|
||||||
virtual std::string get_default_app_name();
|
|
||||||
virtual void subscribe(ISrsReloadHandler *handler);
|
|
||||||
virtual void unsubscribe(ISrsReloadHandler *handler);
|
|
||||||
virtual srs_error_t reload(SrsReloadState *pstate) { return srs_success; }
|
|
||||||
virtual srs_error_t persistence() { return srs_success; }
|
|
||||||
virtual std::string config() { return ""; }
|
|
||||||
virtual SrsConfDirective *get_root() { return NULL; }
|
|
||||||
virtual std::string cwd() { return "./"; }
|
|
||||||
virtual int get_max_connections() { return 1000; }
|
|
||||||
virtual std::string get_pid_file() { return ""; }
|
|
||||||
virtual bool empty_ip_ok() { return false; }
|
|
||||||
virtual bool get_asprocess() { return false; }
|
|
||||||
virtual srs_utime_t get_grace_start_wait() { return 0; }
|
|
||||||
virtual srs_utime_t get_grace_final_wait() { return 0; }
|
|
||||||
virtual bool is_force_grace_quit() { return false; }
|
|
||||||
virtual bool inotify_auto_reload() { return false; }
|
|
||||||
virtual bool auto_reload_for_docker() { return false; }
|
|
||||||
virtual std::vector<std::string> get_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual bool get_rtmps_enabled() { return false; }
|
|
||||||
virtual std::vector<std::string> get_rtmps_listen() { return std::vector<std::string>(); }
|
|
||||||
virtual bool get_http_api_enabled() { return false; }
|
|
||||||
virtual std::vector<std::string> get_http_api_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual bool get_https_api_enabled() { return false; }
|
|
||||||
virtual std::vector<std::string> get_https_api_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual std::string get_https_api_ssl_key() { return ""; }
|
|
||||||
virtual std::string get_https_api_ssl_cert() { return ""; }
|
|
||||||
virtual bool get_raw_api() { return false; }
|
|
||||||
virtual bool get_raw_api_allow_reload() { return false; }
|
|
||||||
virtual bool get_raw_api_allow_query() { return false; }
|
|
||||||
virtual bool get_raw_api_allow_update() { return false; }
|
|
||||||
virtual bool get_http_api_auth_enabled() { return false; }
|
|
||||||
virtual std::string get_http_api_auth_username() { return ""; }
|
|
||||||
virtual std::string get_http_api_auth_password() { return ""; }
|
|
||||||
virtual srs_error_t raw_to_json(SrsJsonObject *obj) { return srs_success; }
|
|
||||||
virtual bool get_http_stream_enabled() { return false; }
|
|
||||||
virtual std::vector<std::string> get_http_stream_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual bool get_https_stream_enabled() { return false; }
|
|
||||||
virtual std::vector<std::string> get_https_stream_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual std::string get_https_stream_ssl_key() { return ""; }
|
|
||||||
virtual std::string get_https_stream_ssl_cert() { return ""; }
|
|
||||||
virtual std::string get_http_stream_dir() { return ""; }
|
|
||||||
virtual bool get_http_stream_crossdomain() { return false; }
|
|
||||||
virtual bool get_rtc_server_enabled() { return false; }
|
|
||||||
virtual bool get_rtc_server_tcp_enabled() { return false; }
|
|
||||||
virtual std::vector<std::string> get_rtc_server_tcp_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual std::string get_rtc_server_protocol() { return "udp"; }
|
|
||||||
virtual std::vector<std::string> get_rtc_server_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual int get_rtc_server_reuseport() { return 1; }
|
|
||||||
virtual bool get_rtc_server_encrypt() { return false; }
|
|
||||||
virtual bool get_api_as_candidates() { return api_as_candidates_; }
|
|
||||||
virtual bool get_resolve_api_domain() { return resolve_api_domain_; }
|
|
||||||
virtual bool get_keep_api_domain() { return keep_api_domain_; }
|
|
||||||
virtual std::string get_rtc_server_candidates() { return "*"; }
|
|
||||||
virtual bool get_use_auto_detect_network_ip() { return true; }
|
|
||||||
virtual std::string get_rtc_server_ip_family() { return "ipv4"; }
|
|
||||||
virtual bool get_rtsp_server_enabled() { return false; }
|
|
||||||
virtual std::vector<std::string> get_rtsp_server_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual std::vector<std::string> get_srt_listens() { return std::vector<std::string>(); }
|
|
||||||
virtual std::vector<SrsConfDirective *> get_stream_casters() { return std::vector<SrsConfDirective *>(); }
|
|
||||||
virtual bool get_stream_caster_enabled(SrsConfDirective *conf) { return false; }
|
|
||||||
virtual std::string get_stream_caster_engine(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::string get_stream_caster_output(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual int get_stream_caster_listen(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual bool get_exporter_enabled() { return false; }
|
|
||||||
virtual std::string get_exporter_listen() { return ""; }
|
|
||||||
virtual std::string get_exporter_label() { return ""; }
|
|
||||||
virtual std::string get_exporter_tag() { return ""; }
|
|
||||||
virtual bool get_stats_enabled() { return false; }
|
|
||||||
virtual int get_stats_network() { return 0; }
|
|
||||||
virtual bool get_heartbeat_enabled() { return false; }
|
|
||||||
virtual srs_utime_t get_heartbeat_interval() { return 0; }
|
|
||||||
virtual std::string get_heartbeat_url() { return ""; }
|
|
||||||
virtual std::string get_heartbeat_device_id() { return ""; }
|
|
||||||
virtual bool get_heartbeat_summaries() { return false; }
|
|
||||||
virtual bool get_heartbeat_ports() { return false; }
|
|
||||||
virtual bool get_circuit_breaker() { return false; }
|
|
||||||
virtual int get_high_threshold() { return 0; }
|
|
||||||
virtual int get_high_pulse() { return 0; }
|
|
||||||
virtual int get_critical_threshold() { return 0; }
|
|
||||||
virtual int get_critical_pulse() { return 0; }
|
|
||||||
virtual int get_dying_threshold() { return 0; }
|
|
||||||
virtual int get_dying_pulse() { return 0; }
|
|
||||||
virtual std::string get_rtmps_ssl_cert() { return ""; }
|
|
||||||
virtual std::string get_rtmps_ssl_key() { return ""; }
|
|
||||||
virtual SrsConfDirective *get_vhost(std::string vhost, bool try_default_vhost = true) { return NULL; }
|
|
||||||
virtual bool get_vhost_enabled(std::string vhost) { return true; }
|
|
||||||
virtual bool get_debug_srs_upnode(std::string vhost) { return true; }
|
|
||||||
virtual int get_out_ack_size(std::string vhost) { return 2500000; }
|
|
||||||
virtual int get_in_ack_size(std::string vhost) { return 2500000; }
|
|
||||||
virtual int get_chunk_size(std::string vhost) { return 60000; }
|
|
||||||
virtual bool get_gop_cache(std::string vhost) { return true; }
|
|
||||||
virtual int get_gop_cache_max_frames(std::string vhost) { return 2500; }
|
|
||||||
virtual bool get_tcp_nodelay(std::string vhost) { return false; }
|
|
||||||
virtual srs_utime_t get_mw_sleep(std::string vhost, bool is_rtc = false) { return 350 * SRS_UTIME_MILLISECONDS; }
|
|
||||||
virtual srs_utime_t get_send_min_interval(std::string vhost) { return 0; }
|
|
||||||
virtual bool get_mr_enabled(std::string vhost) { return false; }
|
|
||||||
virtual srs_utime_t get_mr_sleep(std::string vhost) { return 350 * SRS_UTIME_MILLISECONDS; }
|
|
||||||
virtual srs_utime_t get_publish_1stpkt_timeout(std::string vhost) { return 20000 * SRS_UTIME_MILLISECONDS; }
|
|
||||||
virtual srs_utime_t get_publish_normal_timeout(std::string vhost) { return 5000 * SRS_UTIME_MILLISECONDS; }
|
|
||||||
virtual srs_utime_t get_publish_kickoff_for_idle(std::string vhost) { return 0; }
|
|
||||||
virtual bool get_refer_enabled(std::string vhost) { return false; }
|
|
||||||
virtual SrsConfDirective *get_refer_all(std::string vhost) { return NULL; }
|
|
||||||
virtual SrsConfDirective *get_refer_play(std::string vhost) { return NULL; }
|
|
||||||
virtual SrsConfDirective *get_refer_publish(std::string vhost) { return NULL; }
|
|
||||||
virtual bool get_vhost_origin_cluster(std::string vhost) { return false; }
|
|
||||||
virtual std::vector<std::string> get_vhost_coworkers(std::string vhost) { return std::vector<std::string>(); }
|
|
||||||
virtual bool get_vhost_edge_token_traverse(std::string vhost) { return false; }
|
|
||||||
virtual SrsConfDirective *get_vhost_edge_origin(std::string vhost) { return NULL; }
|
|
||||||
virtual SrsConfDirective *get_vhost_on_connect(std::string vhost) { return NULL; }
|
|
||||||
virtual SrsConfDirective *get_vhost_on_close(std::string vhost) { return NULL; }
|
|
||||||
virtual SrsConfDirective *get_vhost_on_publish(std::string vhost) { return NULL; }
|
|
||||||
virtual SrsConfDirective *get_vhost_on_play(std::string vhost) { return NULL; }
|
|
||||||
virtual bool get_rtc_enabled(std::string vhost) { return false; }
|
|
||||||
virtual bool get_rtsp_enabled(std::string vhost) { return false; }
|
|
||||||
virtual bool get_rtc_from_rtmp(std::string vhost) { return false; }
|
|
||||||
virtual bool get_rtsp_from_rtmp(std::string vhost) { return false; }
|
|
||||||
// ISrsAppConfig methods
|
|
||||||
virtual bool get_vhost_http_hooks_enabled(std::string vhost);
|
|
||||||
virtual SrsConfDirective *get_vhost_on_stop(std::string vhost);
|
|
||||||
virtual SrsConfDirective *get_vhost_on_unpublish(std::string vhost);
|
|
||||||
virtual SrsConfDirective *get_vhost_on_dvr(std::string vhost);
|
|
||||||
virtual bool get_rtc_nack_enabled(std::string vhost);
|
|
||||||
virtual bool get_rtc_nack_no_copy(std::string vhost);
|
|
||||||
virtual bool get_realtime_enabled(std::string vhost, bool is_rtc);
|
|
||||||
virtual int get_mw_msgs(std::string vhost, bool is_realtime, bool is_rtc);
|
|
||||||
virtual int get_rtc_drop_for_pt(std::string vhost);
|
|
||||||
virtual bool get_rtc_twcc_enabled(std::string vhost);
|
|
||||||
virtual bool get_srt_enabled();
|
|
||||||
virtual bool get_srt_enabled(std::string vhost);
|
|
||||||
virtual std::string get_srt_default_streamid();
|
|
||||||
virtual bool get_srt_to_rtmp(std::string vhost);
|
|
||||||
virtual bool get_rtc_to_rtmp(std::string vhost);
|
|
||||||
virtual srs_utime_t get_rtc_stun_timeout(std::string vhost);
|
|
||||||
virtual bool get_rtc_stun_strict_check(std::string vhost);
|
|
||||||
virtual std::string get_rtc_dtls_role(std::string vhost);
|
|
||||||
virtual std::string get_rtc_dtls_version(std::string vhost);
|
|
||||||
virtual SrsConfDirective *get_vhost_on_hls(std::string vhost);
|
|
||||||
virtual SrsConfDirective *get_vhost_on_hls_notify(std::string vhost);
|
|
||||||
// HLS methods
|
|
||||||
virtual bool get_hls_enabled(std::string vhost);
|
|
||||||
virtual bool get_hls_enabled(SrsConfDirective *vhost);
|
|
||||||
virtual bool get_hls_use_fmp4(std::string vhost);
|
|
||||||
virtual std::string get_hls_entry_prefix(std::string vhost);
|
|
||||||
virtual std::string get_hls_path(std::string vhost);
|
|
||||||
virtual std::string get_hls_m3u8_file(std::string vhost);
|
|
||||||
virtual std::string get_hls_ts_file(std::string vhost);
|
|
||||||
virtual std::string get_hls_fmp4_file(std::string vhost);
|
|
||||||
virtual std::string get_hls_init_file(std::string vhost);
|
|
||||||
virtual bool get_hls_ts_floor(std::string vhost);
|
|
||||||
virtual srs_utime_t get_hls_fragment(std::string vhost);
|
|
||||||
virtual double get_hls_td_ratio(std::string vhost);
|
|
||||||
virtual double get_hls_aof_ratio(std::string vhost);
|
|
||||||
virtual srs_utime_t get_hls_window(std::string vhost);
|
|
||||||
virtual std::string get_hls_on_error(std::string vhost);
|
|
||||||
virtual bool get_hls_cleanup(std::string vhost);
|
|
||||||
virtual srs_utime_t get_hls_dispose(std::string vhost);
|
|
||||||
virtual bool get_hls_wait_keyframe(std::string vhost);
|
|
||||||
virtual bool get_hls_keys(std::string vhost);
|
|
||||||
virtual int get_hls_fragments_per_key(std::string vhost);
|
|
||||||
virtual std::string get_hls_key_file(std::string vhost);
|
|
||||||
virtual std::string get_hls_key_file_path(std::string vhost);
|
|
||||||
virtual std::string get_hls_key_url(std::string vhost);
|
|
||||||
virtual int get_vhost_hls_nb_notify(std::string vhost);
|
|
||||||
virtual bool get_vhost_hls_dts_directly(std::string vhost);
|
|
||||||
virtual bool get_hls_ctx_enabled(std::string vhost);
|
|
||||||
virtual bool get_hls_ts_ctx_enabled(std::string vhost);
|
|
||||||
virtual bool get_hls_recover(std::string vhost);
|
|
||||||
virtual bool get_forward_enabled(std::string vhost);
|
|
||||||
virtual SrsConfDirective *get_forwards(std::string vhost);
|
|
||||||
virtual srs_utime_t get_queue_length(std::string vhost);
|
|
||||||
virtual SrsConfDirective *get_forward_backend(std::string vhost);
|
|
||||||
virtual bool get_atc(std::string vhost);
|
|
||||||
virtual int get_time_jitter(std::string vhost);
|
|
||||||
virtual bool get_mix_correct(std::string vhost);
|
|
||||||
virtual bool try_annexb_first(std::string vhost);
|
|
||||||
virtual bool get_vhost_is_edge(std::string vhost);
|
|
||||||
virtual bool get_atc_auto(std::string vhost);
|
|
||||||
virtual bool get_reduce_sequence_header(std::string vhost);
|
|
||||||
virtual bool get_parse_sps(std::string vhost);
|
|
||||||
// DVR methods
|
|
||||||
virtual std::string get_dvr_path(std::string vhost) { return "./[vhost]/[app]/[stream]/[2006]/[01]/[02]/[15].[04].[05].[999].flv"; }
|
|
||||||
virtual std::string get_dvr_plan(std::string vhost) { return "session"; }
|
|
||||||
virtual bool get_dvr_enabled(std::string vhost) { return false; }
|
|
||||||
virtual SrsConfDirective *get_dvr_apply(std::string vhost) { return NULL; }
|
|
||||||
virtual srs_utime_t get_dvr_duration(std::string vhost) { return 30 * SRS_UTIME_SECONDS; }
|
|
||||||
virtual int get_dvr_time_jitter(std::string vhost) { return 0; }
|
|
||||||
virtual bool get_dvr_wait_keyframe(std::string vhost) { return true; }
|
|
||||||
virtual bool get_vhost_enabled(SrsConfDirective *conf) { return true; }
|
|
||||||
virtual bool get_vhost_http_remux_enabled(std::string vhost) { return false; }
|
|
||||||
virtual bool get_vhost_http_remux_enabled(SrsConfDirective *vhost) { return false; }
|
|
||||||
virtual srs_utime_t get_vhost_http_remux_fast_cache(std::string vhost) { return 0; }
|
|
||||||
virtual bool get_vhost_http_remux_drop_if_not_match(std::string vhost) { return false; }
|
|
||||||
virtual bool get_vhost_http_remux_has_audio(std::string vhost) { return true; }
|
|
||||||
virtual bool get_vhost_http_remux_has_video(std::string vhost) { return true; }
|
|
||||||
virtual bool get_vhost_http_remux_guess_has_av(std::string vhost) { return true; }
|
|
||||||
virtual std::string get_vhost_http_remux_mount(std::string vhost) { return ""; }
|
|
||||||
virtual std::string get_vhost_edge_protocol(std::string vhost) { return "rtmp"; }
|
|
||||||
virtual bool get_vhost_edge_follow_client(std::string vhost) { return false; }
|
|
||||||
virtual std::string get_vhost_edge_transform_vhost(std::string vhost) { return ""; }
|
|
||||||
// DASH methods
|
|
||||||
virtual bool get_dash_enabled(std::string vhost) { return dash_enabled_; }
|
|
||||||
virtual bool get_dash_enabled(SrsConfDirective *vhost) { return dash_enabled_; }
|
|
||||||
virtual srs_utime_t get_dash_fragment(std::string vhost) { return 30 * SRS_UTIME_SECONDS; }
|
|
||||||
virtual srs_utime_t get_dash_update_period(std::string vhost) { return 30 * SRS_UTIME_SECONDS; }
|
|
||||||
virtual srs_utime_t get_dash_timeshift(std::string vhost) { return 300 * SRS_UTIME_SECONDS; }
|
|
||||||
virtual std::string get_dash_path(std::string vhost) { return "./[vhost]/[app]/[stream]/"; }
|
|
||||||
virtual std::string get_dash_mpd_file(std::string vhost) { return "[stream].mpd"; }
|
|
||||||
virtual int get_dash_window_size(std::string vhost) { return 10; }
|
|
||||||
virtual bool get_dash_cleanup(std::string vhost) { return true; }
|
|
||||||
virtual srs_utime_t get_dash_dispose(std::string vhost) { return dash_dispose_; }
|
|
||||||
// Exec config
|
|
||||||
virtual bool get_exec_enabled(std::string vhost) { return false; }
|
|
||||||
virtual std::vector<SrsConfDirective *> get_exec_publishs(std::string vhost) { return std::vector<SrsConfDirective *>(); }
|
|
||||||
// Ingest config
|
|
||||||
virtual void get_vhosts(std::vector<SrsConfDirective *> &vhosts) {}
|
|
||||||
virtual std::vector<SrsConfDirective *> get_ingesters(std::string vhost) { return std::vector<SrsConfDirective *>(); }
|
|
||||||
virtual SrsConfDirective *get_ingest_by_id(std::string vhost, std::string ingest_id) { return NULL; }
|
|
||||||
virtual bool get_ingest_enabled(SrsConfDirective *conf) { return false; }
|
|
||||||
virtual std::string get_ingest_ffmpeg(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::string get_ingest_input_type(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::string get_ingest_input_url(SrsConfDirective *conf) { return ""; }
|
|
||||||
// FFmpeg log config
|
|
||||||
virtual bool get_ff_log_enabled() { return false; }
|
|
||||||
virtual std::string get_ff_log_dir() { return ""; }
|
|
||||||
virtual std::string get_ff_log_level() { return ""; }
|
|
||||||
// Transcode/Engine config
|
|
||||||
virtual SrsConfDirective *get_transcode(std::string vhost, std::string scope) { return NULL; }
|
|
||||||
virtual bool get_transcode_enabled(SrsConfDirective *conf) { return false; }
|
|
||||||
virtual std::string get_transcode_ffmpeg(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::vector<SrsConfDirective *> get_transcode_engines(SrsConfDirective *conf) { return std::vector<SrsConfDirective *>(); }
|
|
||||||
virtual bool get_engine_enabled(SrsConfDirective *conf) { return false; }
|
|
||||||
virtual std::vector<std::string> get_engine_perfile(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
|
||||||
virtual std::string get_engine_iformat(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::vector<std::string> get_engine_vfilter(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
|
||||||
virtual std::string get_engine_vcodec(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual int get_engine_vbitrate(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual double get_engine_vfps(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual int get_engine_vwidth(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual int get_engine_vheight(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual int get_engine_vthreads(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual std::string get_engine_vprofile(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::string get_engine_vpreset(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::vector<std::string> get_engine_vparams(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
|
||||||
virtual std::string get_engine_acodec(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual int get_engine_abitrate(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual int get_engine_asample_rate(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual int get_engine_achannels(SrsConfDirective *conf) { return 0; }
|
|
||||||
virtual std::vector<std::string> get_engine_aparams(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
|
||||||
virtual std::string get_engine_oformat(SrsConfDirective *conf) { return ""; }
|
|
||||||
virtual std::string get_engine_output(SrsConfDirective *conf) { return ""; }
|
|
||||||
void set_http_hooks_enabled(bool enabled);
|
|
||||||
void set_on_stop_urls(const std::vector<std::string> &urls);
|
|
||||||
void clear_on_stop_directive();
|
|
||||||
void set_on_unpublish_urls(const std::vector<std::string> &urls);
|
|
||||||
void clear_on_unpublish_directive();
|
|
||||||
void set_rtc_nack_enabled(bool enabled);
|
|
||||||
void set_rtc_nack_no_copy(bool no_copy);
|
|
||||||
void set_rtc_drop_for_pt(int pt);
|
|
||||||
void set_rtc_twcc_enabled(bool enabled);
|
|
||||||
void set_srt_enabled(bool enabled);
|
|
||||||
void set_rtc_to_rtmp(bool enabled);
|
|
||||||
void set_api_as_candidates(bool enabled);
|
|
||||||
void set_resolve_api_domain(bool enabled);
|
|
||||||
void set_keep_api_domain(bool enabled);
|
|
||||||
};
|
|
||||||
|
|
||||||
// Mock request for testing SrsRtcAsyncCallOnStop
|
|
||||||
class MockRtcAsyncCallRequest : public ISrsRequest
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
std::string vhost_;
|
|
||||||
std::string app_;
|
|
||||||
std::string stream_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
MockRtcAsyncCallRequest(std::string vhost = "__defaultVhost__", std::string app = "live", std::string stream = "test");
|
|
||||||
virtual ~MockRtcAsyncCallRequest();
|
|
||||||
virtual ISrsRequest *copy();
|
|
||||||
virtual std::string get_stream_url();
|
|
||||||
virtual void update_auth(ISrsRequest *req);
|
|
||||||
virtual void strip();
|
|
||||||
virtual ISrsRequest *as_http();
|
|
||||||
};
|
|
||||||
|
|
||||||
// Mock RTC source manager for testing SrsRtcPlayStream
|
|
||||||
class MockRtcSourceManager : public ISrsRtcSourceManager
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
srs_error_t initialize_error_;
|
|
||||||
srs_error_t fetch_or_create_error_;
|
|
||||||
int initialize_count_;
|
|
||||||
int fetch_or_create_count_;
|
|
||||||
SrsSharedPtr<SrsRtcSource> mock_source_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
MockRtcSourceManager();
|
|
||||||
virtual ~MockRtcSourceManager();
|
|
||||||
virtual srs_error_t initialize();
|
|
||||||
virtual srs_error_t fetch_or_create(ISrsRequest *r, SrsSharedPtr<SrsRtcSource> &pps);
|
|
||||||
virtual SrsSharedPtr<SrsRtcSource> fetch(ISrsRequest *r);
|
|
||||||
void set_initialize_error(srs_error_t err);
|
|
||||||
void set_fetch_or_create_error(srs_error_t err);
|
|
||||||
void reset();
|
|
||||||
};
|
|
||||||
|
|
||||||
// Mock statistic for testing SrsRtcPlayStream
|
|
||||||
class MockRtcStatistic : public ISrsStatistic
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
srs_error_t on_client_error_;
|
|
||||||
int on_client_count_;
|
|
||||||
int on_disconnect_count_;
|
|
||||||
std::string last_client_id_;
|
|
||||||
ISrsRequest *last_client_req_;
|
|
||||||
ISrsExpire *last_client_conn_;
|
|
||||||
SrsRtmpConnType last_client_type_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
MockRtcStatistic();
|
|
||||||
virtual ~MockRtcStatistic();
|
|
||||||
virtual void on_disconnect(std::string id, srs_error_t err);
|
|
||||||
virtual srs_error_t on_client(std::string id, ISrsRequest *req, ISrsExpire *conn, SrsRtmpConnType type);
|
|
||||||
virtual srs_error_t on_video_info(ISrsRequest *req, SrsVideoCodecId vcodec, int avc_profile, int avc_level, int width, int height);
|
|
||||||
virtual srs_error_t on_audio_info(ISrsRequest *req, SrsAudioCodecId acodec, SrsAudioSampleRate asample_rate, SrsAudioChannels asound_type, SrsAacObjectType aac_object);
|
|
||||||
virtual void on_stream_publish(ISrsRequest *req, std::string publisher_id);
|
|
||||||
virtual void on_stream_close(ISrsRequest *req);
|
|
||||||
virtual void kbps_add_delta(std::string id, ISrsKbpsDelta *delta);
|
|
||||||
virtual void kbps_sample();
|
|
||||||
virtual srs_error_t on_video_frames(ISrsRequest *req, int nb_frames);
|
|
||||||
virtual std::string server_id();
|
|
||||||
virtual std::string service_id();
|
|
||||||
virtual std::string service_pid();
|
|
||||||
virtual SrsStatisticVhost *find_vhost_by_id(std::string vid);
|
|
||||||
virtual SrsStatisticStream *find_stream(std::string sid);
|
|
||||||
virtual SrsStatisticStream *find_stream_by_url(std::string url);
|
|
||||||
virtual SrsStatisticClient *find_client(std::string client_id);
|
|
||||||
virtual srs_error_t dumps_vhosts(SrsJsonArray *arr);
|
|
||||||
virtual srs_error_t dumps_streams(SrsJsonArray *arr, int start, int count);
|
|
||||||
virtual srs_error_t dumps_clients(SrsJsonArray *arr, int start, int count);
|
|
||||||
virtual srs_error_t dumps_metrics(int64_t &send_bytes, int64_t &recv_bytes, int64_t &nstreams, int64_t &nclients, int64_t &total_nclients, int64_t &nerrs);
|
|
||||||
void set_on_client_error(srs_error_t err);
|
|
||||||
void reset();
|
|
||||||
};
|
|
||||||
|
|
||||||
// Mock RTC async task executor for testing SrsRtcPlayStream::send_packet
|
|
||||||
class MockRtcAsyncTaskExecutor : public ISrsExecRtcAsyncTask
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
srs_error_t exec_error_;
|
|
||||||
int exec_count_;
|
|
||||||
ISrsAsyncCallTask *last_task_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
MockRtcAsyncTaskExecutor();
|
|
||||||
virtual ~MockRtcAsyncTaskExecutor();
|
|
||||||
|
|
||||||
public:
|
|
||||||
virtual srs_error_t exec_rtc_async_work(ISrsAsyncCallTask *t);
|
|
||||||
void set_exec_error(srs_error_t err);
|
|
||||||
void reset();
|
|
||||||
};
|
|
||||||
|
|
||||||
// Mock RTC packet sender for testing SrsRtcPlayStream::send_packet
|
|
||||||
class MockRtcPacketSender : public ISrsRtcPacketSender
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
srs_error_t send_packet_error_;
|
|
||||||
int send_packet_count_;
|
|
||||||
SrsRtpPacket *last_sent_packet_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
MockRtcPacketSender();
|
|
||||||
virtual ~MockRtcPacketSender();
|
|
||||||
|
|
||||||
public:
|
|
||||||
virtual srs_error_t do_send_packet(SrsRtpPacket *pkt);
|
|
||||||
void set_send_packet_error(srs_error_t err);
|
|
||||||
void reset();
|
|
||||||
};
|
|
||||||
|
|
||||||
// Mock RTC send track for testing SrsRtcPlayStream::send_packet
|
// Mock RTC send track for testing SrsRtcPlayStream::send_packet
|
||||||
class MockRtcSendTrack : public SrsRtcSendTrack
|
class MockRtcSendTrack : public SrsRtcSendTrack
|
||||||
{
|
{
|
||||||
|
|
|
||||||
808
trunk/src/utest/srs_utest_mock.cpp
Normal file
808
trunk/src/utest/srs_utest_mock.cpp
Normal file
|
|
@ -0,0 +1,808 @@
|
||||||
|
//
|
||||||
|
// Copyright (c) 2013-2025 The SRS Authors
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: MIT
|
||||||
|
//
|
||||||
|
|
||||||
|
#include <srs_utest_mock.hpp>
|
||||||
|
|
||||||
|
#include <srs_app_rtc_conn.hpp>
|
||||||
|
#include <srs_app_rtc_source.hpp>
|
||||||
|
#include <srs_kernel_error.hpp>
|
||||||
|
#include <srs_app_rtmp_source.hpp>
|
||||||
|
|
||||||
|
// MockRtcTrackDescriptionFactory implementation
|
||||||
|
MockRtcTrackDescriptionFactory::MockRtcTrackDescriptionFactory()
|
||||||
|
{
|
||||||
|
audio_ssrc_ = 12345;
|
||||||
|
video_ssrc_ = 67890;
|
||||||
|
screen_ssrc_ = 98765;
|
||||||
|
}
|
||||||
|
|
||||||
|
MockRtcTrackDescriptionFactory::~MockRtcTrackDescriptionFactory()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
std::map<uint32_t, SrsRtcTrackDescription *> MockRtcTrackDescriptionFactory::create_audio_video_tracks()
|
||||||
|
{
|
||||||
|
std::map<uint32_t, SrsRtcTrackDescription *> sub_relations;
|
||||||
|
|
||||||
|
// Create audio track
|
||||||
|
SrsRtcTrackDescription *audio_desc = create_audio_track(audio_ssrc_, "audio-track-1", "0");
|
||||||
|
sub_relations[audio_desc->ssrc_] = audio_desc;
|
||||||
|
|
||||||
|
// Create video track
|
||||||
|
SrsRtcTrackDescription *video_desc = create_video_track(video_ssrc_, "video-track-1", "1");
|
||||||
|
sub_relations[video_desc->ssrc_] = video_desc;
|
||||||
|
|
||||||
|
// Create screen share track
|
||||||
|
SrsRtcTrackDescription *screen_desc = create_video_track(screen_ssrc_, "screen-track-1", "2");
|
||||||
|
sub_relations[screen_desc->ssrc_] = screen_desc;
|
||||||
|
|
||||||
|
return sub_relations;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsRtcTrackDescription *MockRtcTrackDescriptionFactory::create_audio_track(uint32_t ssrc, std::string id, std::string mid)
|
||||||
|
{
|
||||||
|
SrsRtcTrackDescription *audio_desc = new SrsRtcTrackDescription();
|
||||||
|
audio_desc->type_ = "audio";
|
||||||
|
audio_desc->ssrc_ = ssrc;
|
||||||
|
audio_desc->id_ = id;
|
||||||
|
audio_desc->is_active_ = true;
|
||||||
|
audio_desc->direction_ = "sendrecv";
|
||||||
|
audio_desc->mid_ = mid;
|
||||||
|
audio_desc->media_ = new SrsAudioPayload(111, "opus", 48000, 2);
|
||||||
|
return audio_desc;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsRtcTrackDescription *MockRtcTrackDescriptionFactory::create_video_track(uint32_t ssrc, std::string id, std::string mid)
|
||||||
|
{
|
||||||
|
SrsRtcTrackDescription *video_desc = new SrsRtcTrackDescription();
|
||||||
|
video_desc->type_ = "video";
|
||||||
|
video_desc->ssrc_ = ssrc;
|
||||||
|
video_desc->id_ = id;
|
||||||
|
video_desc->is_active_ = true;
|
||||||
|
video_desc->direction_ = "sendrecv";
|
||||||
|
video_desc->mid_ = mid;
|
||||||
|
video_desc->media_ = new SrsVideoPayload(96, "H264", 90000);
|
||||||
|
return video_desc;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MockExpire implementation
|
||||||
|
MockExpire::MockExpire() : expired_(false)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
MockExpire::~MockExpire()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockExpire::expire()
|
||||||
|
{
|
||||||
|
expired_ = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MockRtcAsyncCallRequest implementation
|
||||||
|
MockRtcAsyncCallRequest::MockRtcAsyncCallRequest(std::string vhost, std::string app, std::string stream)
|
||||||
|
{
|
||||||
|
vhost_ = vhost;
|
||||||
|
app_ = app;
|
||||||
|
stream_ = stream;
|
||||||
|
}
|
||||||
|
|
||||||
|
MockRtcAsyncCallRequest::~MockRtcAsyncCallRequest()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
ISrsRequest *MockRtcAsyncCallRequest::copy()
|
||||||
|
{
|
||||||
|
MockRtcAsyncCallRequest *cp = new MockRtcAsyncCallRequest(vhost_, app_, stream_);
|
||||||
|
return cp;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockRtcAsyncCallRequest::get_stream_url()
|
||||||
|
{
|
||||||
|
return "rtmp://" + vhost_ + "/" + app_ + "/" + stream_;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcAsyncCallRequest::update_auth(ISrsRequest *req)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcAsyncCallRequest::strip()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
ISrsRequest *MockRtcAsyncCallRequest::as_http()
|
||||||
|
{
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MockRtcSourceManager implementation
|
||||||
|
MockRtcSourceManager::MockRtcSourceManager()
|
||||||
|
{
|
||||||
|
initialize_error_ = srs_success;
|
||||||
|
fetch_or_create_error_ = srs_success;
|
||||||
|
initialize_count_ = 0;
|
||||||
|
fetch_or_create_count_ = 0;
|
||||||
|
mock_source_ = SrsSharedPtr<SrsRtcSource>(new SrsRtcSource());
|
||||||
|
}
|
||||||
|
|
||||||
|
MockRtcSourceManager::~MockRtcSourceManager()
|
||||||
|
{
|
||||||
|
srs_freep(initialize_error_);
|
||||||
|
srs_freep(fetch_or_create_error_);
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcSourceManager::initialize()
|
||||||
|
{
|
||||||
|
initialize_count_++;
|
||||||
|
return srs_error_copy(initialize_error_);
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcSourceManager::fetch_or_create(ISrsRequest *r, SrsSharedPtr<SrsRtcSource> &pps)
|
||||||
|
{
|
||||||
|
fetch_or_create_count_++;
|
||||||
|
if (fetch_or_create_error_ != srs_success) {
|
||||||
|
return srs_error_copy(fetch_or_create_error_);
|
||||||
|
}
|
||||||
|
pps = mock_source_;
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsSharedPtr<SrsRtcSource> MockRtcSourceManager::fetch(ISrsRequest *r)
|
||||||
|
{
|
||||||
|
return mock_source_;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcSourceManager::set_initialize_error(srs_error_t err)
|
||||||
|
{
|
||||||
|
srs_freep(initialize_error_);
|
||||||
|
initialize_error_ = srs_error_copy(err);
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcSourceManager::set_fetch_or_create_error(srs_error_t err)
|
||||||
|
{
|
||||||
|
srs_freep(fetch_or_create_error_);
|
||||||
|
fetch_or_create_error_ = srs_error_copy(err);
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcSourceManager::reset()
|
||||||
|
{
|
||||||
|
srs_freep(initialize_error_);
|
||||||
|
srs_freep(fetch_or_create_error_);
|
||||||
|
initialize_error_ = srs_success;
|
||||||
|
fetch_or_create_error_ = srs_success;
|
||||||
|
initialize_count_ = 0;
|
||||||
|
fetch_or_create_count_ = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MockRtcStatistic implementation
|
||||||
|
MockRtcStatistic::MockRtcStatistic()
|
||||||
|
{
|
||||||
|
on_client_error_ = srs_success;
|
||||||
|
on_client_count_ = 0;
|
||||||
|
on_disconnect_count_ = 0;
|
||||||
|
last_client_id_ = "";
|
||||||
|
last_client_req_ = NULL;
|
||||||
|
last_client_conn_ = NULL;
|
||||||
|
last_client_type_ = SrsRtmpConnUnknown;
|
||||||
|
}
|
||||||
|
|
||||||
|
MockRtcStatistic::~MockRtcStatistic()
|
||||||
|
{
|
||||||
|
srs_freep(on_client_error_);
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcStatistic::on_disconnect(std::string id, srs_error_t err)
|
||||||
|
{
|
||||||
|
on_disconnect_count_++;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::on_client(std::string id, ISrsRequest *req, ISrsExpire *conn, SrsRtmpConnType type)
|
||||||
|
{
|
||||||
|
on_client_count_++;
|
||||||
|
last_client_id_ = id;
|
||||||
|
last_client_req_ = req;
|
||||||
|
last_client_conn_ = conn;
|
||||||
|
last_client_type_ = type;
|
||||||
|
return srs_error_copy(on_client_error_);
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::on_video_info(ISrsRequest *req, SrsVideoCodecId vcodec, int avc_profile, int avc_level, int width, int height)
|
||||||
|
{
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::on_audio_info(ISrsRequest *req, SrsAudioCodecId acodec, SrsAudioSampleRate asample_rate, SrsAudioChannels asound_type, SrsAacObjectType aac_object)
|
||||||
|
{
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcStatistic::on_stream_publish(ISrsRequest *req, std::string publisher_id)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcStatistic::on_stream_close(ISrsRequest *req)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcStatistic::kbps_add_delta(std::string id, ISrsKbpsDelta *delta)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcStatistic::kbps_sample()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::on_video_frames(ISrsRequest *req, int nb_frames)
|
||||||
|
{
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockRtcStatistic::server_id()
|
||||||
|
{
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockRtcStatistic::service_id()
|
||||||
|
{
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockRtcStatistic::service_pid()
|
||||||
|
{
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsStatisticVhost *MockRtcStatistic::find_vhost_by_id(std::string vid)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsStatisticStream *MockRtcStatistic::find_stream(std::string sid)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsStatisticStream *MockRtcStatistic::find_stream_by_url(std::string url)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsStatisticClient *MockRtcStatistic::find_client(std::string client_id)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::dumps_vhosts(SrsJsonArray *arr)
|
||||||
|
{
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::dumps_streams(SrsJsonArray *arr, int start, int count)
|
||||||
|
{
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::dumps_clients(SrsJsonArray *arr, int start, int count)
|
||||||
|
{
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcStatistic::dumps_metrics(int64_t &send_bytes, int64_t &recv_bytes, int64_t &nstreams, int64_t &nclients, int64_t &total_nclients, int64_t &nerrs)
|
||||||
|
{
|
||||||
|
return srs_success;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcStatistic::set_on_client_error(srs_error_t err)
|
||||||
|
{
|
||||||
|
srs_freep(on_client_error_);
|
||||||
|
on_client_error_ = srs_error_copy(err);
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcStatistic::reset()
|
||||||
|
{
|
||||||
|
srs_freep(on_client_error_);
|
||||||
|
on_client_error_ = srs_success;
|
||||||
|
on_client_count_ = 0;
|
||||||
|
on_disconnect_count_ = 0;
|
||||||
|
last_client_id_ = "";
|
||||||
|
last_client_req_ = NULL;
|
||||||
|
last_client_conn_ = NULL;
|
||||||
|
last_client_type_ = SrsRtmpConnUnknown;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MockRtcAsyncTaskExecutor implementation
|
||||||
|
MockRtcAsyncTaskExecutor::MockRtcAsyncTaskExecutor()
|
||||||
|
{
|
||||||
|
exec_error_ = srs_success;
|
||||||
|
exec_count_ = 0;
|
||||||
|
last_task_ = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
MockRtcAsyncTaskExecutor::~MockRtcAsyncTaskExecutor()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcAsyncTaskExecutor::exec_rtc_async_work(ISrsAsyncCallTask *t)
|
||||||
|
{
|
||||||
|
exec_count_++;
|
||||||
|
last_task_ = t;
|
||||||
|
return exec_error_;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcAsyncTaskExecutor::set_exec_error(srs_error_t err)
|
||||||
|
{
|
||||||
|
exec_error_ = err;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcAsyncTaskExecutor::reset()
|
||||||
|
{
|
||||||
|
exec_error_ = srs_success;
|
||||||
|
exec_count_ = 0;
|
||||||
|
last_task_ = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MockRtcPacketSender implementation
|
||||||
|
MockRtcPacketSender::MockRtcPacketSender()
|
||||||
|
{
|
||||||
|
send_packet_error_ = srs_success;
|
||||||
|
send_packet_count_ = 0;
|
||||||
|
last_sent_packet_ = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
MockRtcPacketSender::~MockRtcPacketSender()
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_error_t MockRtcPacketSender::do_send_packet(SrsRtpPacket *pkt)
|
||||||
|
{
|
||||||
|
send_packet_count_++;
|
||||||
|
last_sent_packet_ = pkt;
|
||||||
|
return send_packet_error_;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcPacketSender::set_send_packet_error(srs_error_t err)
|
||||||
|
{
|
||||||
|
send_packet_error_ = err;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockRtcPacketSender::reset()
|
||||||
|
{
|
||||||
|
send_packet_error_ = srs_success;
|
||||||
|
send_packet_count_ = 0;
|
||||||
|
last_sent_packet_ = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
// MockAppConfig implementation
|
||||||
|
MockAppConfig::MockAppConfig()
|
||||||
|
{
|
||||||
|
http_hooks_enabled_ = true;
|
||||||
|
on_stop_directive_ = NULL;
|
||||||
|
on_unpublish_directive_ = NULL;
|
||||||
|
rtc_nack_enabled_ = true;
|
||||||
|
rtc_nack_no_copy_ = false;
|
||||||
|
rtc_drop_for_pt_ = 0;
|
||||||
|
rtc_twcc_enabled_ = true;
|
||||||
|
srt_enabled_ = false;
|
||||||
|
rtc_to_rtmp_ = false;
|
||||||
|
dash_dispose_ = 0;
|
||||||
|
dash_enabled_ = false;
|
||||||
|
api_as_candidates_ = true;
|
||||||
|
resolve_api_domain_ = true;
|
||||||
|
keep_api_domain_ = false;
|
||||||
|
mw_msgs_ = 8;
|
||||||
|
}
|
||||||
|
|
||||||
|
MockAppConfig::~MockAppConfig()
|
||||||
|
{
|
||||||
|
clear_on_stop_directive();
|
||||||
|
clear_on_unpublish_directive();
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_utime_t MockAppConfig::get_pithy_print()
|
||||||
|
{
|
||||||
|
return 10 * SRS_UTIME_SECONDS;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_default_app_name()
|
||||||
|
{
|
||||||
|
return "live";
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::subscribe(ISrsReloadHandler *handler)
|
||||||
|
{
|
||||||
|
// Do nothing in mock
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::unsubscribe(ISrsReloadHandler *handler)
|
||||||
|
{
|
||||||
|
// Do nothing in mock
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_vhost_http_hooks_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return http_hooks_enabled_;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsConfDirective *MockAppConfig::get_vhost_on_stop(std::string vhost)
|
||||||
|
{
|
||||||
|
return on_stop_directive_;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsConfDirective *MockAppConfig::get_vhost_on_unpublish(std::string vhost)
|
||||||
|
{
|
||||||
|
return on_unpublish_directive_;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsConfDirective *MockAppConfig::get_vhost_on_dvr(std::string vhost)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_rtc_nack_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return rtc_nack_enabled_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_rtc_nack_no_copy(std::string vhost)
|
||||||
|
{
|
||||||
|
return rtc_nack_no_copy_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_realtime_enabled(std::string vhost, bool is_rtc)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
int MockAppConfig::get_mw_msgs(std::string vhost, bool is_realtime, bool is_rtc)
|
||||||
|
{
|
||||||
|
return mw_msgs_;
|
||||||
|
}
|
||||||
|
|
||||||
|
int MockAppConfig::get_rtc_drop_for_pt(std::string vhost)
|
||||||
|
{
|
||||||
|
return rtc_drop_for_pt_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_rtc_twcc_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return rtc_twcc_enabled_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_srt_enabled()
|
||||||
|
{
|
||||||
|
return srt_enabled_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_srt_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return srt_enabled_;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_srt_default_streamid()
|
||||||
|
{
|
||||||
|
return "#!::r=live/livestream,m=request";
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_srt_to_rtmp(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_rtc_to_rtmp(std::string vhost)
|
||||||
|
{
|
||||||
|
return rtc_to_rtmp_;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_utime_t MockAppConfig::get_rtc_stun_timeout(std::string vhost)
|
||||||
|
{
|
||||||
|
return 30 * SRS_UTIME_SECONDS;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_rtc_stun_strict_check(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_rtc_dtls_role(std::string vhost)
|
||||||
|
{
|
||||||
|
return "passive";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_rtc_dtls_version(std::string vhost)
|
||||||
|
{
|
||||||
|
return "auto";
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsConfDirective *MockAppConfig::get_vhost_on_hls(std::string vhost)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsConfDirective *MockAppConfig::get_vhost_on_hls_notify(std::string vhost)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_enabled(SrsConfDirective *vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_use_fmp4(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_entry_prefix(std::string vhost)
|
||||||
|
{
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_path(std::string vhost)
|
||||||
|
{
|
||||||
|
return "./objs/nginx/html";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_m3u8_file(std::string vhost)
|
||||||
|
{
|
||||||
|
return "[app]/[stream].m3u8";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_ts_file(std::string vhost)
|
||||||
|
{
|
||||||
|
return "[app]/[stream]-[seq].ts";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_fmp4_file(std::string vhost)
|
||||||
|
{
|
||||||
|
return "[app]/[stream]-[seq].m4s";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_init_file(std::string vhost)
|
||||||
|
{
|
||||||
|
return "[app]/[stream]-init.mp4";
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_ts_floor(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_utime_t MockAppConfig::get_hls_fragment(std::string vhost)
|
||||||
|
{
|
||||||
|
return 10 * SRS_UTIME_SECONDS;
|
||||||
|
}
|
||||||
|
|
||||||
|
double MockAppConfig::get_hls_td_ratio(std::string vhost)
|
||||||
|
{
|
||||||
|
return 1.5;
|
||||||
|
}
|
||||||
|
|
||||||
|
double MockAppConfig::get_hls_aof_ratio(std::string vhost)
|
||||||
|
{
|
||||||
|
return 2.0;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_utime_t MockAppConfig::get_hls_window(std::string vhost)
|
||||||
|
{
|
||||||
|
return 60 * SRS_UTIME_SECONDS;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_on_error(std::string vhost)
|
||||||
|
{
|
||||||
|
return "continue";
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_cleanup(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_utime_t MockAppConfig::get_hls_dispose(std::string vhost)
|
||||||
|
{
|
||||||
|
return 120 * SRS_UTIME_SECONDS;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_wait_keyframe(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_keys(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int MockAppConfig::get_hls_fragments_per_key(std::string vhost)
|
||||||
|
{
|
||||||
|
return 5;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_key_file(std::string vhost)
|
||||||
|
{
|
||||||
|
return "[app]/[stream]-[seq].key";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_key_file_path(std::string vhost)
|
||||||
|
{
|
||||||
|
return "./objs/nginx/html";
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MockAppConfig::get_hls_key_url(std::string vhost)
|
||||||
|
{
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
int MockAppConfig::get_vhost_hls_nb_notify(std::string vhost)
|
||||||
|
{
|
||||||
|
return 64;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_vhost_hls_dts_directly(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_ctx_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_ts_ctx_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_hls_recover(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_forward_enabled(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsConfDirective *MockAppConfig::get_forwards(std::string vhost)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
srs_utime_t MockAppConfig::get_queue_length(std::string vhost)
|
||||||
|
{
|
||||||
|
return 30 * SRS_UTIME_SECONDS;
|
||||||
|
}
|
||||||
|
|
||||||
|
SrsConfDirective *MockAppConfig::get_forward_backend(std::string vhost)
|
||||||
|
{
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_atc(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int MockAppConfig::get_time_jitter(std::string vhost)
|
||||||
|
{
|
||||||
|
return SrsRtmpJitterAlgorithmFULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_mix_correct(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::try_annexb_first(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_vhost_is_edge(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_atc_auto(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_reduce_sequence_header(std::string vhost)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MockAppConfig::get_parse_sps(std::string vhost)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_http_hooks_enabled(bool enabled)
|
||||||
|
{
|
||||||
|
http_hooks_enabled_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_on_stop_urls(const std::vector<std::string> &urls)
|
||||||
|
{
|
||||||
|
clear_on_stop_directive();
|
||||||
|
if (!urls.empty()) {
|
||||||
|
on_stop_directive_ = new SrsConfDirective();
|
||||||
|
on_stop_directive_->name_ = "on_stop";
|
||||||
|
on_stop_directive_->args_ = urls;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::clear_on_stop_directive()
|
||||||
|
{
|
||||||
|
srs_freep(on_stop_directive_);
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_on_unpublish_urls(const std::vector<std::string> &urls)
|
||||||
|
{
|
||||||
|
clear_on_unpublish_directive();
|
||||||
|
if (!urls.empty()) {
|
||||||
|
on_unpublish_directive_ = new SrsConfDirective();
|
||||||
|
on_unpublish_directive_->name_ = "on_unpublish";
|
||||||
|
on_unpublish_directive_->args_ = urls;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::clear_on_unpublish_directive()
|
||||||
|
{
|
||||||
|
srs_freep(on_unpublish_directive_);
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_rtc_nack_enabled(bool enabled)
|
||||||
|
{
|
||||||
|
rtc_nack_enabled_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_rtc_nack_no_copy(bool no_copy)
|
||||||
|
{
|
||||||
|
rtc_nack_no_copy_ = no_copy;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_rtc_drop_for_pt(int pt)
|
||||||
|
{
|
||||||
|
rtc_drop_for_pt_ = pt;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_rtc_twcc_enabled(bool enabled)
|
||||||
|
{
|
||||||
|
rtc_twcc_enabled_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_srt_enabled(bool enabled)
|
||||||
|
{
|
||||||
|
srt_enabled_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_rtc_to_rtmp(bool enabled)
|
||||||
|
{
|
||||||
|
rtc_to_rtmp_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_api_as_candidates(bool enabled)
|
||||||
|
{
|
||||||
|
api_as_candidates_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_resolve_api_domain(bool enabled)
|
||||||
|
{
|
||||||
|
resolve_api_domain_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
|
void MockAppConfig::set_keep_api_domain(bool enabled)
|
||||||
|
{
|
||||||
|
keep_api_domain_ = enabled;
|
||||||
|
}
|
||||||
|
|
||||||
471
trunk/src/utest/srs_utest_mock.hpp
Normal file
471
trunk/src/utest/srs_utest_mock.hpp
Normal file
|
|
@ -0,0 +1,471 @@
|
||||||
|
//
|
||||||
|
// Copyright (c) 2013-2025 The SRS Authors
|
||||||
|
//
|
||||||
|
// SPDX-License-Identifier: MIT
|
||||||
|
//
|
||||||
|
|
||||||
|
#ifndef SRS_UTEST_MOCK_HPP
|
||||||
|
#define SRS_UTEST_MOCK_HPP
|
||||||
|
|
||||||
|
#include <srs_utest.hpp>
|
||||||
|
|
||||||
|
#include <map>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
// Include necessary SRS headers for interfaces
|
||||||
|
#include <srs_app_config.hpp>
|
||||||
|
#include <srs_app_rtc_conn.hpp>
|
||||||
|
#include <srs_app_rtc_source.hpp>
|
||||||
|
#include <srs_app_statistic.hpp>
|
||||||
|
#include <srs_protocol_conn.hpp>
|
||||||
|
#include <srs_protocol_rtmp_conn.hpp>
|
||||||
|
|
||||||
|
// Forward declarations
|
||||||
|
class SrsRtcTrackDescription;
|
||||||
|
class SrsRtpPacket;
|
||||||
|
|
||||||
|
// Helper class to create mock track descriptions for testing
|
||||||
|
class MockRtcTrackDescriptionFactory
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
MockRtcTrackDescriptionFactory();
|
||||||
|
virtual ~MockRtcTrackDescriptionFactory();
|
||||||
|
|
||||||
|
public:
|
||||||
|
// Default SSRCs for audio and video tracks
|
||||||
|
uint32_t audio_ssrc_;
|
||||||
|
uint32_t video_ssrc_;
|
||||||
|
uint32_t screen_ssrc_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
// Create a map of track descriptions with audio and video tracks
|
||||||
|
std::map<uint32_t, SrsRtcTrackDescription *> create_audio_video_tracks();
|
||||||
|
|
||||||
|
// Create a single audio track description
|
||||||
|
SrsRtcTrackDescription *create_audio_track(uint32_t ssrc, std::string id, std::string mid);
|
||||||
|
|
||||||
|
// Create a single video track description
|
||||||
|
SrsRtcTrackDescription *create_video_track(uint32_t ssrc, std::string id, std::string mid);
|
||||||
|
};
|
||||||
|
|
||||||
|
// Mock expire for testing
|
||||||
|
class MockExpire : public ISrsExpire
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
bool expired_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
MockExpire();
|
||||||
|
virtual ~MockExpire();
|
||||||
|
|
||||||
|
public:
|
||||||
|
virtual void expire();
|
||||||
|
};
|
||||||
|
|
||||||
|
// Mock request for testing
|
||||||
|
class MockRtcAsyncCallRequest : public ISrsRequest
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
std::string vhost_;
|
||||||
|
std::string app_;
|
||||||
|
std::string stream_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
MockRtcAsyncCallRequest(std::string vhost = "__defaultVhost__", std::string app = "live", std::string stream = "test");
|
||||||
|
virtual ~MockRtcAsyncCallRequest();
|
||||||
|
virtual ISrsRequest *copy();
|
||||||
|
virtual std::string get_stream_url();
|
||||||
|
virtual void update_auth(ISrsRequest *req);
|
||||||
|
virtual void strip();
|
||||||
|
virtual ISrsRequest *as_http();
|
||||||
|
};
|
||||||
|
|
||||||
|
// Mock RTC source manager for testing
|
||||||
|
class MockRtcSourceManager : public ISrsRtcSourceManager
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
srs_error_t initialize_error_;
|
||||||
|
srs_error_t fetch_or_create_error_;
|
||||||
|
int initialize_count_;
|
||||||
|
int fetch_or_create_count_;
|
||||||
|
SrsSharedPtr<SrsRtcSource> mock_source_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
MockRtcSourceManager();
|
||||||
|
virtual ~MockRtcSourceManager();
|
||||||
|
virtual srs_error_t initialize();
|
||||||
|
virtual srs_error_t fetch_or_create(ISrsRequest *r, SrsSharedPtr<SrsRtcSource> &pps);
|
||||||
|
virtual SrsSharedPtr<SrsRtcSource> fetch(ISrsRequest *r);
|
||||||
|
void set_initialize_error(srs_error_t err);
|
||||||
|
void set_fetch_or_create_error(srs_error_t err);
|
||||||
|
void reset();
|
||||||
|
};
|
||||||
|
|
||||||
|
// Mock statistic for testing
|
||||||
|
class MockRtcStatistic : public ISrsStatistic
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
srs_error_t on_client_error_;
|
||||||
|
int on_client_count_;
|
||||||
|
int on_disconnect_count_;
|
||||||
|
std::string last_client_id_;
|
||||||
|
ISrsRequest *last_client_req_;
|
||||||
|
ISrsExpire *last_client_conn_;
|
||||||
|
SrsRtmpConnType last_client_type_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
MockRtcStatistic();
|
||||||
|
virtual ~MockRtcStatistic();
|
||||||
|
virtual void on_disconnect(std::string id, srs_error_t err);
|
||||||
|
virtual srs_error_t on_client(std::string id, ISrsRequest *req, ISrsExpire *conn, SrsRtmpConnType type);
|
||||||
|
virtual srs_error_t on_video_info(ISrsRequest *req, SrsVideoCodecId vcodec, int avc_profile, int avc_level, int width, int height);
|
||||||
|
virtual srs_error_t on_audio_info(ISrsRequest *req, SrsAudioCodecId acodec, SrsAudioSampleRate asample_rate, SrsAudioChannels asound_type, SrsAacObjectType aac_object);
|
||||||
|
virtual void on_stream_publish(ISrsRequest *req, std::string publisher_id);
|
||||||
|
virtual void on_stream_close(ISrsRequest *req);
|
||||||
|
virtual void kbps_add_delta(std::string id, ISrsKbpsDelta *delta);
|
||||||
|
virtual void kbps_sample();
|
||||||
|
virtual srs_error_t on_video_frames(ISrsRequest *req, int nb_frames);
|
||||||
|
virtual std::string server_id();
|
||||||
|
virtual std::string service_id();
|
||||||
|
virtual std::string service_pid();
|
||||||
|
virtual SrsStatisticVhost *find_vhost_by_id(std::string vid);
|
||||||
|
virtual SrsStatisticStream *find_stream(std::string sid);
|
||||||
|
virtual SrsStatisticStream *find_stream_by_url(std::string url);
|
||||||
|
virtual SrsStatisticClient *find_client(std::string client_id);
|
||||||
|
virtual srs_error_t dumps_vhosts(SrsJsonArray *arr);
|
||||||
|
virtual srs_error_t dumps_streams(SrsJsonArray *arr, int start, int count);
|
||||||
|
virtual srs_error_t dumps_clients(SrsJsonArray *arr, int start, int count);
|
||||||
|
virtual srs_error_t dumps_metrics(int64_t &send_bytes, int64_t &recv_bytes, int64_t &nstreams, int64_t &nclients, int64_t &total_nclients, int64_t &nerrs);
|
||||||
|
void set_on_client_error(srs_error_t err);
|
||||||
|
void reset();
|
||||||
|
};
|
||||||
|
|
||||||
|
// Mock RTC async task executor for testing
|
||||||
|
class MockRtcAsyncTaskExecutor : public ISrsExecRtcAsyncTask
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
srs_error_t exec_error_;
|
||||||
|
int exec_count_;
|
||||||
|
ISrsAsyncCallTask *last_task_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
MockRtcAsyncTaskExecutor();
|
||||||
|
virtual ~MockRtcAsyncTaskExecutor();
|
||||||
|
|
||||||
|
public:
|
||||||
|
virtual srs_error_t exec_rtc_async_work(ISrsAsyncCallTask *t);
|
||||||
|
void set_exec_error(srs_error_t err);
|
||||||
|
void reset();
|
||||||
|
};
|
||||||
|
|
||||||
|
// Mock RTC packet sender for testing
|
||||||
|
class MockRtcPacketSender : public ISrsRtcPacketSender
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
srs_error_t send_packet_error_;
|
||||||
|
int send_packet_count_;
|
||||||
|
SrsRtpPacket *last_sent_packet_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
MockRtcPacketSender();
|
||||||
|
virtual ~MockRtcPacketSender();
|
||||||
|
|
||||||
|
public:
|
||||||
|
virtual srs_error_t do_send_packet(SrsRtpPacket *pkt);
|
||||||
|
void set_send_packet_error(srs_error_t err);
|
||||||
|
void reset();
|
||||||
|
};
|
||||||
|
|
||||||
|
// Mock app config for testing
|
||||||
|
class MockAppConfig : public ISrsAppConfig
|
||||||
|
{
|
||||||
|
public:
|
||||||
|
bool http_hooks_enabled_;
|
||||||
|
SrsConfDirective *on_stop_directive_;
|
||||||
|
SrsConfDirective *on_unpublish_directive_;
|
||||||
|
bool rtc_nack_enabled_;
|
||||||
|
bool rtc_nack_no_copy_;
|
||||||
|
int rtc_drop_for_pt_;
|
||||||
|
bool rtc_twcc_enabled_;
|
||||||
|
bool srt_enabled_;
|
||||||
|
bool rtc_to_rtmp_;
|
||||||
|
srs_utime_t dash_dispose_;
|
||||||
|
bool dash_enabled_;
|
||||||
|
bool api_as_candidates_;
|
||||||
|
bool resolve_api_domain_;
|
||||||
|
bool keep_api_domain_;
|
||||||
|
int mw_msgs_;
|
||||||
|
|
||||||
|
public:
|
||||||
|
MockAppConfig();
|
||||||
|
virtual ~MockAppConfig();
|
||||||
|
// ISrsConfig methods
|
||||||
|
virtual srs_utime_t get_pithy_print();
|
||||||
|
virtual std::string get_default_app_name();
|
||||||
|
virtual void subscribe(ISrsReloadHandler *handler);
|
||||||
|
virtual void unsubscribe(ISrsReloadHandler *handler);
|
||||||
|
virtual srs_error_t reload(SrsReloadState *pstate) { return srs_success; }
|
||||||
|
virtual srs_error_t persistence() { return srs_success; }
|
||||||
|
virtual std::string config() { return ""; }
|
||||||
|
virtual SrsConfDirective *get_root() { return NULL; }
|
||||||
|
virtual std::string cwd() { return "./"; }
|
||||||
|
virtual int get_max_connections() { return 1000; }
|
||||||
|
virtual std::string get_pid_file() { return ""; }
|
||||||
|
virtual bool empty_ip_ok() { return false; }
|
||||||
|
virtual bool get_asprocess() { return false; }
|
||||||
|
virtual srs_utime_t get_grace_start_wait() { return 0; }
|
||||||
|
virtual srs_utime_t get_grace_final_wait() { return 0; }
|
||||||
|
virtual bool is_force_grace_quit() { return false; }
|
||||||
|
virtual bool inotify_auto_reload() { return false; }
|
||||||
|
virtual bool auto_reload_for_docker() { return false; }
|
||||||
|
virtual std::vector<std::string> get_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual bool get_rtmps_enabled() { return false; }
|
||||||
|
virtual std::vector<std::string> get_rtmps_listen() { return std::vector<std::string>(); }
|
||||||
|
virtual bool get_http_api_enabled() { return false; }
|
||||||
|
virtual std::vector<std::string> get_http_api_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual bool get_https_api_enabled() { return false; }
|
||||||
|
virtual std::vector<std::string> get_https_api_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual std::string get_https_api_ssl_key() { return ""; }
|
||||||
|
virtual std::string get_https_api_ssl_cert() { return ""; }
|
||||||
|
virtual bool get_raw_api() { return false; }
|
||||||
|
virtual bool get_raw_api_allow_reload() { return false; }
|
||||||
|
virtual bool get_raw_api_allow_query() { return false; }
|
||||||
|
virtual bool get_raw_api_allow_update() { return false; }
|
||||||
|
virtual bool get_http_api_auth_enabled() { return false; }
|
||||||
|
virtual std::string get_http_api_auth_username() { return ""; }
|
||||||
|
virtual std::string get_http_api_auth_password() { return ""; }
|
||||||
|
virtual srs_error_t raw_to_json(SrsJsonObject *obj) { return srs_success; }
|
||||||
|
virtual bool get_http_stream_enabled() { return false; }
|
||||||
|
virtual std::vector<std::string> get_http_stream_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual bool get_https_stream_enabled() { return false; }
|
||||||
|
virtual std::vector<std::string> get_https_stream_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual std::string get_https_stream_ssl_key() { return ""; }
|
||||||
|
virtual std::string get_https_stream_ssl_cert() { return ""; }
|
||||||
|
virtual std::string get_http_stream_dir() { return ""; }
|
||||||
|
virtual bool get_http_stream_crossdomain() { return false; }
|
||||||
|
virtual bool get_rtc_server_enabled() { return false; }
|
||||||
|
virtual bool get_rtc_server_tcp_enabled() { return false; }
|
||||||
|
virtual std::vector<std::string> get_rtc_server_tcp_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual std::string get_rtc_server_protocol() { return "udp"; }
|
||||||
|
virtual std::vector<std::string> get_rtc_server_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual int get_rtc_server_reuseport() { return 1; }
|
||||||
|
virtual bool get_rtc_server_encrypt() { return false; }
|
||||||
|
virtual bool get_api_as_candidates() { return api_as_candidates_; }
|
||||||
|
virtual bool get_resolve_api_domain() { return resolve_api_domain_; }
|
||||||
|
virtual bool get_keep_api_domain() { return keep_api_domain_; }
|
||||||
|
virtual std::string get_rtc_server_candidates() { return "*"; }
|
||||||
|
virtual bool get_use_auto_detect_network_ip() { return true; }
|
||||||
|
virtual std::string get_rtc_server_ip_family() { return "ipv4"; }
|
||||||
|
virtual bool get_rtsp_server_enabled() { return false; }
|
||||||
|
virtual std::vector<std::string> get_rtsp_server_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual std::vector<std::string> get_srt_listens() { return std::vector<std::string>(); }
|
||||||
|
virtual std::vector<SrsConfDirective *> get_stream_casters() { return std::vector<SrsConfDirective *>(); }
|
||||||
|
virtual bool get_stream_caster_enabled(SrsConfDirective *conf) { return false; }
|
||||||
|
virtual std::string get_stream_caster_engine(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::string get_stream_caster_output(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual int get_stream_caster_listen(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual bool get_exporter_enabled() { return false; }
|
||||||
|
virtual std::string get_exporter_listen() { return ""; }
|
||||||
|
virtual std::string get_exporter_label() { return ""; }
|
||||||
|
virtual std::string get_exporter_tag() { return ""; }
|
||||||
|
virtual bool get_stats_enabled() { return false; }
|
||||||
|
virtual int get_stats_network() { return 0; }
|
||||||
|
virtual bool get_heartbeat_enabled() { return false; }
|
||||||
|
virtual srs_utime_t get_heartbeat_interval() { return 0; }
|
||||||
|
virtual std::string get_heartbeat_url() { return ""; }
|
||||||
|
virtual std::string get_heartbeat_device_id() { return ""; }
|
||||||
|
virtual bool get_heartbeat_summaries() { return false; }
|
||||||
|
virtual bool get_heartbeat_ports() { return false; }
|
||||||
|
virtual bool get_circuit_breaker() { return false; }
|
||||||
|
virtual int get_high_threshold() { return 0; }
|
||||||
|
virtual int get_high_pulse() { return 0; }
|
||||||
|
virtual int get_critical_threshold() { return 0; }
|
||||||
|
virtual int get_critical_pulse() { return 0; }
|
||||||
|
virtual int get_dying_threshold() { return 0; }
|
||||||
|
virtual int get_dying_pulse() { return 0; }
|
||||||
|
virtual std::string get_rtmps_ssl_cert() { return ""; }
|
||||||
|
virtual std::string get_rtmps_ssl_key() { return ""; }
|
||||||
|
virtual SrsConfDirective *get_vhost(std::string vhost, bool try_default_vhost = true) { return NULL; }
|
||||||
|
virtual bool get_vhost_enabled(std::string vhost) { return true; }
|
||||||
|
virtual bool get_debug_srs_upnode(std::string vhost) { return true; }
|
||||||
|
virtual int get_out_ack_size(std::string vhost) { return 2500000; }
|
||||||
|
virtual int get_in_ack_size(std::string vhost) { return 2500000; }
|
||||||
|
virtual int get_chunk_size(std::string vhost) { return 60000; }
|
||||||
|
virtual bool get_gop_cache(std::string vhost) { return true; }
|
||||||
|
virtual int get_gop_cache_max_frames(std::string vhost) { return 2500; }
|
||||||
|
virtual bool get_tcp_nodelay(std::string vhost) { return false; }
|
||||||
|
virtual srs_utime_t get_mw_sleep(std::string vhost, bool is_rtc = false) { return 350 * SRS_UTIME_MILLISECONDS; }
|
||||||
|
virtual srs_utime_t get_send_min_interval(std::string vhost) { return 0; }
|
||||||
|
virtual bool get_mr_enabled(std::string vhost) { return false; }
|
||||||
|
virtual srs_utime_t get_mr_sleep(std::string vhost) { return 350 * SRS_UTIME_MILLISECONDS; }
|
||||||
|
virtual srs_utime_t get_publish_1stpkt_timeout(std::string vhost) { return 20000 * SRS_UTIME_MILLISECONDS; }
|
||||||
|
virtual srs_utime_t get_publish_normal_timeout(std::string vhost) { return 5000 * SRS_UTIME_MILLISECONDS; }
|
||||||
|
virtual srs_utime_t get_publish_kickoff_for_idle(std::string vhost) { return 0; }
|
||||||
|
virtual bool get_refer_enabled(std::string vhost) { return false; }
|
||||||
|
virtual SrsConfDirective *get_refer_all(std::string vhost) { return NULL; }
|
||||||
|
virtual SrsConfDirective *get_refer_play(std::string vhost) { return NULL; }
|
||||||
|
virtual SrsConfDirective *get_refer_publish(std::string vhost) { return NULL; }
|
||||||
|
virtual bool get_vhost_origin_cluster(std::string vhost) { return false; }
|
||||||
|
virtual std::vector<std::string> get_vhost_coworkers(std::string vhost) { return std::vector<std::string>(); }
|
||||||
|
virtual bool get_vhost_edge_token_traverse(std::string vhost) { return false; }
|
||||||
|
virtual SrsConfDirective *get_vhost_edge_origin(std::string vhost) { return NULL; }
|
||||||
|
virtual SrsConfDirective *get_vhost_on_connect(std::string vhost) { return NULL; }
|
||||||
|
virtual SrsConfDirective *get_vhost_on_close(std::string vhost) { return NULL; }
|
||||||
|
virtual SrsConfDirective *get_vhost_on_publish(std::string vhost) { return NULL; }
|
||||||
|
virtual SrsConfDirective *get_vhost_on_play(std::string vhost) { return NULL; }
|
||||||
|
virtual bool get_rtc_enabled(std::string vhost) { return false; }
|
||||||
|
virtual bool get_rtsp_enabled(std::string vhost) { return false; }
|
||||||
|
virtual bool get_rtc_from_rtmp(std::string vhost) { return false; }
|
||||||
|
virtual bool get_rtsp_from_rtmp(std::string vhost) { return false; }
|
||||||
|
// ISrsAppConfig methods
|
||||||
|
virtual bool get_vhost_http_hooks_enabled(std::string vhost);
|
||||||
|
virtual SrsConfDirective *get_vhost_on_stop(std::string vhost);
|
||||||
|
virtual SrsConfDirective *get_vhost_on_unpublish(std::string vhost);
|
||||||
|
virtual SrsConfDirective *get_vhost_on_dvr(std::string vhost);
|
||||||
|
virtual bool get_rtc_nack_enabled(std::string vhost);
|
||||||
|
virtual bool get_rtc_nack_no_copy(std::string vhost);
|
||||||
|
virtual bool get_realtime_enabled(std::string vhost, bool is_rtc);
|
||||||
|
virtual int get_mw_msgs(std::string vhost, bool is_realtime, bool is_rtc);
|
||||||
|
virtual int get_rtc_drop_for_pt(std::string vhost);
|
||||||
|
virtual bool get_rtc_twcc_enabled(std::string vhost);
|
||||||
|
virtual bool get_srt_enabled();
|
||||||
|
virtual bool get_srt_enabled(std::string vhost);
|
||||||
|
virtual std::string get_srt_default_streamid();
|
||||||
|
virtual bool get_srt_to_rtmp(std::string vhost);
|
||||||
|
virtual bool get_rtc_to_rtmp(std::string vhost);
|
||||||
|
virtual srs_utime_t get_rtc_stun_timeout(std::string vhost);
|
||||||
|
virtual bool get_rtc_stun_strict_check(std::string vhost);
|
||||||
|
virtual std::string get_rtc_dtls_role(std::string vhost);
|
||||||
|
virtual std::string get_rtc_dtls_version(std::string vhost);
|
||||||
|
virtual SrsConfDirective *get_vhost_on_hls(std::string vhost);
|
||||||
|
virtual SrsConfDirective *get_vhost_on_hls_notify(std::string vhost);
|
||||||
|
// HLS methods
|
||||||
|
virtual bool get_hls_enabled(std::string vhost);
|
||||||
|
virtual bool get_hls_enabled(SrsConfDirective *vhost);
|
||||||
|
virtual bool get_hls_use_fmp4(std::string vhost);
|
||||||
|
virtual std::string get_hls_entry_prefix(std::string vhost);
|
||||||
|
virtual std::string get_hls_path(std::string vhost);
|
||||||
|
virtual std::string get_hls_m3u8_file(std::string vhost);
|
||||||
|
virtual std::string get_hls_ts_file(std::string vhost);
|
||||||
|
virtual std::string get_hls_fmp4_file(std::string vhost);
|
||||||
|
virtual std::string get_hls_init_file(std::string vhost);
|
||||||
|
virtual bool get_hls_ts_floor(std::string vhost);
|
||||||
|
virtual srs_utime_t get_hls_fragment(std::string vhost);
|
||||||
|
virtual double get_hls_td_ratio(std::string vhost);
|
||||||
|
virtual double get_hls_aof_ratio(std::string vhost);
|
||||||
|
virtual srs_utime_t get_hls_window(std::string vhost);
|
||||||
|
virtual std::string get_hls_on_error(std::string vhost);
|
||||||
|
virtual bool get_hls_cleanup(std::string vhost);
|
||||||
|
virtual srs_utime_t get_hls_dispose(std::string vhost);
|
||||||
|
virtual bool get_hls_wait_keyframe(std::string vhost);
|
||||||
|
virtual bool get_hls_keys(std::string vhost);
|
||||||
|
virtual int get_hls_fragments_per_key(std::string vhost);
|
||||||
|
virtual std::string get_hls_key_file(std::string vhost);
|
||||||
|
virtual std::string get_hls_key_file_path(std::string vhost);
|
||||||
|
virtual std::string get_hls_key_url(std::string vhost);
|
||||||
|
virtual int get_vhost_hls_nb_notify(std::string vhost);
|
||||||
|
virtual bool get_vhost_hls_dts_directly(std::string vhost);
|
||||||
|
virtual bool get_hls_ctx_enabled(std::string vhost);
|
||||||
|
virtual bool get_hls_ts_ctx_enabled(std::string vhost);
|
||||||
|
virtual bool get_hls_recover(std::string vhost);
|
||||||
|
virtual bool get_forward_enabled(std::string vhost);
|
||||||
|
virtual SrsConfDirective *get_forwards(std::string vhost);
|
||||||
|
virtual srs_utime_t get_queue_length(std::string vhost);
|
||||||
|
virtual SrsConfDirective *get_forward_backend(std::string vhost);
|
||||||
|
virtual bool get_atc(std::string vhost);
|
||||||
|
virtual int get_time_jitter(std::string vhost);
|
||||||
|
virtual bool get_mix_correct(std::string vhost);
|
||||||
|
virtual bool try_annexb_first(std::string vhost);
|
||||||
|
virtual bool get_vhost_is_edge(std::string vhost);
|
||||||
|
virtual bool get_atc_auto(std::string vhost);
|
||||||
|
virtual bool get_reduce_sequence_header(std::string vhost);
|
||||||
|
virtual bool get_parse_sps(std::string vhost);
|
||||||
|
// DVR methods
|
||||||
|
virtual std::string get_dvr_path(std::string vhost) { return "./[vhost]/[app]/[stream]/[2006]/[01]/[02]/[15].[04].[05].[999].flv"; }
|
||||||
|
virtual std::string get_dvr_plan(std::string vhost) { return "session"; }
|
||||||
|
virtual bool get_dvr_enabled(std::string vhost) { return false; }
|
||||||
|
virtual SrsConfDirective *get_dvr_apply(std::string vhost) { return NULL; }
|
||||||
|
virtual srs_utime_t get_dvr_duration(std::string vhost) { return 30 * SRS_UTIME_SECONDS; }
|
||||||
|
virtual int get_dvr_time_jitter(std::string vhost) { return 0; }
|
||||||
|
virtual bool get_dvr_wait_keyframe(std::string vhost) { return true; }
|
||||||
|
virtual bool get_vhost_enabled(SrsConfDirective *conf) { return true; }
|
||||||
|
virtual bool get_vhost_http_remux_enabled(std::string vhost) { return false; }
|
||||||
|
virtual bool get_vhost_http_remux_enabled(SrsConfDirective *vhost) { return false; }
|
||||||
|
virtual srs_utime_t get_vhost_http_remux_fast_cache(std::string vhost) { return 0; }
|
||||||
|
virtual bool get_vhost_http_remux_drop_if_not_match(std::string vhost) { return false; }
|
||||||
|
virtual bool get_vhost_http_remux_has_audio(std::string vhost) { return true; }
|
||||||
|
virtual bool get_vhost_http_remux_has_video(std::string vhost) { return true; }
|
||||||
|
virtual bool get_vhost_http_remux_guess_has_av(std::string vhost) { return true; }
|
||||||
|
virtual std::string get_vhost_http_remux_mount(std::string vhost) { return ""; }
|
||||||
|
virtual std::string get_vhost_edge_protocol(std::string vhost) { return "rtmp"; }
|
||||||
|
virtual bool get_vhost_edge_follow_client(std::string vhost) { return false; }
|
||||||
|
virtual std::string get_vhost_edge_transform_vhost(std::string vhost) { return ""; }
|
||||||
|
// DASH methods
|
||||||
|
virtual bool get_dash_enabled(std::string vhost) { return dash_enabled_; }
|
||||||
|
virtual bool get_dash_enabled(SrsConfDirective *vhost) { return dash_enabled_; }
|
||||||
|
virtual srs_utime_t get_dash_fragment(std::string vhost) { return 30 * SRS_UTIME_SECONDS; }
|
||||||
|
virtual srs_utime_t get_dash_update_period(std::string vhost) { return 30 * SRS_UTIME_SECONDS; }
|
||||||
|
virtual srs_utime_t get_dash_timeshift(std::string vhost) { return 300 * SRS_UTIME_SECONDS; }
|
||||||
|
virtual std::string get_dash_path(std::string vhost) { return "./[vhost]/[app]/[stream]/"; }
|
||||||
|
virtual std::string get_dash_mpd_file(std::string vhost) { return "[stream].mpd"; }
|
||||||
|
virtual int get_dash_window_size(std::string vhost) { return 10; }
|
||||||
|
virtual bool get_dash_cleanup(std::string vhost) { return true; }
|
||||||
|
virtual srs_utime_t get_dash_dispose(std::string vhost) { return dash_dispose_; }
|
||||||
|
// Exec config
|
||||||
|
virtual bool get_exec_enabled(std::string vhost) { return false; }
|
||||||
|
virtual std::vector<SrsConfDirective *> get_exec_publishs(std::string vhost) { return std::vector<SrsConfDirective *>(); }
|
||||||
|
// Ingest config
|
||||||
|
virtual void get_vhosts(std::vector<SrsConfDirective *> &vhosts) {}
|
||||||
|
virtual std::vector<SrsConfDirective *> get_ingesters(std::string vhost) { return std::vector<SrsConfDirective *>(); }
|
||||||
|
virtual SrsConfDirective *get_ingest_by_id(std::string vhost, std::string ingest_id) { return NULL; }
|
||||||
|
virtual bool get_ingest_enabled(SrsConfDirective *conf) { return false; }
|
||||||
|
virtual std::string get_ingest_ffmpeg(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::string get_ingest_input_type(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::string get_ingest_input_url(SrsConfDirective *conf) { return ""; }
|
||||||
|
// FFmpeg log config
|
||||||
|
virtual bool get_ff_log_enabled() { return false; }
|
||||||
|
virtual std::string get_ff_log_dir() { return ""; }
|
||||||
|
virtual std::string get_ff_log_level() { return ""; }
|
||||||
|
// Transcode/Engine config
|
||||||
|
virtual SrsConfDirective *get_transcode(std::string vhost, std::string scope) { return NULL; }
|
||||||
|
virtual bool get_transcode_enabled(SrsConfDirective *conf) { return false; }
|
||||||
|
virtual std::string get_transcode_ffmpeg(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::vector<SrsConfDirective *> get_transcode_engines(SrsConfDirective *conf) { return std::vector<SrsConfDirective *>(); }
|
||||||
|
virtual bool get_engine_enabled(SrsConfDirective *conf) { return false; }
|
||||||
|
virtual std::vector<std::string> get_engine_perfile(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
||||||
|
virtual std::string get_engine_iformat(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::vector<std::string> get_engine_vfilter(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
||||||
|
virtual std::string get_engine_vcodec(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual int get_engine_vbitrate(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual double get_engine_vfps(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual int get_engine_vwidth(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual int get_engine_vheight(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual int get_engine_vthreads(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual std::string get_engine_vprofile(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::string get_engine_vpreset(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::vector<std::string> get_engine_vparams(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
||||||
|
virtual std::string get_engine_acodec(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual int get_engine_abitrate(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual int get_engine_asample_rate(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual int get_engine_achannels(SrsConfDirective *conf) { return 0; }
|
||||||
|
virtual std::vector<std::string> get_engine_aparams(SrsConfDirective *conf) { return std::vector<std::string>(); }
|
||||||
|
virtual std::string get_engine_oformat(SrsConfDirective *conf) { return ""; }
|
||||||
|
virtual std::string get_engine_output(SrsConfDirective *conf) { return ""; }
|
||||||
|
void set_http_hooks_enabled(bool enabled);
|
||||||
|
void set_on_stop_urls(const std::vector<std::string> &urls);
|
||||||
|
void clear_on_stop_directive();
|
||||||
|
void set_on_unpublish_urls(const std::vector<std::string> &urls);
|
||||||
|
void clear_on_unpublish_directive();
|
||||||
|
void set_rtc_nack_enabled(bool enabled);
|
||||||
|
void set_rtc_nack_no_copy(bool no_copy);
|
||||||
|
void set_rtc_drop_for_pt(int pt);
|
||||||
|
void set_rtc_twcc_enabled(bool enabled);
|
||||||
|
void set_srt_enabled(bool enabled);
|
||||||
|
void set_rtc_to_rtmp(bool enabled);
|
||||||
|
void set_api_as_candidates(bool enabled);
|
||||||
|
void set_resolve_api_domain(bool enabled);
|
||||||
|
void set_keep_api_domain(bool enabled);
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
|
@ -77,19 +77,6 @@ const SrsContextId &MockConnection::get_id()
|
||||||
return id;
|
return id;
|
||||||
}
|
}
|
||||||
|
|
||||||
MockExpire::MockExpire() : expired_(false)
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
MockExpire::~MockExpire()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
void MockExpire::expire()
|
|
||||||
{
|
|
||||||
expired_ = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
VOID TEST(ProtocolConnTest, ISrsConnectionInterface)
|
VOID TEST(ProtocolConnTest, ISrsConnectionInterface)
|
||||||
{
|
{
|
||||||
MockConnection conn("192.168.1.100");
|
MockConnection conn("192.168.1.100");
|
||||||
|
|
|
||||||
|
|
@ -30,17 +30,4 @@ public:
|
||||||
virtual const SrsContextId &get_id();
|
virtual const SrsContextId &get_id();
|
||||||
};
|
};
|
||||||
|
|
||||||
class MockExpire : public ISrsExpire
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
bool expired_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
MockExpire();
|
|
||||||
virtual ~MockExpire();
|
|
||||||
|
|
||||||
public:
|
|
||||||
virtual void expire();
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
||||||
|
|
@ -11,65 +11,11 @@
|
||||||
#include <srs_kernel_error.hpp>
|
#include <srs_kernel_error.hpp>
|
||||||
#include <srs_utest_app.hpp>
|
#include <srs_utest_app.hpp>
|
||||||
#include <srs_utest_app6.hpp>
|
#include <srs_utest_app6.hpp>
|
||||||
|
#include <srs_utest_mock.hpp>
|
||||||
|
|
||||||
// MockRtcTrackDescriptionFactory implementation
|
// This test is used to verify the basic workflow of the RTC play stream.
|
||||||
MockRtcTrackDescriptionFactory::MockRtcTrackDescriptionFactory()
|
// It's finished with the help of AI, but each step is manually designed
|
||||||
{
|
// and verified. So this is not dominated by AI, but by humanbeing.
|
||||||
audio_ssrc_ = 12345;
|
|
||||||
video_ssrc_ = 67890;
|
|
||||||
screen_ssrc_ = 98765;
|
|
||||||
}
|
|
||||||
|
|
||||||
MockRtcTrackDescriptionFactory::~MockRtcTrackDescriptionFactory()
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
std::map<uint32_t, SrsRtcTrackDescription *> MockRtcTrackDescriptionFactory::create_audio_video_tracks()
|
|
||||||
{
|
|
||||||
std::map<uint32_t, SrsRtcTrackDescription *> sub_relations;
|
|
||||||
|
|
||||||
// Create audio track
|
|
||||||
SrsRtcTrackDescription *audio_desc = create_audio_track(audio_ssrc_, "audio-track-1", "0");
|
|
||||||
sub_relations[audio_desc->ssrc_] = audio_desc;
|
|
||||||
|
|
||||||
// Create video track
|
|
||||||
SrsRtcTrackDescription *video_desc = create_video_track(video_ssrc_, "video-track-1", "1");
|
|
||||||
sub_relations[video_desc->ssrc_] = video_desc;
|
|
||||||
|
|
||||||
// Create screen share track
|
|
||||||
SrsRtcTrackDescription *screen_desc = create_video_track(screen_ssrc_, "screen-track-1", "2");
|
|
||||||
sub_relations[screen_desc->ssrc_] = screen_desc;
|
|
||||||
|
|
||||||
return sub_relations;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsRtcTrackDescription *MockRtcTrackDescriptionFactory::create_audio_track(uint32_t ssrc, std::string id, std::string mid)
|
|
||||||
{
|
|
||||||
SrsRtcTrackDescription *audio_desc = new SrsRtcTrackDescription();
|
|
||||||
audio_desc->type_ = "audio";
|
|
||||||
audio_desc->ssrc_ = ssrc;
|
|
||||||
audio_desc->id_ = id;
|
|
||||||
audio_desc->is_active_ = true;
|
|
||||||
audio_desc->direction_ = "sendrecv";
|
|
||||||
audio_desc->mid_ = mid;
|
|
||||||
audio_desc->media_ = new SrsAudioPayload(111, "opus", 48000, 2);
|
|
||||||
return audio_desc;
|
|
||||||
}
|
|
||||||
|
|
||||||
SrsRtcTrackDescription *MockRtcTrackDescriptionFactory::create_video_track(uint32_t ssrc, std::string id, std::string mid)
|
|
||||||
{
|
|
||||||
SrsRtcTrackDescription *video_desc = new SrsRtcTrackDescription();
|
|
||||||
video_desc->type_ = "video";
|
|
||||||
video_desc->ssrc_ = ssrc;
|
|
||||||
video_desc->id_ = id;
|
|
||||||
video_desc->is_active_ = true;
|
|
||||||
video_desc->direction_ = "sendrecv";
|
|
||||||
video_desc->mid_ = mid;
|
|
||||||
video_desc->media_ = new SrsVideoPayload(96, "H264", 90000);
|
|
||||||
return video_desc;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Test SrsRtcPlayStream::start() - Basic success scenario
|
|
||||||
VOID TEST(RtcPlayStreamTest, ManuallyVerifyBasicWorkflow)
|
VOID TEST(RtcPlayStreamTest, ManuallyVerifyBasicWorkflow)
|
||||||
{
|
{
|
||||||
srs_error_t err;
|
srs_error_t err;
|
||||||
|
|
@ -207,6 +153,72 @@ VOID TEST(RtcPlayStreamTest, ManuallyVerifyBasicWorkflow)
|
||||||
EXPECT_EQ(pkt->header_.get_ssrc(), track_factory.screen_ssrc_);
|
EXPECT_EQ(pkt->header_.get_ssrc(), track_factory.screen_ssrc_);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Receive a video packet again, to verify the hit cache track.
|
||||||
|
if (true) {
|
||||||
|
SrsUniquePtr<SrsRtpPacket> test_pkt(new SrsRtpPacket());
|
||||||
|
test_pkt->frame_type_ = SrsFrameTypeVideo;
|
||||||
|
test_pkt->header_.set_sequence(1001);
|
||||||
|
test_pkt->header_.set_timestamp(5001);
|
||||||
|
test_pkt->header_.set_ssrc(track_factory.video_ssrc_);
|
||||||
|
|
||||||
|
// Push packet to source - this will feed all consumers including the play stream's consumer
|
||||||
|
HELPER_EXPECT_SUCCESS(play_stream->source_->on_rtp(test_pkt.get()));
|
||||||
|
|
||||||
|
// Wait for coroutine to process the packet
|
||||||
|
srs_usleep(1 * SRS_UTIME_MILLISECONDS);
|
||||||
|
|
||||||
|
// Check sender should have received the packet
|
||||||
|
EXPECT_EQ(mock_sender.send_packet_count_, 4);
|
||||||
|
// Check NACK ring buffer.
|
||||||
|
SrsRtpPacket *pkt = play_stream->cache_track0_->rtp_queue_->at(1001);
|
||||||
|
EXPECT_TRUE(pkt != NULL);
|
||||||
|
EXPECT_EQ(pkt->header_.get_ssrc(), track_factory.video_ssrc_);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Receive a audio packet again, to verify the hit cache track.
|
||||||
|
if (true) {
|
||||||
|
SrsUniquePtr<SrsRtpPacket> test_pkt(new SrsRtpPacket());
|
||||||
|
test_pkt->frame_type_ = SrsFrameTypeAudio;
|
||||||
|
test_pkt->header_.set_sequence(1001);
|
||||||
|
test_pkt->header_.set_timestamp(5001);
|
||||||
|
test_pkt->header_.set_ssrc(track_factory.audio_ssrc_);
|
||||||
|
|
||||||
|
// Push packet to source - this will feed all consumers including the play stream's consumer
|
||||||
|
HELPER_EXPECT_SUCCESS(play_stream->source_->on_rtp(test_pkt.get()));
|
||||||
|
|
||||||
|
// Wait for coroutine to process the packet
|
||||||
|
srs_usleep(1 * SRS_UTIME_MILLISECONDS);
|
||||||
|
|
||||||
|
// Check sender should have received the packet
|
||||||
|
EXPECT_EQ(mock_sender.send_packet_count_, 5);
|
||||||
|
// Check NACK ring buffer.
|
||||||
|
SrsRtpPacket *pkt = play_stream->cache_track1_->rtp_queue_->at(1001);
|
||||||
|
EXPECT_TRUE(pkt != NULL);
|
||||||
|
EXPECT_EQ(pkt->header_.get_ssrc(), track_factory.audio_ssrc_);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Receive a screen share packet again, to verify the hit cache track.
|
||||||
|
if (true) {
|
||||||
|
SrsUniquePtr<SrsRtpPacket> test_pkt(new SrsRtpPacket());
|
||||||
|
test_pkt->frame_type_ = SrsFrameTypeVideo;
|
||||||
|
test_pkt->header_.set_sequence(1001);
|
||||||
|
test_pkt->header_.set_timestamp(5001);
|
||||||
|
test_pkt->header_.set_ssrc(track_factory.screen_ssrc_);
|
||||||
|
|
||||||
|
// Push packet to source - this will feed all consumers including the play stream's consumer
|
||||||
|
HELPER_EXPECT_SUCCESS(play_stream->source_->on_rtp(test_pkt.get()));
|
||||||
|
|
||||||
|
// Wait for coroutine to process the packet
|
||||||
|
srs_usleep(1 * SRS_UTIME_MILLISECONDS);
|
||||||
|
|
||||||
|
// Check sender should have received the packet
|
||||||
|
EXPECT_EQ(mock_sender.send_packet_count_, 6);
|
||||||
|
// Check NACK ring buffer.
|
||||||
|
SrsRtpPacket *pkt = play_stream->cache_track2_->rtp_queue_->at(1001);
|
||||||
|
EXPECT_TRUE(pkt != NULL);
|
||||||
|
EXPECT_EQ(pkt->header_.get_ssrc(), track_factory.screen_ssrc_);
|
||||||
|
}
|
||||||
|
|
||||||
// Stop the play stream
|
// Stop the play stream
|
||||||
play_stream->stop();
|
play_stream->stop();
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -8,32 +8,5 @@
|
||||||
#define SRS_UTEST_RTC_PLAYSTREAM_HPP
|
#define SRS_UTEST_RTC_PLAYSTREAM_HPP
|
||||||
|
|
||||||
#include <srs_utest.hpp>
|
#include <srs_utest.hpp>
|
||||||
#include <map>
|
|
||||||
|
|
||||||
class SrsRtcTrackDescription;
|
|
||||||
|
|
||||||
// Helper class to create mock track descriptions for testing
|
|
||||||
class MockRtcTrackDescriptionFactory
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
MockRtcTrackDescriptionFactory();
|
|
||||||
virtual ~MockRtcTrackDescriptionFactory();
|
|
||||||
|
|
||||||
public:
|
|
||||||
// Default SSRCs for audio and video tracks
|
|
||||||
uint32_t audio_ssrc_;
|
|
||||||
uint32_t video_ssrc_;
|
|
||||||
uint32_t screen_ssrc_;
|
|
||||||
|
|
||||||
public:
|
|
||||||
// Create a map of track descriptions with audio and video tracks
|
|
||||||
std::map<uint32_t, SrsRtcTrackDescription *> create_audio_video_tracks();
|
|
||||||
|
|
||||||
// Create a single audio track description
|
|
||||||
SrsRtcTrackDescription *create_audio_track(uint32_t ssrc, std::string id, std::string mid);
|
|
||||||
|
|
||||||
// Create a single video track description
|
|
||||||
SrsRtcTrackDescription *create_video_track(uint32_t ssrc, std::string id, std::string mid);
|
|
||||||
};
|
|
||||||
|
|
||||||
#endif
|
#endif
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user