diff --git a/src/main/java/jenkins/plugins/rocketchatnotifier/rocket/LegacyRocketChatClient.java b/src/main/java/jenkins/plugins/rocketchatnotifier/rocket/LegacyRocketChatClient.java index 8f66bdf0..7de9557a 100644 --- a/src/main/java/jenkins/plugins/rocketchatnotifier/rocket/LegacyRocketChatClient.java +++ b/src/main/java/jenkins/plugins/rocketchatnotifier/rocket/LegacyRocketChatClient.java @@ -14,10 +14,7 @@ import sun.security.validator.ValidatorException; import java.io.IOException; -import java.util.HashMap; -import java.util.HashSet; -import java.util.Map; -import java.util.Set; +import java.util.*; /** * Created by mreinhardt on 26.12.16. @@ -163,6 +160,6 @@ public User getUser(String userId) throws IOException { @Override public Room[] getChannels() throws IOException { - return new Room[0]; + return this.getPublicRooms().toArray(new Room[0]); } } diff --git a/src/test/java/jenkins/plugins/rocketchatnotifier/rocket/RocketChatClientIT.java b/src/test/java/jenkins/plugins/rocketchatnotifier/rocket/RocketChatClientIT.java index d8b58b5d..3689ce65 100644 --- a/src/test/java/jenkins/plugins/rocketchatnotifier/rocket/RocketChatClientIT.java +++ b/src/test/java/jenkins/plugins/rocketchatnotifier/rocket/RocketChatClientIT.java @@ -16,7 +16,13 @@ public class RocketChatClientIT { @Before public void setup() throws Exception { - this.client = new RocketChatClientImpl("http://localhost:4443/api/", "admin", "supersecret"); // TODO read from env + String rocketVersionString = System.getProperty("rocket.version", "0.48.2"); + int rocketVersion = Integer.parseInt(rocketVersionString.split("\\.")[1]); + if (rocketVersion >= 48) { + this.client = new RocketChatClientImpl("http://localhost:4443/api/", "admin", "supersecret"); // TODO read from env + } else { + this.client = new LegacyRocketChatClient("http://localhost:4443/api/", "admin", "supersecret"); + } } @Test