def METHOD_NAME(self, generated_rows):
    yaml = """
    - object: foo
      just_once: True
    - object: bar
    """
    generate(StringIO(yaml), stopping_criteria=StoppingCriteria("bar", 3))
    assert generated_rows.mock_calls == [
        mock.call("foo", {"id": 1}),
        mock.call("bar", {"id": 1}),
        mock.call("bar", {"id": 2}),
        mock.call("bar", {"id": 3}),
    ]