andreacavalli
/
tdlib-session-container
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Enhance test client

This commit is contained in:
Andrea Cavalli 2021-08-01 00:49:12 +02:00
parent abbf91cdf1
commit ebbd96d849
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/main/java/it/tdlight/tdlibsession/td/direct/TestClient.java
View File

@ -10,6 +10,7 @@ import it.tdlight.jni.TdApi.Close;
import it.tdlight.jni.TdApi.ConnectionStateReady; import it.tdlight.jni.TdApi.ConnectionStateReady;
import it.tdlight.jni.TdApi.FormattedText; import it.tdlight.jni.TdApi.FormattedText;
import it.tdlight.jni.TdApi.Function; import it.tdlight.jni.TdApi.Function;
import it.tdlight.jni.TdApi.GetMe;
import it.tdlight.jni.TdApi.Message; import it.tdlight.jni.TdApi.Message;
import it.tdlight.jni.TdApi.MessageSenderUser; import it.tdlight.jni.TdApi.MessageSenderUser;
import it.tdlight.jni.TdApi.MessageText; import it.tdlight.jni.TdApi.MessageText;
@ -22,6 +23,7 @@ import it.tdlight.jni.TdApi.TextEntity;
import it.tdlight.jni.TdApi.UpdateAuthorizationState; import it.tdlight.jni.TdApi.UpdateAuthorizationState;
import it.tdlight.jni.TdApi.UpdateConnectionState; import it.tdlight.jni.TdApi.UpdateConnectionState;
import it.tdlight.jni.TdApi.UpdateNewMessage; import it.tdlight.jni.TdApi.UpdateNewMessage;
import it.tdlight.jni.TdApi.User;
import it.tdlight.tdlibsession.td.ReactorTelegramClient; import it.tdlight.tdlibsession.td.ReactorTelegramClient;
import java.util.ArrayList; import java.util.ArrayList;
import java.util.Iterator; import java.util.Iterator;
@ -74,14 +76,17 @@ public class TestClient implements ReactorTelegramClient {
return Flux.fromIterable(features).flatMap(featureName -> { return Flux.fromIterable(features).flatMap(featureName -> {
switch (featureName) { switch (featureName) {
case "status-update": case "status-update":
return Flux.<TdApi.Object>just( return Flux
new UpdateAuthorizationState(new AuthorizationStateReady()), .<TdApi.Object>just(
new UpdateConnectionState(new ConnectionStateReady()) new UpdateAuthorizationState(new AuthorizationStateReady()),
).mergeWith(closedSink new UpdateConnectionState(new ConnectionStateReady())
.asMono() )
.thenMany(Flux.just(new UpdateAuthorizationState(new AuthorizationStateClosing()), .mergeWith(closedSink
new UpdateAuthorizationState(new AuthorizationStateClosed()) .asMono()
))); .thenMany(Flux.just(new UpdateAuthorizationState(new AuthorizationStateClosing()),
new UpdateAuthorizationState(new AuthorizationStateClosed())
))
);
case "infinite-messages": case "infinite-messages":
var randomSenders = features.contains("random-senders"); var randomSenders = features.contains("random-senders");
var randomChats = features.contains("random-chats"); var randomChats = features.contains("random-chats");
@ -134,6 +139,13 @@ public class TestClient implements ReactorTelegramClient {
case SetTdlibParameters.CONSTRUCTOR: case SetTdlibParameters.CONSTRUCTOR:
case SetOption.CONSTRUCTOR: case SetOption.CONSTRUCTOR:
return new Ok(); return new Ok();
case GetMe.CONSTRUCTOR:
var user = new User();
user.id = 420;
user.firstName = "Test";
user.lastName = "Test";
user.phoneNumber = "+77";
return user;
case Close.CONSTRUCTOR: case Close.CONSTRUCTOR:
closedSink.tryEmitEmpty(); closedSink.tryEmitEmpty();
return new Ok(); return new Ok();