1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49

ash / assistant / test / mocked_assistant_interaction.h [blame]

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

#ifndef ASH_ASSISTANT_TEST_MOCKED_ASSISTANT_INTERACTION_H_
#define ASH_ASSISTANT_TEST_MOCKED_ASSISTANT_INTERACTION_H_

#include <memory>
#include <string>
#include "ash/assistant/test/test_assistant_service.h"
#include "base/memory/raw_ptr.h"

namespace ash {

class AssistantTestApi;

// Builder for an Assistant interaction.
// It contains methods to set the query and the response.
// The interaction will be auto submitted in the destructor.
class MockedAssistantInteraction {
 public:
  using Resolution = InteractionResponse::Resolution;

  MockedAssistantInteraction(AssistantTestApi* test_api,
                             TestAssistantService* service);
  ~MockedAssistantInteraction();
  MockedAssistantInteraction(MockedAssistantInteraction&&);
  MockedAssistantInteraction& operator=(MockedAssistantInteraction&&);

  MockedAssistantInteraction& WithQuery(const std::string& text_query);
  MockedAssistantInteraction& WithTextResponse(
      const std::string& text_response);
  MockedAssistantInteraction& WithSuggestionChip(const std::string& text);
  MockedAssistantInteraction& WithResolution(Resolution);

 private:
  void Submit();

  raw_ptr<AssistantTestApi> test_api_;
  raw_ptr<TestAssistantService> service_;
  std::unique_ptr<InteractionResponse> response_;

  std::string query_ = "<fake-query>";
  Resolution resolution_ = Resolution::kNormal;
};

}  // namespace ash

#endif  // ASH_ASSISTANT_TEST_MOCKED_ASSISTANT_INTERACTION_H_