content: // Copyright (c) 2011-present, Facebook, Inc.  All rights reserved.
  //  This source code is licensed under both the GPLv2 (found in the
  //  COPYING file in the root directory) and Apache 2.0 License
  //  (found in the LICENSE.Apache file in the root directory).
  package org.rocksdb;
  
  import org.junit.ClassRule;
  import org.junit.Test;
  
  import static org.assertj.core.api.Assertions.assertThat;
  
  public class RateLimiterTest {
  
    @ClassRule
    public static final RocksMemoryResource rocksMemoryResource =
        new RocksMemoryResource();
  
    @Test
    public void setBytesPerSecond() {
      try(final RateLimiter rateLimiter =
              new RateLimiter(1000, 100 * 1000, 1)) {
        rateLimiter.setBytesPerSecond(2000);
      }
    }
  
    @Test
    public void getSingleBurstBytes() {
      try(final RateLimiter rateLimiter =
              new RateLimiter(1000, 100 * 1000, 1)) {
        assertThat(rateLimiter.getSingleBurstBytes()).isEqualTo(100);
      }
    }
  
    @Test
    public void getTotalBytesThrough() {
      try(final RateLimiter rateLimiter =
              new RateLimiter(1000, 100 * 1000, 1)) {
        assertThat(rateLimiter.getTotalBytesThrough()).isEqualTo(0);
      }
    }
  
    @Test
    public void getTotalRequests() {
      try(final RateLimiter rateLimiter =
              new RateLimiter(1000, 100 * 1000, 1)) {
        assertThat(rateLimiter.getTotalRequests()).isEqualTo(0);
      }
    }
  }
  