Project-SDE-WP-Suite/tests/aggregates_check.py

#!/usr/bin/env python3
"""Do the counts come from the server? — B4 / T4.1.

The defect B4 names is not "the numbers are wrong". It is that they were derived
from the caller's own localStorage, so two people on the same project saw
different numbers and neither was told. A test that only checks the totals are
correct would have passed before this change, because on one browser with one
cache they were correct.

So this checks the thing that was actually broken:

  1. the same project reports the same aggregates to two different users
  2. the dashboard shows the SERVER's total even when the browser's own cache has
     been poisoned with a different one — which it cannot do if it is summing
     localStorage
  3. a failed aggregate request renders an explicit error and a retry, not a zero
     and not the last good answer
  4. the launcher's SOP status survives a poisoned cache the same way
  5. the aggregate response can be grouped by location without a schema change

Self-contained: throwaway SQLite, its own uvicorn, headless Edge or Chrome over
CDP, all torn down. Exit 0 all passed, 1 a failure, 2 could not run.
"""
import json
import os
import subprocess
import sys
import tempfile
import time
import urllib.error
import urllib.request

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

import cdp                                                     # noqa: E402
from browser_check import seed, start_server, chk, _PASS, _FAIL, _c   # noqa: E402


def api(base, path, token, method="GET", body=None):
    req = urllib.request.Request(base + path, method=method)
    req.add_header("Cookie", "wp_session=" + token)
    req.add_header("Accept", "application/json")
    data = None
    if body is not None:
        data = json.dumps(body).encode()
        req.add_header("Content-Type", "application/json")
    with urllib.request.urlopen(req, data, timeout=15) as r:
        return json.loads(r.read().decode() or "null")


