Adding tests for direct message with allowed channels

attzonko · Feb 3, 2019 · f1c50c5 · f1c50c5
1 parent 9b93646
commit f1c50c5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/behavior_tests/test_allowed_channels.py b/tests/behavior_tests/test_allowed_channels.py
@@ -2,9 +2,15 @@
 
 
 def test_allowed_channels(driver):  # noqa: F811
+
+    driver.send_direct_message('allowed_channel', tobot=True)
+    driver.wait_for_bot_direct_message(
+        'Driver in channel allowed!')
+
     driver.send_private_channel_message('allowed_channel', tobot=True)
     driver.wait_for_bot_private_channel_message(
         'Driver in channel allowed!', tosender=True)
+
     driver.send_private_channel_message('not_allowed_channel', tobot=True)
     try:
         driver.wait_for_bot_private_channel_message(