From 3a6f877a9f79157424d5a86d05f9f1b7a98d1585 Mon Sep 17 00:00:00 2001
From: Michael Pollind <mpollind@gmail.com>
Date: Tue, 14 Dec 2021 23:05:13 -0800
Subject: [PATCH] chore: add first test

Signed-off-by: Michael Pollind <mpollind@gmail.com>
---
 .../Input/QtEventToAzInputMapper.cpp          |  8 ++---
 .../Input/QtEventToAzInputMapperTests.cpp     | 32 +++++++++++++++++++
 2 files changed, 36 insertions(+), 4 deletions(-)

diff --git a/Code/Framework/AzToolsFramework/AzToolsFramework/Input/QtEventToAzInputMapper.cpp b/Code/Framework/AzToolsFramework/AzToolsFramework/Input/QtEventToAzInputMapper.cpp
index 77f297b98c..7852a4a5d5 100644
--- a/Code/Framework/AzToolsFramework/AzToolsFramework/Input/QtEventToAzInputMapper.cpp
+++ b/Code/Framework/AzToolsFramework/AzToolsFramework/Input/QtEventToAzInputMapper.cpp
@@ -447,11 +447,11 @@ namespace AzToolsFramework
 
     void wrapCursorX(const QRect& rect, QPoint& point)
     {
-        if (rect.left() < point.x())
+        if (point.x() < rect.left())
         {
             point.setX(rect.right() - 1);
         }
-        else if (rect.right() > point.x())
+        else if (point.x() > rect.right())
         {
             point.setX(rect.left() + 1);
         }
@@ -459,11 +459,11 @@ namespace AzToolsFramework
 
     void wrapCursorY(const QRect& rect, QPoint& point)
     {
-        if (rect.top() < point.y())
+        if (point.y() < rect.top())
         {
             point.setY(rect.bottom() - 1);
         }
-        else if (rect.bottom() > point.y())
+        else if (point.y() > rect.bottom())
         {
             point.setY(rect.top() + 1);
         }
diff --git a/Code/Framework/AzToolsFramework/Tests/Input/QtEventToAzInputMapperTests.cpp b/Code/Framework/AzToolsFramework/Tests/Input/QtEventToAzInputMapperTests.cpp
index 4c813a4bdd..e645c365e2 100644
--- a/Code/Framework/AzToolsFramework/Tests/Input/QtEventToAzInputMapperTests.cpp
+++ b/Code/Framework/AzToolsFramework/Tests/Input/QtEventToAzInputMapperTests.cpp
@@ -44,6 +44,10 @@ namespace UnitTest
             QObject::connect(m_inputChannelMapper.get(), &AzToolsFramework::QtEventToAzInputMapper::InputChannelUpdated, m_rootWidget.get(),
                 [this]([[maybe_unused]] const AzFramework::InputChannel* inputChannel, QEvent* event)
                 {
+                    if(event == nullptr) {
+                        return;
+                    }
+
                     const QEvent::Type eventType = event->type();
 
                     if (eventType == QEvent::Type::MouseButtonPress ||
@@ -512,4 +516,32 @@ namespace UnitTest
             return info.param.m_az.GetName();
         }
     );
+
+    TEST_F(QtEventToAzInputMapperFixture, MouseWrapMouseViewportQtEventToAzInputMapperFixture) 
+    {
+        AzFramework::InputChannelNotificationBus::Handler::BusConnect();
+        m_captureAzEvents = false;
+
+        
+        const auto startPos = QPoint(WidgetSize.width()  - 2, WidgetSize.height() / 2);
+        MouseMove(m_rootWidget.get(), startPos, QPoint(0,0));
+        m_inputChannelMapper->SetCursorMode(AzToolsFramework::QtEventToAzInputMapper::CursorInputMode::CursorModeWrappedX);
+
+        const auto deltaPos = QPoint(200.0f, 0);
+        const auto expectedPosition = m_rootWidget->mapToGlobal(QPoint(200, WidgetSize.height() / 2));
+        const int iterations = 50;
+  
+        for(float i = 0; i < iterations; i++) {
+            MouseMove(m_rootWidget.get(), m_rootWidget->mapFromGlobal(QCursor::pos()), (deltaPos / iterations));
+        }
+        
+        QPointF endPosition = QCursor::pos();
+        EXPECT_NEAR(endPosition.x(), expectedPosition.x(), 5.0f);
+        EXPECT_NEAR(endPosition.y(), expectedPosition.y(), 5.0f);
+        
+        // cleanup
+        m_inputChannelMapper->SetCursorMode(AzToolsFramework::QtEventToAzInputMapper::CursorInputMode::CursorModeNone);
+        AzFramework::InputChannelNotificationBus::Handler::BusDisconnect();
+    }
+
 } // namespace UnitTest