/*******************************************************************************
 * Copyright (c) 2017, Rockwell Automation, Inc.
 * All rights reserved.
 *
 ******************************************************************************/

#include <CppUTest/TestHarness.h>
#include <stdint.h>
#include <string.h>

extern "C" {

#include "cipconnectionobject.h"

}

TEST_GROUP(CipConnectionObject) {

};

TEST(CipConnectionObject, StateNonExistent) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 0;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateNonExistent, state);
}

TEST(CipConnectionObject, StateConfiguring) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 1;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateConfiguring, state);
}

TEST(CipConnectionObject, StateWaitingForConnectionID) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 2;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateWaitingForConnectionID, state);
}

TEST(CipConnectionObject, StateEstablished) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 3;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateEstablished, state);
}

TEST(CipConnectionObject, StateTimedOut) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 4;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateTimedOut, state);
}

TEST(CipConnectionObject, StateDeferredDelete) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 5;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateDeferredDelete, state);
}

TEST(CipConnectionObject, StateClosing) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 6;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateClosing, state);
}

TEST(CipConnectionObject, StateInvalid) {
  CipConnectionObject connection_object = { 0 };
  connection_object.state = 7;
  ConnectionObjectState state = GetConnectionObjectState(&connection_object);
  CHECK_EQUAL(kConnectionObjectStateInvalid, state);
}

TEST(CipConnectionObject, InstanceTypeInvalid) {
  CipConnectionObject connection_object = { 0 };
  connection_object.instance_type = 4;
  ConnectionObjectInstanceType type = GetConnectionObjectInstanceType(
    &connection_object);
  CHECK_EQUAL(kConnectionObjectInstanceTypeInvalid, type);
}

TEST(CipConnectionObject, InstanceTypeIExplicitMessaging) {
  CipConnectionObject connection_object = { 0 };
  connection_object.instance_type = 0;
  ConnectionObjectInstanceType type = GetConnectionObjectInstanceType(
    &connection_object);
  CHECK_EQUAL(kConnectionObjectInstanceTypeExplicitMessaging, type);
}

TEST(CipConnectionObject, InstanceTypeIO) {
  CipConnectionObject connection_object = { 0 };
  connection_object.instance_type = 1;
  ConnectionObjectInstanceType type = GetConnectionObjectInstanceType(
    &connection_object);
  CHECK_EQUAL(kConnectionObjectInstanceTypeIO, type);
}

TEST(CipConnectionObject, InstanceTypeCipBridged) {
  CipConnectionObject connection_object = { 0 };
  connection_object.instance_type = 2;
  ConnectionObjectInstanceType type = GetConnectionObjectInstanceType(
    &connection_object);
  CHECK_EQUAL(kConnectionObjectInstanceTypeCipBridged, type);
}

TEST(CipConnectionObject, TransportClassTriggerDirectionServer) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 0x80;
  ConnectionObjectTransportClassTriggerDirection direction =
    GetConnectionObjectTransportClassTriggerDirection(&connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerDirectionServer, direction);
}

TEST(CipConnectionObject, TransportClassTriggerDirectionClient) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 0x00;
  ConnectionObjectTransportClassTriggerDirection direction =
    GetConnectionObjectTransportClassTriggerDirection(&connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerDirectionClient, direction);
}

TEST(CipConnectionObject, TransportClassTriggerProductionTriggerInvalid) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 3 << 4;
  ConnectionObjectTransportClassTriggerProductionTrigger production_trigger =
    GetConnectionObjectTransportClassTriggerProductionTrigger(
      &connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerProductionTriggerInvalid,
              production_trigger);
}

TEST(CipConnectionObject, TransportClassTriggerProductionTriggerCyclic) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 0x00;
  ConnectionObjectTransportClassTriggerProductionTrigger production_trigger =
    GetConnectionObjectTransportClassTriggerProductionTrigger(
      &connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerProductionTriggerCyclic,
              production_trigger);
}

TEST(CipConnectionObject,
     TransportClassTriggerProductionTriggerChangeOfState) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 1 << 4;
  ConnectionObjectTransportClassTriggerProductionTrigger production_trigger =
    GetConnectionObjectTransportClassTriggerProductionTrigger(
      &connection_object);
  CHECK_EQUAL(
    kConnectionObjectTransportClassTriggerProductionTriggerChangeOfState,
    production_trigger);
}

TEST(CipConnectionObject,
     TransportClassTriggerProductionTriggerApplicationObject) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 2 << 4;
  ConnectionObjectTransportClassTriggerProductionTrigger production_trigger =
    GetConnectionObjectTransportClassTriggerProductionTrigger(
      &connection_object);
  CHECK_EQUAL(
    kConnectionObjectTransportClassTriggerProductionTriggerApplicationObject,
    production_trigger);
}

TEST(CipConnectionObject, TransportClassTriggerClassInvalid) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 5;
  ConnectionObjectTransportClassTriggerTransportClass transport_class =
    GetConnectionObjectTransportClassTriggerTransportClass(&connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerTransportClassInvalid,
              transport_class);
}

TEST(CipConnectionObject, TransportClassTriggerClass0) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 0;
  ConnectionObjectTransportClassTriggerTransportClass transport_class =
    GetConnectionObjectTransportClassTriggerTransportClass(&connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerTransportClass0,
              transport_class);
}

TEST(CipConnectionObject, TransportClassTriggerClass1) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 1;
  ConnectionObjectTransportClassTriggerTransportClass transport_class =
    GetConnectionObjectTransportClassTriggerTransportClass(&connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerTransportClass1,
              transport_class);
}

TEST(CipConnectionObject, TransportClassTriggerClass2) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 2;
  ConnectionObjectTransportClassTriggerTransportClass transport_class =
    GetConnectionObjectTransportClassTriggerTransportClass(&connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerTransportClass2,
              transport_class);
}

TEST(CipConnectionObject, TransportClassTriggerClass3) {
  CipConnectionObject connection_object = { 0 };
  connection_object.transport_class_trigger = 3;
  ConnectionObjectTransportClassTriggerTransportClass transport_class =
    GetConnectionObjectTransportClassTriggerTransportClass(&connection_object);
  CHECK_EQUAL(kConnectionObjectTransportClassTriggerTransportClass3,
              transport_class);
}