98 lines
2.6 KiB
C++
98 lines
2.6 KiB
C++
|
/*
|
||
|
|
* synergy -- mouse and keyboard sharing utility
|
||
|
|
* Copyright (C) 2013 Bolton Software Ltd.
|
||
|
|
*
|
||
|
|
* This package is free software; you can redistribute it and/or
|
||
|
|
* modify it under the terms of the GNU General Public License
|
||
|
|
* found in the file COPYING that should have accompanied this file.
|
||
|
|
*
|
||
|
|
* This package is distributed in the hope that it will be useful,
|
||
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
* GNU General Public License for more details.
|
||
|
|
*
|
||
|
|
* You should have received a copy of the GNU General Public License
|
||
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
#include "CCryptoStream.h"
|
||
|
|
#include "CMockStream.h"
|
||
|
|
#include "CMockEventQueue.h"
|
||
|
|
|
||
|
|
using ::testing::_;
|
||
|
|
using ::testing::Invoke;
|
||
|
|
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
void assertWrite(const void* in, UInt32 n);
|
||
|
|
UInt8 mockRead(void* out, UInt32 n);
|
||
|
|
|
||
|
|
TEST(CCryptoTests, write)
|
||
|
|
{
|
||
|
|
const UInt32 size = 4;
|
||
|
|
UInt8* buffer = new UInt8[size];
|
||
|
|
buffer[0] = 'D';
|
||
|
|
buffer[1] = 'K';
|
||
|
|
buffer[2] = 'D';
|
||
|
|
buffer[3] = 'N';
|
||
|
|
|
||
|
|
CMockEventQueue eventQueue;
|
||
|
|
CMockStream innerStream(eventQueue);
|
||
|
|
|
||
|
|
ON_CALL(innerStream, write(_, _)).WillByDefault(Invoke(assertWrite));
|
||
|
|
EXPECT_CALL(innerStream, write(_, _)).Times(1);
|
||
|
|
EXPECT_CALL(innerStream, getEventTarget()).Times(3);
|
||
|
|
EXPECT_CALL(eventQueue, removeHandlers(_)).Times(1);
|
||
|
|
EXPECT_CALL(eventQueue, adoptHandler(_, _, _)).Times(1);
|
||
|
|
EXPECT_CALL(eventQueue, removeHandler(_, _)).Times(1);
|
||
|
|
|
||
|
|
CCryptoStream cs(eventQueue, &innerStream);
|
||
|
|
cs.write(buffer, size);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(CCryptoTests, read)
|
||
|
|
{
|
||
|
|
CMockEventQueue eventQueue;
|
||
|
|
CMockStream innerStream(eventQueue);
|
||
|
|
|
||
|
|
ON_CALL(innerStream, read(_, _)).WillByDefault(Invoke(mockRead));
|
||
|
|
EXPECT_CALL(innerStream, read(_, _)).Times(1);
|
||
|
|
EXPECT_CALL(innerStream, getEventTarget()).Times(3);
|
||
|
|
EXPECT_CALL(eventQueue, removeHandlers(_)).Times(1);
|
||
|
|
EXPECT_CALL(eventQueue, adoptHandler(_, _, _)).Times(1);
|
||
|
|
EXPECT_CALL(eventQueue, removeHandler(_, _)).Times(1);
|
||
|
|
|
||
|
|
CCryptoStream cs(eventQueue, &innerStream);
|
||
|
|
|
||
|
|
const UInt32 size = 4;
|
||
|
|
UInt8* buffer = new UInt8[size];
|
||
|
|
cs.read(buffer, size);
|
||
|
|
|
||
|
|
EXPECT_EQ('D', buffer[0]);
|
||
|
|
EXPECT_EQ('K', buffer[1]);
|
||
|
|
EXPECT_EQ('D', buffer[2]);
|
||
|
|
EXPECT_EQ('N', buffer[3]);
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
assertWrite(const void* in, UInt32 n)
|
||
|
|
{
|
||
|
|
UInt8* buffer = static_cast<UInt8*>(const_cast<void*>(in));
|
||
|
|
EXPECT_EQ(55, buffer[0]);
|
||
|
|
EXPECT_EQ(142, buffer[1]);
|
||
|
|
EXPECT_EQ(189, buffer[2]);
|
||
|
|
EXPECT_EQ(237, buffer[3]);
|
||
|
|
}
|
||
|
|
|
||
|
|
UInt8
|
||
|
|
mockRead(void* out, UInt32 n)
|
||
|
|
{
|
||
|
|
UInt8* buffer = static_cast<UInt8*>(out);
|
||
|
|
buffer[0] = 55;
|
||
|
|
buffer[1] = 142;
|
||
|
|
buffer[2] = 189;
|
||
|
|
buffer[3] = 237;
|
||
|
|
return n;
|
||
|
|
}
|