Make FIFA17 roster hostname configurable

fifa17-recon/tools/test_roster_host_config.py

@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+"""Standalone contract test for Blaze roster-host advertisement."""
+
+import importlib
+import os
+import sys
+
+TOOLS = os.path.dirname(os.path.abspath(__file__))
+if TOOLS not in sys.path:
+    sys.path.insert(0, TOOLS)
+
+ADVERTISE = "192.0.2.10"
+DNS_HOST = "winter15.gosredirector.ea.com:8081"
+
+
+def assert_roster_config(blaze, host):
+    config = dict(blaze.OSDK_ROSTER)
+    assert blaze.ROSTER_HOST == host
+    assert config["ROSTERUPDATE_URL"] == (
+        f"https://{host}/fifa17/fut/rosterupdate.xml"
+    )
+    assert config["ROSTER_URL"] == f"https://{host}/fifa17/roster/"
+    assert config["ROSTER_VER"] == "0"
+    assert config["ROSTER_CSUM"] == ""
+
+
+def main():
+    old_advertise = os.environ.get("OPENFUT_ADVERTISE")
+    old_roster_host = os.environ.get("OPENFUT_ROSTER_HOST")
+    try:
+        os.environ["OPENFUT_ADVERTISE"] = ADVERTISE
+        os.environ.pop("OPENFUT_ROSTER_HOST", None)
+
+        import blaze_responder_v3b as blaze
+
+        blaze = importlib.reload(blaze)
+        assert_roster_config(blaze, f"{ADVERTISE}:8081")
+
+        os.environ["OPENFUT_ROSTER_HOST"] = DNS_HOST
+        blaze = importlib.reload(blaze)
+        assert_roster_config(blaze, DNS_HOST)
+
+        os.environ["OPENFUT_ROSTER_HOST"] = ""
+        blaze = importlib.reload(blaze)
+        assert_roster_config(blaze, f"{ADVERTISE}:8081")
+    finally:
+        if old_advertise is None:
+            os.environ.pop("OPENFUT_ADVERTISE", None)
+        else:
+            os.environ["OPENFUT_ADVERTISE"] = old_advertise
+        if old_roster_host is None:
+            os.environ.pop("OPENFUT_ROSTER_HOST", None)
+        else:
+            os.environ["OPENFUT_ROSTER_HOST"] = old_roster_host
+
+    print("PASS: roster host defaults, override, and URLs")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())