#ifndef TESTS_SRC_FSFW_TESTS_UNIT_DEVICEHANDLER_DEVICEHANDLERMOCK_H_
#define TESTS_SRC_FSFW_TESTS_UNIT_DEVICEHANDLER_DEVICEHANDLERMOCK_H_

#include <fsfw/devicehandlers/DeviceHandlerBase.h>
#include <fsfw/introspection/Enum.h>
#include <fsfw/action/TemplateAction.h>

FSFW_ENUM(MockCommands, ActionId_t, ((SIMPLE_COMMAND, 1, "Simple Command")) ((PERIODIC_REPLY, 2, "Periodic Reply")))

class DeviceHandlerMock;

class MockAction : public TemplateAction<DeviceHandlerMock, MockAction, MockCommands> {
  public:
  MockAction(DeviceHandlerMock *owner): TemplateAction(owner, MockCommands::SIMPLE_COMMAND) {

  }
};

class DeviceHandlerMock : public DeviceHandlerBase {
 public:

  static const uint8_t SIMPLE_COMMAND_DATA = 1;
  static const uint8_t PERIODIC_REPLY_DATA = 2;

  DeviceHandlerMock(object_id_t objectId, object_id_t deviceCommunication, CookieIF *comCookie,
                    FailureIsolationBase *fdirInstance);
  virtual ~DeviceHandlerMock();
  void changePeriodicReplyCountdown(uint32_t timeout);
  void changeSimpleCommandReplyCountdown(uint32_t timeout);
  void resetPeriodicReplyState();
  bool getPeriodicReplyReceived();
  ReturnValue_t enablePeriodicReply(DeviceCommandId_t replyId);
  ReturnValue_t disablePeriodicReply(DeviceCommandId_t replyId);

  ReturnValue_t handleAction(MockAction * action);

 protected:
  void doStartUp() override;
  void doShutDown() override;
  ReturnValue_t buildNormalDeviceCommand(DeviceCommandId_t *id) override;
  ReturnValue_t buildTransitionDeviceCommand(DeviceCommandId_t *id) override;
  ReturnValue_t buildCommandFromCommand(DeviceCommandId_t deviceCommand, const uint8_t *commandData,
                                        size_t commandDataLen) override;
  ReturnValue_t scanForReply(const uint8_t *start, size_t len, DeviceCommandId_t *foundId,
                             size_t *foundLen) override;
  ReturnValue_t interpretDeviceReply(DeviceCommandId_t id, const uint8_t *packet) override;
  void fillCommandAndReplyMap() override;
  uint32_t getTransitionDelayMs(Mode_t modeFrom, Mode_t modeTo) override;

 private:
  Countdown simpleCommandReplyTimeout = Countdown(1000);
  Countdown periodicReplyCountdown = Countdown(1000);

  uint8_t commandBuffer[1];

  bool periodicReplyReceived = false;

  MockAction mockAction = MockAction(this);
};

#endif /* TESTS_SRC_FSFW_TESTS_UNIT_DEVICEHANDLER_DEVICEHANDLERMOCK_H_ */