65 lines
1.9 KiB
C++
65 lines
1.9 KiB
C++
/*
|
|
* Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License").
|
|
* You may not use this file except in compliance with the License.
|
|
* A copy of the License is located at
|
|
*
|
|
* http://aws.amazon.com/apache2.0/
|
|
*
|
|
* or in the "license" file accompanying this file. This file 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 "AVSCommon/Utils/Common/TestableAttachmentWriter.h"
|
|
#include "AVSCommon/Utils/Common/Common.h"
|
|
|
|
namespace alexaClientSDK {
|
|
namespace avsCommon {
|
|
namespace utils {
|
|
|
|
using namespace avsCommon::avs::attachment;
|
|
using namespace avsCommon::utils::sds;
|
|
|
|
TestableAttachmentWriter::TestableAttachmentWriter(
|
|
std::shared_ptr<InProcessSDS> dummySDS,
|
|
std::unique_ptr<AttachmentWriter> writer) :
|
|
InProcessAttachmentWriter{dummySDS},
|
|
m_writer{std::move(writer)},
|
|
m_hasWriteBeenInvoked{false} {
|
|
}
|
|
|
|
std::size_t TestableAttachmentWriter::write(
|
|
const void* buf,
|
|
std::size_t numBytes,
|
|
WriteStatus* writeStatus,
|
|
std::chrono::milliseconds timeout) {
|
|
bool simulatePause = false;
|
|
|
|
if (!m_hasWriteBeenInvoked) {
|
|
simulatePause = true;
|
|
m_hasWriteBeenInvoked = true;
|
|
} else {
|
|
// Let's throw a random number, and simulate a pause if it's 1!
|
|
simulatePause = (generateRandomNumber(0, 1) == 1);
|
|
}
|
|
|
|
if (simulatePause) {
|
|
*writeStatus = WriteStatus::OK_BUFFER_FULL;
|
|
return 0;
|
|
}
|
|
|
|
// Otherwise, let's have the encapsulated writer do the actual work.
|
|
return m_writer->write(buf, numBytes, writeStatus);
|
|
}
|
|
|
|
void TestableAttachmentWriter::close() {
|
|
m_writer->close();
|
|
}
|
|
|
|
} // namespace utils
|
|
} // namespace avsCommon
|
|
} // namespace alexaClientSDK
|