def test_like_unlike_artist_ok():
# create artist
created, data, errors = create(**artist_data)
artist_id = data['artist'].id
# create user
created, data, errors = register('*****@*****.**', '1234')
user_id = data['user'].id
# like artist
like(user_id, artist_id)
# check user artist likes
user_likes_key = 'user:{}:likes'.format(user_id)
user_likes_artist_ids = r.smembers(user_likes_key)
assert user_likes_artist_ids
assert len(user_likes_artist_ids) == 1
assert int(user_likes_artist_ids.pop()) == artist_id
# unlike artist
unlike(user_id, artist_id)
# remove user artist like
assert not user_likes_artist_ids
assert len(user_likes_artist_ids) == 0
def post(self):
parser = reqparse.RequestParser()
parser.add_argument('Name',
required=True,
type=str,
help='Name cannot be found')
parser.add_argument('Phoneno',
required=True,
type=int,
help='Phoneno cannot be found')
parser.add_argument('AadharNo',
required=True,
type=int,
help='AadharNo cannot be found')
parser.add_argument('Latitude',
required=True,
type=str,
help='Latitude cannot be found')
parser.add_argument('Longitude',
required=True,
type=str,
help='Longitude cannot be found')
parser.add_argument('Photo',
type=str,
required=True,
help='Photo From Camera cannot be found')
args = parser.parse_args()
return jsonify(user.register(args))
def post(self, request, data):
success, result, errors = register(
data["register_data"].email,
data["register_data"].password
)
if errors:
return BadRequest(errors)
return Ok({"user": result["user"].toJSONDict()})
def test_register_fail():
user = build_user()
with patch('services.user.get_user_by_email') as mock_get_user:
with patch('services.user.create_user') as mock_create_user:
with patch('services.user.send_email') as mock_send_email:
mock_get_user.return_value = user
ok, data, errors = register("*****@*****.**", "aaa")
assert mock_create_user.call_count == 0
assert mock_send_email.call_count == 0
assert ok is False
assert 'email' in errors
def test_register_ok():
user = build_user()
with patch('services.user.get_user_by_email') as mock_get_user:
with patch('services.user.create_user') as mock_create_user:
with patch('services.user.send_email') as mock_send_email:
mock_get_user.return_value = None
mock_create_user.return_value = user
ok, data, errors = register("*****@*****.**", "aaa")
assert mock_create_user.call_count == 1
#assert mock_send_email.call_count == 1
assert ok is True
assert 'user' in data
