55 from mock import Mock , create_autospec
66
77from testfixtures import LogCapture
8- from slackclient import SlackClient , _channel , _server , _util
8+ from slackclient import SlackClient
9+ from slackclient .server import Server
10+ from slackclient .channel import Channel
11+ from slackclient .util import SearchList
912from rtmbot .core import RtmBot , Plugin
1013
1114def init_rtmbot ():
@@ -36,13 +39,13 @@ def test_output():
3639
3740 # Mock the slack_client object with Server, Channel objects and needed methods
3841 slackclient_mock = create_autospec (SlackClient )
39- server_mock = create_autospec (_server . Server )
42+ server_mock = create_autospec (Server )
4043
4144 # Mock Server with channels method and correct return value
4245 slackclient_mock .server = server_mock
43- searchlist_mock = create_autospec (_util . SearchList )
46+ searchlist_mock = create_autospec (SearchList )
4447 server_mock .channels = searchlist_mock
45- channel_mock = create_autospec (_channel . Channel )
48+ channel_mock = create_autospec (Channel )
4649 slackclient_mock .server .channels .find .return_value = channel_mock
4750
4851 rtmbot .slack_client = slackclient_mock
@@ -72,4 +75,4 @@ def test_output():
7275 plugin_mock .do_output .return_value = [['C12345678' , 'ù hœø3ö' ]]
7376 rtmbot .output ()
7477
75- channel_mock .send_message .assert_called_with ('ù hœø3ö' )
78+ channel_mock .send_message .assert_called_with ('ù hœø3ö' )
0 commit comments