class TestLineBotApi(unittest.TestCase): def setUp(self): self.tested = LineBotApi('channel_secret') @responses.activate def test_get_group_member_ids(self): responses.add( responses.GET, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/group/group_id/members/ids', json={ 'memberIds': ['U1', 'U2'] }, status=200 ) member_ids_response = self.tested.get_group_member_ids('group_id') request = responses.calls[0].request self.assertEqual(request.method, 'GET') self.assertEqual( request.url, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/group/group_id/members/ids') self.assertEqual(member_ids_response.member_ids, ['U1', 'U2']) self.assertEqual(member_ids_response.next, None) @responses.activate def test_get_group_member_ids_with_start(self): responses.add( responses.GET, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/group/group_id/members/ids', json={ 'memberIds': ['U1', 'U2'], 'next': 'continuationToken2' }, status=200 ) member_ids_response = self.tested.get_group_member_ids('group_id', start='continuationToken1') request = responses.calls[0].request self.assertEqual(request.method, 'GET') self.assertEqual( request.url, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/group/group_id/members/ids?start=continuationToken1') self.assertEqual(member_ids_response.member_ids, ['U1', 'U2']) self.assertEqual(member_ids_response.next, 'continuationToken2') @responses.activate def test_get_room_member_ids(self): responses.add( responses.GET, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/room/room_id/members/ids', json={ 'memberIds': ['U1', 'U2'] }, status=200 ) member_ids_response = self.tested.get_room_member_ids('room_id') request = responses.calls[0].request self.assertEqual(request.method, 'GET') self.assertEqual( request.url, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/room/room_id/members/ids') self.assertEqual(member_ids_response.member_ids, ['U1', 'U2']) self.assertEqual(member_ids_response.next, None) @responses.activate def test_get_room_member_ids_with_start(self): responses.add( responses.GET, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/room/room_id/members/ids', json={ 'memberIds': ['U1', 'U2'], 'next': 'continuationToken2' }, status=200 ) member_ids_response = self.tested.get_room_member_ids('room_id', start='continuationToken1') request = responses.calls[0].request self.assertEqual(request.method, 'GET') self.assertEqual( request.url, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/room/room_id/members/ids?start=continuationToken1') self.assertEqual(member_ids_response.member_ids, ['U1', 'U2']) self.assertEqual(member_ids_response.next, 'continuationToken2') @responses.activate def test_get_follower_user_ids(self): responses.add( responses.GET, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/followers/ids', json={ 'userIds': ['U1', 'U2'] }, status=200 ) member_ids_response = self.tested.get_followers_ids() request = responses.calls[0].request self.assertEqual(request.method, 'GET') self.assertEqual( request.url, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/followers/ids?limit=300') self.assertEqual(member_ids_response.user_ids, ['U1', 'U2']) self.assertEqual(member_ids_response.next, None) @responses.activate def test_get_follower_user_ids_with_start(self): responses.add( responses.GET, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/followers/ids', json={ 'userIds': ['U1', 'U2'], 'next': 'continuationToken2' }, status=200 ) member_ids_response = self.tested.get_followers_ids(start='continuationToken1') request = responses.calls[0].request self.assertEqual(request.method, 'GET') self.assertEqual( request.url, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/followers/ids?limit=300&start=continuationToken1') self.assertEqual(member_ids_response.user_ids, ['U1', 'U2']) self.assertEqual(member_ids_response.next, 'continuationToken2') @responses.activate def test_get_follower_user_ids_with_start_and_limit(self): responses.add( responses.GET, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/followers/ids', json={ 'userIds': ['U1', 'U2'], 'next': 'continuationToken2' }, status=200 ) member_ids_response = self.tested.get_followers_ids(limit=2, start='continuationToken1') request = responses.calls[0].request self.assertEqual(request.method, 'GET') self.assertEqual( request.url, LineBotApi.DEFAULT_API_ENDPOINT + '/v2/bot/followers/ids?limit=2&start=continuationToken1') self.assertEqual(member_ids_response.user_ids, ['U1', 'U2']) self.assertEqual(member_ids_response.next, 'continuationToken2')