test_workflow_generator.h

Go to the documentation of this file.

// test_workflow_generator.h
// Converts natural language prompts into GUI automation workflows
 
#ifndef YAZE_CLI_SERVICE_TEST_WORKFLOW_GENERATOR_H
#define YAZE_CLI_SERVICE_TEST_WORKFLOW_GENERATOR_H
 
#include "absl/status/statusor.h"
 
#include <string>
#include <vector>
 
namespace yaze {
namespace cli {
 
enum class TestStepType {
  kClick,       // Click a button or element
  kType,        // Type text into an input
  kWait,        // Wait for a condition
  kAssert,      // Assert a condition is true
  kScreenshot   // Capture a screenshot
};
 
struct TestStep {
  TestStepType type;
  std::string target;      // Widget/element target (e.g., "button:Overworld")
  std::string text;        // Text to type (for kType steps)
  std::string condition;   // Condition to wait for or assert
  int timeout_ms = 5000;   // Timeout for wait operations
  bool clear_first = false; // Clear text before typing
  
  std::string ToString() const;
};
 
struct TestWorkflow {
  std::string description;
  std::vector<TestStep> steps;
  
  std::string ToString() const;
};
 
class TestWorkflowGenerator {
 public:
  TestWorkflowGenerator() = default;
  
  absl::StatusOr<TestWorkflow> GenerateWorkflow(const std::string& prompt);
  
 private:
  // Pattern matchers for different prompt types
  bool MatchesOpenEditor(const std::string& prompt, std::string* editor_name);
  bool MatchesOpenAndVerify(const std::string& prompt, std::string* editor_name);
  bool MatchesTypeInput(const std::string& prompt, std::string* input_name,
                        std::string* text);
  bool MatchesClickButton(const std::string& prompt, std::string* button_name);
  bool MatchesMultiStep(const std::string& prompt);
  
  // Workflow builders
  TestWorkflow BuildOpenEditorWorkflow(const std::string& editor_name);
  TestWorkflow BuildOpenAndVerifyWorkflow(const std::string& editor_name);
  TestWorkflow BuildTypeInputWorkflow(const std::string& input_name,
                                      const std::string& text);
  TestWorkflow BuildClickButtonWorkflow(const std::string& button_name);
  
  // Helper to normalize editor names (e.g., "overworld" → "Overworld Editor")
  std::string NormalizeEditorName(const std::string& name);
};
 
}  // namespace cli
}  // namespace yaze
 
#endif  // YAZE_CLI_SERVICE_TEST_WORKFLOW_GENERATOR_H