github/azure-sdk-for-cpp
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fixed AV in attestation tests (#4480)

Browse Source 
* Fixed AV in attestation tests

* Fixed another test AV
This commit is contained in:
Larry Osterman 2023-04-06 11:03:39 -07:00 committed by GitHub
parent 779c268a7f
commit b4be76fec8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
sdk/attestation/azure-security-attestation/test/ut/tpmattestation_test.cpp
View File

@ -53,7 +53,10 @@ namespace Azure { namespace Security { namespace Attestation { namespace Test {
virtual void TearDown() override
{
// Reset the attestation policy for this instance back to the default.
m_adminClient->ResetAttestationPolicy(AttestationType::Tpm);
if (m_adminClient)
{
m_adminClient->ResetAttestationPolicy(AttestationType::Tpm);
}
// Make sure you call the base classes TearDown method to ensure recordings are made.
TestBase::TearDown();

15
sdk/core/azure-core-test/src/test_base.cpp
View File

@ -16,12 +16,15 @@ void Azure::Core::Test::TestBase::TearDown()
{
return;
}
if (m_testProxy->IsRecordMode())
if (m_testProxy)
{
m_testProxy->StopPlaybackRecord(TestMode::RECORD);
}
if (m_testProxy->IsPlaybackMode())
{
m_testProxy->StopPlaybackRecord(TestMode::PLAYBACK);
if (m_testProxy->IsRecordMode())
{
m_testProxy->StopPlaybackRecord(TestMode::RECORD);
}
if (m_testProxy->IsPlaybackMode())
{
m_testProxy->StopPlaybackRecord(TestMode::PLAYBACK);
}
}
}

7
sdk/core/azure-core/test/ut/transport_adapter_base_test.cpp
View File

@ -268,8 +268,11 @@ namespace Azure { namespace Core { namespace Test {
auto response = m_pipeline->Send(request, Azure::Core::Context::ApplicationContext);
checkResponseCode(response->GetStatusCode());
auto expectedResponseBodySize = std::stoull(response->GetHeaders().at("content-length"));
CheckBodyFromStream(*response, expectedResponseBodySize);
if (response->GetStatusCode() == Http::HttpStatusCode::Ok)
{
auto expectedResponseBodySize = std::stoull(response->GetHeaders().at("content-length"));
CheckBodyFromStream(*response, expectedResponseBodySize);
}
}
TEST_P(TransportAdapter, getChunkWithStream)