FilesExpand file tree

 master

/

test_views.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

150 lines (120 loc) · 3.95 KB

 master

/

test_views.py

Copy path

Top

File metadata and controls

• Code
• Blame

150 lines (120 loc) · 3.95 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

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

import responses

from fitbit.exceptions import BadResponse

from app.fitbit_client import get_auth_url

from app.models import save_fitbit_token, User

def register_batman(test_client):

return test_client.post(

'/register',

data=dict(

username='batman',

password='robin',

confirm='robin'

),

follow_redirects=True

)

def login_batman(test_client):

return test_client.post(

'/login',

data=dict(

username='batman',

password='robin'

),

follow_redirects=True

)

def test_unauthed_index(test_client):

response = test_client.get('/')

assert b'login' in response.data

def test_unsuccessful_login(test_client):

response = register_batman(test_client)

assert response.status_code == 200

response = test_client.post(

'/login',

data=dict(

username='batman',

password='joker'

),

follow_redirects=True

)

assert response.status_code == 401

assert b'Invalid Credentials' in response.data

response = test_client.post(

'/login',

data=dict(

username='joker',

password='robin'

),

follow_redirects=True

)

assert b'Invalid Credentials' in response.data

assert response.status_code == 401

def test_logout(test_client):

response = register_batman(test_client)

assert response.status_code == 200

response = login_batman(test_client)

assert response.status_code == 200

assert b'batman' in response.data

response = test_client.get('/logout', follow_redirects=True)

assert b'Logged Out' in response.data

assert b'login' in response.data # Redirected to login

def test_register_and_login_new_user(test_client):

response = register_batman(test_client)

assert response.status_code == 200

response = login_batman(test_client)

assert response.status_code == 200

assert b'batman' in response.data

def test_reregistering_is_error(test_client):

response = register_batman(test_client)

assert response.status_code == 200

response = register_batman(test_client)

assert response.status_code == 400

assert b'Username batman already taken' in response.data

@responses.activate

def test_handle_redirect(test_client):

code = 'iamacode'

# First we will do the authorization with the code

# sent

responses.add(

responses.POST,

get_auth_url(code),

json={'access_token': 'acc', 'refresh_token': 'ref'},

match_querystring=True

)

# Then we will get a token

responses.add(

responses.POST,

'https://api.fitbit.com/oauth2/token',

json={'access_token': 'acc', 'refresh_token': 'ref'}

)

# After getting the token we redirect to index which requests the profile

responses.add(

responses.GET,

'https://api.fitbit.com/1/user/-/profile.json',

json={'user': {'fullName': 'bat man', 'memberSince': '12/12/12'}}

)

register_batman(test_client)

login_batman(test_client)

response = test_client.get(

'/oauth-redirect?code={}'.format(code),

follow_redirects=True

)

assert b'batman' in response.data

assert b'bat man' in response.data

assert b'12/12/12' in response.data

@responses.activate

def test_api_call_failed(test_client):

register_batman(test_client)

save_fitbit_token(User.query.filter_by(username='batman').first().id, 'acc', 'ref')

# Then we will get a token

responses.add(

responses.POST,

'https://api.fitbit.com/oauth2/token',

json={'access_token': 'acc', 'refresh_token': 'ref'}

)

# After getting the token we redirect to index which requests the profile

responses.add(

responses.GET,

'https://api.fitbit.com/1/user/-/profile.json',

body=BadResponse("I FAILED")

)

response = login_batman(test_client)

assert b'Api Call Failed' in response.data