# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from google.protobuf import text_format as textpb
from recipe_engine import recipe_test_api


class TextProtoTestApi(recipe_test_api.RecipeTestApi):
    def spec_not_found(self, step_name="build_init"):
        """Returns mock step data that simulates a missing spec file."""
        return self.step_data(step_name, retcode=2)

    def spec_parse_error(self, step_name="build_init"):
        """Returns mock step data that simulates a spec file that fails to parse."""
        return self.step_data(
            step_name, stdout=self.m.raw_io.output("not a proto message"), retcode=0,
        )

    def spec_loaded_ok(self, message, step_name="build_init"):
        """Returns mock step data that simulates successfully fetching a spec file.

        Args:
          message: The spec as a protobuf Message.
        """
        return self.step_data(
            step_name,
            retcode=0,
            stdout=self.m.raw_io.output(textpb.MessageToString(message)),
        )