FilesExpand file tree

 develop

/

test_user.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

119 lines (96 loc) · 3.83 KB

 develop

/

test_user.py

Copy path

Top

File metadata and controls

• Code
• Blame

119 lines (96 loc) · 3.83 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

#!/usr/bin/env python

#

# A library that provides a Python interface to the Telegram Bot API

# Copyright (C) 2015-2017

# Leandro Toledo de Souza <devs@python-telegram-bot.org>

#

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU Lesser Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

# GNU Lesser Public License for more details.

#

# You should have received a copy of the GNU Lesser Public License

# along with this program. If not, see [http://www.gnu.org/licenses/].

import pytest

from telegram import User, Update

@pytest.fixture(scope='function')

def json_dict():

return {

'id': TestUser.id,

'is_bot': TestUser.is_bot,

'first_name': TestUser.first_name,

'last_name': TestUser.last_name,

'username': TestUser.username,

'language_code': TestUser.language_code

}

@pytest.fixture(scope='function')

def user(bot):

return User(TestUser.id, TestUser.first_name, TestUser.is_bot, last_name=TestUser.last_name,

username=TestUser.username, language_code=TestUser.language_code, bot=bot)

class TestUser(object):

id = 1

is_bot = True

first_name = 'first_name'

last_name = 'last_name'

username = 'username'

language_code = 'en_us'

def test_de_json(self, json_dict, bot):

user = User.de_json(json_dict, bot)

assert user.id == self.id

assert user.is_bot == self.is_bot

assert user.first_name == self.first_name

assert user.last_name == self.last_name

assert user.username == self.username

assert user.language_code == self.language_code

def test_de_json_without_username(self, json_dict, bot):

del json_dict['username']

user = User.de_json(json_dict, bot)

assert user.id == self.id

assert user.is_bot == self.is_bot

assert user.first_name == self.first_name

assert user.last_name == self.last_name

assert user.username is None

assert user.language_code == self.language_code

def test_de_json_without_username_and_last_name(self, json_dict, bot):

del json_dict['username']

del json_dict['last_name']

user = User.de_json(json_dict, bot)

assert user.id == self.id

assert user.is_bot == self.is_bot

assert user.first_name == self.first_name

assert user.last_name is None

assert user.username is None

assert user.language_code == self.language_code

def test_name(self, user):

assert user.name == '@username'

user.username = None

assert user.name == 'first_name last_name'

user.last_name = None

assert user.name == 'first_name'

user.username = self.username

assert user.name == '@username'

def test_get_profile_photos(self, monkeypatch, user):

def test(_, *args, **kwargs):

return args[0] == user.id

monkeypatch.setattr('telegram.Bot.get_user_profile_photos', test)

assert user.get_profile_photos()

def test_equality(self):

a = User(self.id, self.first_name, self.is_bot, self.last_name)

b = User(self.id, self.first_name, self.is_bot, self.last_name)

c = User(self.id, self.first_name, self.is_bot)

d = User(0, self.first_name, self.is_bot, self.last_name)

e = Update(self.id)

assert a == b

assert hash(a) == hash(b)

assert a is not b

assert a == c

assert hash(a) == hash(c)

assert a != d

assert hash(a) != hash(d)

assert a != e

assert hash(a) != hash(e)