| 47 | self.assertEqual(ctx.client_feature_flags, client_feature_flags) |
| 48 | |
| 49 | def test_environ(self): |
| 50 | environ = {"one": "two", "three": "four"} |
| 51 | auth = auth_lib.AuthContext({}, environ) |
| 52 | client_feature_flags = {"someFlag": True} |
| 53 | req_context = context.from_environ(environ) |
| 54 | self.assertNotEqual(req_context.auth, auth) |
| 55 | self.assertEqual(req_context.remote_ip, None) |
| 56 | self.assertEqual(req_context.x_forwarded_for, ()) |
| 57 | self.assertEqual(req_context.client_feature_flags, {}) |
| 58 | |
| 59 | context.set_in_environ( |
| 60 | environ, |
| 61 | context.from_environ(environ).replace( |
| 62 | auth=auth, |
| 63 | remote_ip=REMOTE_IP, |
| 64 | x_forwarded_for=X_FORWARDED_FOR_IPS, |
| 65 | client_feature_flags=client_feature_flags, |
| 66 | ), |
| 67 | ) |
| 68 | self.assertEqual(environ["one"], "two") |
| 69 | self.assertEqual(environ["three"], "four") |
| 70 | req_context = context.from_environ(environ) |
| 71 | self.assertEqual(req_context.auth, auth) |
| 72 | self.assertEqual(req_context.remote_ip, REMOTE_IP) |
| 73 | self.assertEqual(req_context.x_forwarded_for, X_FORWARDED_FOR_IPS) |
| 74 | self.assertEqual(req_context.client_feature_flags, client_feature_flags) |
| 75 | |
| 76 | def test_replace(self): |
| 77 | environ1 = {"one": "two", "three": "four"} |