def test_multiple_flows_in_storage(): s = Memory() f = Flow("test") g = Flow("other") z = Flow("not") s.add_flow(f) s.add_flow(g) assert "test" in s assert "other" in s assert "not" not in s assert s.get_flow("test") is f assert s.get_flow("other") is g assert s.flows["test"] is f assert s.flows["other"] is g
def test_get_flow_returns_flow(): s = Memory() f = Flow("test") s.add_flow(f) runner = s.get_flow("test") assert runner is f
def test_get_flow_raises_if_flow_not_present(): s = Memory() with pytest.raises(ValueError): s.get_flow("test")