base/test/android/javatests/src/org/chromium/base/test/transit/TrafficControl.java

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.base.test.transit;

import org.chromium.base.test.transit.ConditionalState.Phase;

import java.util.ArrayList;
import java.util.List;

/** Keeps track of all existing TransitStations and which one is active. */
public class TrafficControl {
    private static final List<TransitStation> sAllStations = new ArrayList<>();

    private static TransitStation sActiveStation;

    static void notifyCreatedStation(TransitStation station) {
        sAllStations.add(station);
    }

    static void notifyActiveStationChanged(TransitStation newActiveStation) {
        assert newActiveStation.getPhase() == Phase.ACTIVE : "New active Station must be ACTIVE";
        if (sActiveStation != null) {
            assert sActiveStation.getPhase() != Phase.ACTIVE
                    : "Previously active station was not ACTIVE";
        }
        sActiveStation = newActiveStation;
    }

    static List<TransitStation> getAllStations() {
        return sAllStations;
    }

    static TransitStation getActiveStation() {
        return sActiveStation;
    }
}