netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Additionally expect the custom tos/tclass in TOS_TCLASS_EXPECT_DEFAULT case.

Browse Source 
PiperOrigin-RevId: 565458888
This commit is contained in:
Ayush Ranjan
2023-09-14 13:25:12 -07:00
committed by gVisor bot
parent ad68fe1d68
commit c0a9bb7b3a
1 changed files with 9 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/syscalls/linux/raw_socket.cc
+9 -13
View File
@@ -23,8 +23,7 @@
#include <sys/types.h>
#include <unistd.h>
#include <algorithm>
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "test/syscalls/linux/ip_socket_test_util.h"
#include "test/syscalls/linux/unix_domain_socket_test_util.h"
@@ -41,6 +40,7 @@ namespace testing {
namespace {
using ::testing::AnyOf;
using ::testing::IsNull;
using ::testing::NotNull;
@@ -1224,12 +1224,10 @@ TEST(RawSocketTest, ReceiveTOS) {
if (const char* val = getenv("TOS_TCLASS_EXPECT_DEFAULT");
val != nullptr && strcmp(val, "1") == 0) {
// TODO(https://issuetracker.google.com/issues/217448626): As of writing, it
// seems like at least one Linux environment does not allow setting a custom
// TOS. In this case, we expect the default instead of the TOS that was set
// above.
EXPECT_EQ(recv_buf.ip.tos, 0u);
EXPECT_EQ(recv_tos, 0u);
// TODO(b/217448626): At least one Linux environment does not allow setting
// a custom TOS. In this case, we additionally accept the default.
EXPECT_THAT(recv_buf.ip.tos, AnyOf(kArbitraryTOS, 0u));
EXPECT_THAT(recv_tos, AnyOf(kArbitraryTOS, 0u));
} else {
EXPECT_EQ(recv_buf.ip.tos, static_cast<uint8_t>(kArbitraryTOS));
EXPECT_EQ(recv_tos, kArbitraryTOS);
@@ -1277,11 +1275,9 @@ TEST(RawSocketTest, ReceiveTClass) {
if (const char* val = getenv("TOS_TCLASS_EXPECT_DEFAULT");
val != nullptr && strcmp(val, "1") == 0) {
// TODO(https://issuetracker.google.com/issues/217448626): As of writing, it
// seems like at least one Linux environment does not allow setting a custom
// TCLASS. In this case, we expect the default instead of the TCLASS that
// was set above.
EXPECT_EQ(recv_tclass, 0);
// TODO(b/217448626): At least one Linux environment does not allow setting
// a custom TCLASS. In this case, we additionally accept the default.
EXPECT_THAT(recv_tclass, AnyOf(kArbitraryTClass, 0));
} else {
EXPECT_EQ(recv_tclass, kArbitraryTClass);
}