LoadShedConfigurationTest.cpp File Reference

#include <wangle/acceptor/LoadShedConfiguration.h>
#include <gtest/gtest.h>

Go to the source code of this file.

Functions
	TEST (LoadShedConfigurationTest, TestSettersAndGetters)

Function Documentation

TEST	(	LoadShedConfigurationTest	,
		TestSettersAndGetters
	)

Definition at line 23 of file LoadShedConfigurationTest.cpp.

References wangle::LoadShedConfiguration::addWhitelistAddr(), EXPECT_EQ, EXPECT_FALSE, EXPECT_TRUE, wangle::LoadShedConfiguration::getAcceptPauseOnAcceptorQueueSize(), wangle::LoadShedConfiguration::getAcceptResumeOnAcceptorQueueSize(), wangle::LoadShedConfiguration::getCpuUsageExceedWindowSize(), wangle::LoadShedConfiguration::getLoadUpdatePeriod(), wangle::LoadShedConfiguration::getMaxActiveConnections(), wangle::LoadShedConfiguration::getMaxConnections(), wangle::LoadShedConfiguration::getMaxCpuUsage(), wangle::LoadShedConfiguration::getMaxMemUsage(), wangle::LoadShedConfiguration::getMinCpuIdle(), wangle::LoadShedConfiguration::getMinFreeMem(), wangle::LoadShedConfiguration::getWhitelistAddrs(), wangle::LoadShedConfiguration::getWhitelistNetworks(), wangle::LoadShedConfiguration::isWhitelisted(), wangle::LoadShedConfiguration::setAcceptPauseOnAcceptorQueueSize(), wangle::LoadShedConfiguration::setAcceptResumeOnAcceptorQueueSize(), wangle::LoadShedConfiguration::setCpuUsageExceedWindowSize(), wangle::LoadShedConfiguration::setLoadUpdatePeriod(), wangle::LoadShedConfiguration::setMaxActiveConnections(), wangle::LoadShedConfiguration::setMaxConnections(), wangle::LoadShedConfiguration::setMaxCpuUsage(), wangle::LoadShedConfiguration::setMaxMemUsage(), wangle::LoadShedConfiguration::setMinCpuIdle(), wangle::LoadShedConfiguration::setMinFreeMem(), wangle::LoadShedConfiguration::setWhitelistAddrs(), and wangle::LoadShedConfiguration::setWhitelistNetworks().

                                                       {
  LoadShedConfiguration lsc;

  lsc.setMaxConnections(10);
  EXPECT_EQ(10, lsc.getMaxConnections());

  lsc.setMaxActiveConnections(20);
  EXPECT_EQ(20, lsc.getMaxActiveConnections());

  EXPECT_EQ(0, lsc.getAcceptPauseOnAcceptorQueueSize());
  lsc.setAcceptPauseOnAcceptorQueueSize(40);
  EXPECT_EQ(40, lsc.getAcceptPauseOnAcceptorQueueSize());

  EXPECT_EQ(0, lsc.getAcceptResumeOnAcceptorQueueSize());
  lsc.setAcceptResumeOnAcceptorQueueSize(50);
  EXPECT_EQ(50, lsc.getAcceptResumeOnAcceptorQueueSize());

  lsc.setMinFreeMem(30);
  EXPECT_EQ(30, lsc.getMinFreeMem());

  lsc.setMaxMemUsage(0.1);
  EXPECT_EQ(0.1, lsc.getMaxMemUsage());

  lsc.setMaxCpuUsage(0.2);
  EXPECT_EQ(0.2, lsc.getMaxCpuUsage());

  lsc.setMinCpuIdle(0.03);
  EXPECT_EQ(0.03, lsc.getMinCpuIdle());

  EXPECT_EQ(0, lsc.getCpuUsageExceedWindowSize());
  lsc.setCpuUsageExceedWindowSize(12);
  EXPECT_EQ(12, lsc.getCpuUsageExceedWindowSize());

  lsc.setLoadUpdatePeriod(std::chrono::milliseconds(1200));
  EXPECT_EQ(std::chrono::milliseconds(1200), lsc.getLoadUpdatePeriod());

  LoadShedConfiguration::AddressSet addressSet = {
    folly::SocketAddress("127.0.0.1", 1100),
    folly::SocketAddress("127.0.0.2", 1200),
    folly::SocketAddress("127.0.0.3", 1300),
  };
  lsc.setWhitelistAddrs(addressSet);

  EXPECT_EQ(addressSet, lsc.getWhitelistAddrs());
  EXPECT_TRUE(lsc.isWhitelisted(folly::SocketAddress("127.0.0.1", 1100)));
  EXPECT_TRUE(lsc.isWhitelisted(folly::SocketAddress("127.0.0.2", 1200)));
  EXPECT_TRUE(lsc.isWhitelisted(folly::SocketAddress("127.0.0.3", 1300)));
  EXPECT_FALSE(lsc.isWhitelisted(folly::SocketAddress("127.0.0.4", 1400)));
  lsc.addWhitelistAddr(folly::StringPiece("127.0.0.4"));
  EXPECT_TRUE(lsc.isWhitelisted(folly::SocketAddress("127.0.0.4", 0)));

  LoadShedConfiguration::NetworkSet networkSet = {
    NetworkAddress(folly::SocketAddress("127.0.0.5", 1500), 28),
    NetworkAddress(folly::SocketAddress("127.0.0.6", 1600), 24),
  };
  lsc.setWhitelistNetworks(networkSet);
  EXPECT_EQ(networkSet, lsc.getWhitelistNetworks());
  EXPECT_TRUE(lsc.isWhitelisted(folly::SocketAddress("127.0.0.5", 1500)));
  EXPECT_TRUE(lsc.isWhitelisted(folly::SocketAddress("127.0.0.6", 1300)));
  EXPECT_FALSE(lsc.isWhitelisted(folly::SocketAddress("10.0.0.7", 1700)));
  lsc.addWhitelistAddr(folly::StringPiece("10.0.0.7/20"));
  EXPECT_TRUE(lsc.isWhitelisted(folly::SocketAddress("10.0.0.7", 0)));
}