From d3e58de85196a292609a3d71585207fca63c5470 Mon Sep 17 00:00:00 2001 From: Shang Jihao <11811011@mail.sustech.edu.cn> Date: Sun, 25 Apr 2021 18:09:24 +0800 Subject: [PATCH] fix issue 869 --- .../telegram/abilitybots/api/objects/Ability.java | 5 +++++ .../abilitybots/api/objects/AbilityTest.java | 14 ++++++++++++++ 2 files changed, 19 insertions(+) diff --git a/telegrambots-abilities/src/main/java/org/telegram/abilitybots/api/objects/Ability.java b/telegrambots-abilities/src/main/java/org/telegram/abilitybots/api/objects/Ability.java index 92c6d9b2..34ce39ce 100644 --- a/telegrambots-abilities/src/main/java/org/telegram/abilitybots/api/objects/Ability.java +++ b/telegrambots-abilities/src/main/java/org/telegram/abilitybots/api/objects/Ability.java @@ -200,6 +200,11 @@ public final class Ability { return this; } + public AbilityBuilder setStatsEnabled(boolean statsEnabled) { + this.statsEnabled = statsEnabled; + return this; + } + public AbilityBuilder privacy(Privacy privacy) { this.privacy = privacy; return this; diff --git a/telegrambots-abilities/src/test/java/org/telegram/abilitybots/api/objects/AbilityTest.java b/telegrambots-abilities/src/test/java/org/telegram/abilitybots/api/objects/AbilityTest.java index 91cca733..3f8d0955 100644 --- a/telegrambots-abilities/src/test/java/org/telegram/abilitybots/api/objects/AbilityTest.java +++ b/telegrambots-abilities/src/test/java/org/telegram/abilitybots/api/objects/AbilityTest.java @@ -5,6 +5,8 @@ import org.junit.jupiter.api.Test; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertNotEquals; +import static org.junit.jupiter.api.Assertions.assertTrue; +import static org.junit.jupiter.api.Assertions.assertFalse; import static org.telegram.abilitybots.api.bot.DefaultBot.getDefaultBuilder; class AbilityTest { @@ -55,5 +57,17 @@ class AbilityTest { assertEquals(ability1, ability4, "Abilities should not be equal"); assertNotEquals(ability1, ability3, "Abilities should be equal"); } + + @Test + void abilityBuilderSetStatsEnabledTrueTest() { + Ability statsEnabledAbility = getDefaultBuilder().setStatsEnabled(true).build(); + assertTrue(statsEnabledAbility.statsEnabled()); + } + + @Test + void abilityBuilderSetStatsEnabledFalseTest() { + Ability statsDisabledAbility = getDefaultBuilder().setStatsEnabled(false).build(); + assertFalse(statsDisabledAbility.statsEnabled()); + } }