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')
