Fix test cases and add new test cases

ant-media · Nov 24, 2024 · d607be7 · d607be7
1 parent 39e5e3d
commit d607be7
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 3 deletions.
diff --git a/src/test/java/io/antmedia/test/AntMediaApplicationAdaptorUnitTest.java b/src/test/java/io/antmedia/test/AntMediaApplicationAdaptorUnitTest.java
@@ -96,6 +96,7 @@
 import io.antmedia.rest.model.Result;
 import io.antmedia.security.AcceptOnlyStreamsInDataStore;
 import io.antmedia.settings.ServerSettings;
+import io.antmedia.statistic.HlsViewerStats;
 import io.antmedia.statistic.type.WebRTCAudioReceiveStats;
 import io.antmedia.statistic.type.WebRTCAudioSendStats;
 import io.antmedia.statistic.type.WebRTCVideoReceiveStats;
@@ -827,8 +828,9 @@ public void testHookAfterDefined()
 
 		assertEquals(hookURL, spyAdaptor.getListenerHookURL(broadcast));
 
-
 		spyAdaptor = Mockito.spy(adapter);
+		Mockito.doNothing().when(spyAdaptor).resetHLSStats(Mockito.anyString());
+		Mockito.doNothing().when(spyAdaptor).resetDASHStats(Mockito.anyString());
 		appSettings = new AppSettings();
 		spyAdaptor.setServerSettings(new ServerSettings());
 		spyAdaptor.setAppSettings(appSettings);

diff --git a/src/test/java/io/antmedia/test/db/DBStoresUnitTest.java b/src/test/java/io/antmedia/test/db/DBStoresUnitTest.java
@@ -141,7 +141,7 @@ public void testMapDBStore() throws Exception {
 
 		DataStore dataStore = new MapDBStore("testdb", vertx);
 
-
+		testLocalLiveBroadcast(dataStore);
 		testUpdateBroadcastEncoderSettings(dataStore);
 		testSubscriberMetaData(dataStore);
 		testGetActiveBroadcastCount(dataStore);
@@ -231,6 +231,7 @@ public void testMemoryDataStore() throws Exception {
 
 		testVoDFunctions(dataStore);
 
+		testLocalLiveBroadcast(dataStore);
 		testUpdateBroadcastEncoderSettings(dataStore);
 		testSubscriberMetaData(dataStore);
 		testBlockSubscriber(dataStore);
@@ -293,8 +294,10 @@ public void testMongoStore() throws Exception {
 
 		dataStore = new MongoStore("127.0.0.1", "", "", "testdb");
 
+
 		testSaveDuplicateStreamId((MongoStore)dataStore);
 
+		testLocalLiveBroadcast(dataStore);
 		testUpdateBroadcastEncoderSettings(dataStore);
 		testSubscriberMetaData(dataStore);
 		testBlockSubscriber(dataStore);
@@ -360,6 +363,7 @@ public void testRedisStore() throws Exception {
 		dataStore.close(true);
 		dataStore = new RedisStore("redis://127.0.0.1:6379", "testdb");
 
+		testLocalLiveBroadcast(dataStore);
 		testUpdateBroadcastEncoderSettings(dataStore);
 		testSubscriberMetaData(dataStore);
 		testBlockSubscriber(dataStore);
@@ -644,6 +648,58 @@ public void testUnexpectedBroadcastOffset(DataStore dataStore) {
 		assertNotNull(broadcastList);
 		assertEquals(0, broadcastList.size());
 	}
+
+	public void testLocalLiveBroadcast(DataStore dataStore) {
+		clear(dataStore);
+
+		assertEquals(0, dataStore.getBroadcastCount());
+
+		long streamCount = 10;
+		String streamId = null;
+		for (int i = 0; i < streamCount; i++) {
+			Broadcast broadcast = new Broadcast(null, null);
+			broadcast.setStatus(AntMediaApplicationAdapter.BROADCAST_STATUS_BROADCASTING);
+			broadcast.setUpdateTime(System.currentTimeMillis());
+			streamId = dataStore.save(broadcast);
+			logger.info("Saved streamId:{}", streamId);
+		}
+
+
+		assertEquals(streamCount, dataStore.getActiveBroadcastCount());
+
+		if (dataStore instanceof MapDBStore || dataStore instanceof InMemoryDataStore) {
+			assertEquals(streamCount, dataStore.getLocalLiveBroadcastCount(ServerSettings.getLocalHostAddress()));
+
+			List<Broadcast> localLiveBroadcasts = dataStore.getLocalLiveBroadcasts(ServerSettings.getLocalHostAddress());
+            assertEquals(streamCount, localLiveBroadcasts.size());
+		} else {
+
+			//because there is no origin address registered
+			assertEquals(0, dataStore.getLocalLiveBroadcastCount(ServerSettings.getLocalHostAddress()));
+			List<Broadcast> localLiveBroadcasts = dataStore.getLocalLiveBroadcasts(ServerSettings.getLocalHostAddress());
+			assertEquals(0, localLiveBroadcasts.size());
+		}
+
+		clear(dataStore);
+
+		assertEquals(0, dataStore.getBroadcastCount());
+
+		 streamCount = 15;
+		for (int i = 0; i < streamCount; i++) {
+			Broadcast broadcast = new Broadcast(null, null);
+			broadcast.setStatus(AntMediaApplicationAdapter.BROADCAST_STATUS_BROADCASTING);
+			broadcast.setUpdateTime(System.currentTimeMillis());
+			broadcast.setOriginAdress(ServerSettings.getLocalHostAddress());
+			streamId = dataStore.save(broadcast);
+			logger.info("Saved streamId:{}", streamId);
+		}
+
+		assertEquals(streamCount, dataStore.getLocalLiveBroadcastCount(ServerSettings.getLocalHostAddress()));
+
+		List<Broadcast> localLiveBroadcasts = dataStore.getLocalLiveBroadcasts(ServerSettings.getLocalHostAddress());
+        assertEquals(streamCount, localLiveBroadcasts.size());
+
+	}
 
 	public void testGetActiveBroadcastCount(DataStore dataStore) {
 
@@ -661,7 +717,9 @@ public void testGetActiveBroadcastCount(DataStore dataStore) {
 
 		String streamId = null;
 		for (int i = 0; i < streamCount; i++) {
-			streamId = dataStore.save(new Broadcast(null, null));
+			Broadcast broadcast = new Broadcast(null, null);
+			broadcast.setOriginAdress(ServerSettings.getLocalHostAddress());
+			streamId = dataStore.save(broadcast);
 			logger.info("Saved streamId:{}", streamId);
 		}