From 1df1d84766e09b5de51d7fb22e8ebb2432a929e4 Mon Sep 17 00:00:00 2001
From: winlin <winlin@vip.126.com>
Date: Sun, 3 Feb 2019 21:23:06 +0800
Subject: [PATCH] CI: Cover kernel LB round robin

---
 trunk/src/utest/srs_utest_kernel.cpp | 36 ++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/trunk/src/utest/srs_utest_kernel.cpp b/trunk/src/utest/srs_utest_kernel.cpp
index 6b6336ce9..8c6e47777 100644
--- a/trunk/src/utest/srs_utest_kernel.cpp
+++ b/trunk/src/utest/srs_utest_kernel.cpp
@@ -31,6 +31,7 @@ using namespace std;
 #include <srs_protocol_utility.hpp>
 #include <srs_kernel_buffer.hpp>
 #include <srs_kernel_aac.hpp>
+#include <srs_kernel_balance.hpp>
 
 #define MAX_MOCK_DATA_SIZE 1024 * 1024
 
@@ -2344,3 +2345,38 @@ VOID TEST(KernelAACTest, TransmaxRTMP2AAC)
     }
 }
 
+VOID TEST(KernelLBRRTest, CoverAll)
+{
+    if (true) {
+        SrsLbRoundRobin lb;
+        EXPECT_EQ(0, (int)lb.count);
+        EXPECT_EQ(-1, lb.index);
+        EXPECT_EQ(-1, (int)lb.current());
+        EXPECT_TRUE("" == lb.selected());
+    }
+    
+    if (true) {
+        vector<string> servers;
+        servers.push_back("s0");
+        servers.push_back("s1");
+        servers.push_back("s2");
+        
+        SrsLbRoundRobin lb;
+        lb.select(servers);
+        EXPECT_EQ(0, (int)lb.current());
+        EXPECT_TRUE("s0" == lb.selected());
+        
+        lb.select(servers);
+        EXPECT_EQ(1, (int)lb.current());
+        EXPECT_TRUE("s1" == lb.selected());
+        
+        lb.select(servers);
+        EXPECT_EQ(2, (int)lb.current());
+        EXPECT_TRUE("s2" == lb.selected());
+        
+        lb.select(servers);
+        EXPECT_EQ(0, (int)lb.current());
+        EXPECT_TRUE("s0" == lb.selected());
+    }
+}
+