/*
 * Copyright (C) 2016 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <general_test/test.h>

#include <shared/abort.h>
#include <shared/macros.h>
#include <shared/send_message.h>
#include <shared/time_util.h>
#include <cinttypes>

#include <chre/util/nanoapp/log.h>

#include "chre_api/chre.h"

#define LOG_TAG "[Test]"

using nanoapp_testing::sendInternalFailureToHost;

namespace general_test {

Test::Test(uint32_t minSupportedVersion)
    : mApiVersion(chreGetApiVersion()),
      mIsSupported(mApiVersion >= minSupportedVersion) {}

void Test::testSetUp(uint32_t messageSize, const void *message) {
  if (mIsSupported) {
    setUp(messageSize, message);
  } else {
    sendMessageToHost(nanoapp_testing::MessageType::kSkipped);
  }
}

void Test::testHandleEvent(uint32_t senderInstanceId, uint16_t eventType,
                           const void *eventData) {
  if (mIsSupported) {
    handleEvent(senderInstanceId, eventType, eventData);
  }
}

void Test::unexpectedEvent(uint16_t eventType) {
  uint32_t localEvent = eventType;
  EXPECT_FAIL_RETURN("Test received unexpected event:", &localEvent);
}

void Test::validateChreAsyncResult(const chreAsyncResult *result,
                                   const chreAsyncRequest &request) {
  if (!result->success) {
    EXPECT_FAIL_RETURN_UINT8("chre async result error: ", result->errorCode);
  }
  if (result->success && result->errorCode != CHRE_ERROR_NONE) {
    EXPECT_FAIL_RETURN_UINT8(
        "Request was successfully processed, but got errorCode: ",
        result->errorCode);
  }
  if (result->reserved != 0) {
    EXPECT_FAIL_RETURN_UINT8("reserved should be 0, got: ", result->reserved);
  }
  if (result->cookie != request.cookie) {
    LOGE("Request cookie is %p, got %p", request.cookie, result->cookie);
    EXPECT_FAIL_RETURN("Request cookie mismatch");
  }
  if (result->requestType != request.requestType) {
    LOGE("Request requestType is %d, got %d", request.requestType,
         result->requestType);
    EXPECT_FAIL_RETURN("Request requestType mismatch");
  }
  if (chreGetTime() - request.requestTimeNs > request.timeoutNs) {
    uint32_t time =
        request.timeoutNs / nanoapp_testing::kOneSecondInNanoseconds;
    EXPECT_FAIL_RETURN("Did not receive chreWifiAsyncEvent within time (sec): ",
                       &time);
  }
}

const void *Test::getMessageDataFromHostEvent(
    uint32_t senderInstanceId, uint16_t eventType, const void *eventData,
    nanoapp_testing::MessageType expectedMessageType,
    uint32_t expectedMessageSize) {
  if (senderInstanceId != CHRE_INSTANCE_ID) {
    sendInternalFailureToHost("Unexpected sender ID:", &senderInstanceId);
  }
  if (eventType != CHRE_EVENT_MESSAGE_FROM_HOST) {
    unexpectedEvent(eventType);
  }
  if (eventData == nullptr) {
    sendInternalFailureToHost("NULL eventData given");
  }
  auto data = static_cast<const chreMessageFromHostData *>(eventData);
  if (data->reservedMessageType != uint32_t(expectedMessageType)) {
    sendInternalFailureToHost("Unexpected reservedMessageType:",
                              &(data->reservedMessageType));
  }
  if (data->messageSize != expectedMessageSize) {
    sendInternalFailureToHost("Unexpected messageSize:", &(data->messageSize));
  }
  return data->message;
}

}  // namespace general_test