Add utest for connection manager

trunk/src/utest/srs_utest_rtc.cpp

@@ -30,10 +30,32 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #include <srs_app_rtc_source.hpp>
 #include <srs_app_rtc_conn.hpp>
 #include <srs_kernel_codec.hpp>
+#include <srs_app_conn.hpp>
+
+#include <srs_utest_service.hpp>
 
 #include <vector>
 using namespace std;
 
+VOID TEST(KernelRTCTest, ConnectionManagerTest)
+{
+    srs_error_t err = srs_success;
+
+    if (true) {
+        SrsConnectionManager manager;
+        HELPER_EXPECT_SUCCESS(manager.start());
+        EXPECT_EQ(0, manager.size()); EXPECT_TRUE(manager.empty());
+
+        MockSrsConnection* conn = new MockSrsConnection();
+        manager.add(conn);
+        EXPECT_EQ(1, manager.size()); EXPECT_FALSE(manager.empty());
+
+        manager.remove(conn);
+        srs_usleep(0); // Switch context for manager to dispose connections.
+        EXPECT_EQ(0, manager.size()); EXPECT_TRUE(manager.empty());
+    }
+}
+
 VOID TEST(KernelRTCTest, StringDumpHexTest)
 {
     // Typical normal case.

trunk/src/utest/srs_utest_service.cpp

@@ -42,17 +42,18 @@ using namespace std;
 #include <sys/socket.h>
 #include <netdb.h>
 
-class MockSrsConnection : public ISrsConnection
+MockSrsConnection::MockSrsConnection()
 {
-public:
-    MockSrsConnection() {
-    }
-    virtual ~MockSrsConnection() {
-    }
-    virtual std::string remote_ip() {
-        return "127.0.0.1";
-    }
-};
+}
+
+MockSrsConnection::~MockSrsConnection()
+{
+}
+
+std::string MockSrsConnection::remote_ip()
+{
+    return "127.0.0.1";
+}
 
 VOID TEST(ServiceTimeTest, TimeUnit)
 {

trunk/src/utest/srs_utest_service.hpp

@@ -30,6 +30,15 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 #include <srs_utest.hpp>
 
 #include <srs_app_st.hpp>
+#include <srs_service_conn.hpp>
+
+class MockSrsConnection : public ISrsConnection
+{
+public:
+    MockSrsConnection();
+    virtual ~MockSrsConnection();
+    virtual std::string remote_ip();
+};
 
 #endif