def main():
    exe = cdp.find_browser()
    if not exe:
        print("no headless-capable browser found; set WP_BROWSER.")
        return 2

    tmpdir = tempfile.mkdtemp(prefix="wpsuite-aggregates-")
    db_path = os.path.join(tmpdir, "check.db")
    server = None
    try:
        tok = seed(db_path)
        port = cdp.free_port()
        base = "http://127.0.0.1:%d" % port
        server = start_server(port, db_path)
        if server is None:
            print("the test server would not start.")
            return 2
        print("\nAggregate counts — B4 / T4.1\nTarget: %s" % base)

        # Extra packages so the counts are not all the same number: one on hold,
        # one blocked by an open constraint, one closed, each with a location.
        for i, (num, status, constraint, loc, hours) in enumerate([
            ("WP03-HOLD", "Issue", "cleared", "B100 / L2 / P", "12"),
            ("WP04-GATE", "Draft", "open", "B100 / L2 / P", "8"),
            ("WP05-DONE", "Closed", "cleared", "B100 / L3 / Q", "20"),
        ]):
            api(base, "/api/wps", tok["root"], "POST", {
                "id": "wpX%d" % i, "project_id": "projA", "sop_id": "sopA",
                "number": num, "subject": num, "type": "Conduit Install",
                "status": status,
                "data": {"disciplines": ["Electrical"], "hours": hours,
                         "location": loc,
                         "constraints": [{"name": "Materials", "status": constraint,
                                          "comment": "waiting on delivery"}]},
            })

        print("\n1. the same project reports the same aggregates to two users")
        as_root = api(base, "/api/wps/metrics?project_id=projA", tok["root"])
        as_pat = api(base, "/api/wps/metrics?project_id=projA", tok["pat"])
        comparable = ["total", "release_ready", "on_hold", "overdue",
                      "est_hours", "actual_hours", "by_status", "by_discipline"]
        same = {k: as_root[k] for k in comparable} == {k: as_pat[k] for k in comparable}
        chk("root and pat get identical aggregates for projA", same,
            "root=%s pat=%s" % ({k: as_root[k] for k in comparable},
                                {k: as_pat[k] for k in comparable}))
        chk("the fixture is not degenerate (>=5 packages, a hold, a gate)",
            as_root["total"] >= 5 and as_root["on_hold"] >= 1 and len(as_root["gating"]) >= 1,
            "total=%s on_hold=%s gating=%s" % (as_root["total"], as_root["on_hold"],
                                               len(as_root["gating"])))
        chk("'mine' is per-user, so it is allowed to differ",
            "mine" in as_root and "mine" in as_pat)

        print("\n5. the aggregate can be grouped by location without a schema change")
        loc = as_root.get("by_location") or {}
        chk("by_location carries its dimensions", isinstance(loc.get("dimensions"), list)
            and len(loc["dimensions"]) >= 1, loc.get("dimensions"))
        groups = loc.get("groups") or []
        chk("by_location groups are keyed by those dimensions",
            bool(groups) and all(set(g["key"]) == set(loc["dimensions"]) for g in groups),
            [g.get("key") for g in groups[:3]])
        chk("each group carries its own rollup, not just a count",
            bool(groups) and all({"total", "release_ready", "on_hold", "by_status"} <= set(g)
                                 for g in groups),
            list(groups[0]) if groups else None)
        chk("the groups sum to the project total",
            sum(g["total"] for g in groups) == as_root["total"],
            "%s vs %s" % (sum(g["total"] for g in groups), as_root["total"]))

        browser = cdp.Browser(exe)
        page = browser.page()
        try:
            print("\n2. the dashboard shows the server's total, not the browser's")
            page.clear_cookies()
            page.set_cookie("wp_session", tok["root"])
            page.goto(base + "/wp-creation-index.html?project=projA")
            time.sleep(1.5)
            # Poison this browser's cache with a different number of packages than
            # the server has. If any displayed count still tracks localStorage, it
            # will report 2 and the server's total will not match.
            page.eval("""(() => {
              const fake = [
                {id:'fake1', number:'FAKE-1', subject:'not on the server', status:'Draft',
                 disciplines:['Electrical'], hours:'999', constraints:[]},
                {id:'fake2', number:'FAKE-2', subject:'also not', status:'Draft',
                 disciplines:['Electrical'], hours:'999', constraints:[]}
              ];
              localStorage.setItem('wp_iwp_v1::projA', JSON.stringify(fake));
              localStorage.setItem('wp_iwp_v1', JSON.stringify(fake));
              return true;
            })()""")
            page.goto(base + "/wp-creation-index.html?project=projA&view=dashboard")
            time.sleep(1.2)
            page.eval("typeof showDashboard==='function' && showDashboard()")
            for _ in range(30):
                ready = page.eval("!!document.querySelector('.dash-metric .dm-val')")
                if ready:
                    break
                time.sleep(0.3)
            shown = page.eval(
                "(()=>{const e=[...document.querySelectorAll('.dash-metric')]"
                ".find(x=>/Total WPs/i.test(x.textContent));"
                "return e?e.querySelector('.dm-val').textContent.trim():null})()")
            chk("dashboard 'Total WPs' equals the server total",
                str(shown) == str(as_root["total"]),
                "shown=%r server=%r (poisoned cache said 2)" % (shown, as_root["total"]))
            chk("...and is therefore not the poisoned cache's 2", str(shown) != "2", shown)

            print("\n3. a failed aggregate request is an error, not a zero")
            page.eval("""(() => {
              const real = window.fetch;
              window.fetch = function(u, o){
                if (String(u).indexOf('/api/wps/metrics') !== -1)
                  return Promise.reject(new Error('simulated outage'));
                return real.apply(this, arguments);
              };
              dashMetrics = null; dashMetricsErr = null;
              loadDashMetrics();
              return true;
            })()""")
            time.sleep(1.2)
            txt = page.eval("(document.getElementById('dashboard-view')||{}).textContent||''")
            chk("an explicit error panel is shown", "Counts unavailable" in txt, txt[:120])
            chk("...naming the failure", "simulated outage" in txt, txt[:160])
            chk("...offering a retry",
                page.eval("!!document.querySelector('#dashboard-view button')"))
            chk("no metric tiles are rendered alongside the error",
                page.eval("document.querySelectorAll('#dashboard-view .dash-metric').length") == 0)
            chk("the error region announces itself",
                page.eval("!!document.querySelector('#dashboard-view [role=alert]')"))

            print("\n4. the launcher's SOP status comes from the server too")
            page.goto(base + "/index.html")
            time.sleep(0.6)
            page.eval("""(() => {
              try {
                localStorage.setItem('wp_active_project_id', 'projA');
                localStorage.setItem('wp_suite_sop_complete::projA', '0');
                localStorage.removeItem('wp_suite_sop::projA');
              } catch(e){}
              return true;
            })()""")
            page.goto(base + "/index.html")
            for _ in range(30):
                s = page.eval("(document.querySelector('#card-sop .card-status')||{}).textContent||''")
                if s:
                    break
                time.sleep(0.3)
            status = page.eval(
                "(document.querySelector('#card-sop .card-status')||{}).textContent||''")
            chk("launcher reports the SOP complete despite a cache that says otherwise",
                "SOP complete" in status, "card said %r" % status)

            page.goto(base + "/index.html")
            time.sleep(0.4)
            page.eval("""(() => {
              const real = window.fetch;
              window.fetch = function(u, o){
                if (String(u).indexOf('/summary') !== -1)
                  return Promise.reject(new Error('simulated outage'));
                return real.apply(this, arguments);
              };
              return true;
            })()""")
            page.eval("typeof applyActiveProject==='function' && applyActiveProject()")
            time.sleep(1.0)
            status = page.eval(
                "(document.querySelector('#card-sop .card-status')||{}).textContent||''")
            chk("a failed status request says so on the card",
                "Could not check" in status, "card said %r" % status)
        finally:
            page.close()
            browser.close()
    except urllib.error.HTTPError as e:
        print("API error: %s %s" % (e.code, e.read()[:300]))
        return 2
    finally:
        if server:
            server.kill()
            try:
                server.wait(timeout=10)
            except subprocess.TimeoutExpired:
                pass
        try:
            from server.db import engine
            engine.dispose()
        except Exception:
            pass
        import shutil
        for _ in range(10):
            shutil.rmtree(tmpdir, ignore_errors=True)
            if not os.path.exists(tmpdir):
                break
            time.sleep(0.3)

    total = len(_PASS) + len(_FAIL)
    print("\n%s\n%d/%d checks passed." % ("-" * 54, len(_PASS), total))
    if _FAIL:
        for f in _FAIL:
            print("  - " + f)
        return 1
    print("\nResult: " + _c("ALL PASS — counts come from the server.", "32") + "\n")
    return 0


if __name__ == "__main__":
    sys.exit(main())