Add mock X-Forwarded-For header in trackable tests

pallets-eco · May 6, 2014 · d185407 · d185407
1 parent 58b7fa8
commit d185407
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_trackable.py b/tests/test_trackable.py
@@ -17,12 +17,12 @@ def test_trackable_flag(app, client):
     e = 'matt@lp.com'
     authenticate(client, email=e)
     logout(client)
-    authenticate(client, email=e)
+    authenticate(client, email=e, headers={'X-Forwarded-For': '127.0.0.1'})
 
     with app.app_context():
         user = app.security.datastore.find_user(email=e)
         assert user.last_login_at is not None
         assert user.current_login_at is not None
         assert user.last_login_ip == 'untrackable'
-        assert user.current_login_ip == 'untrackable'
+        assert user.current_login_ip == '127.0.0.1'
         assert user.login_count == 2