All Downloads are FREE. Search and download functionalities are using the official Maven repository.

io.zeebe.test.util.BufferAssert Maven / Gradle / Ivy

There is a newer version: 0.16.4
Show newest version
/*
 * Copyright © 2017 camunda services GmbH ([email protected])
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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.
 */
package io.zeebe.test.util;

import static io.zeebe.util.buffer.BufferUtil.NO_WRAP;
import static io.zeebe.util.buffer.BufferUtil.bytesAsHexString;

import io.zeebe.util.buffer.BufferWriter;
import java.util.Arrays;
import org.agrona.DirectBuffer;
import org.agrona.MutableDirectBuffer;
import org.agrona.concurrent.UnsafeBuffer;
import org.assertj.core.api.AbstractAssert;

public class BufferAssert extends AbstractAssert {

  protected BufferAssert(DirectBuffer actual) {
    super(actual, BufferAssert.class);
  }

  public static BufferAssert assertThatBuffer(DirectBuffer buffer) {
    return new BufferAssert(buffer);
  }

  public BufferAssert hasBytes(byte[] expected, int position) {
    isNotNull();

    final byte[] actualBytes = new byte[expected.length];

    try {
      actual.getBytes(position, actualBytes, 0, actualBytes.length);
    } catch (final Exception e) {
      e.printStackTrace();
      failWithMessage(
          "Unable to read %d bytes from actual: %s", actualBytes.length, e.getMessage());
    }

    if (!Arrays.equals(expected, actualBytes)) {
      failWithMessage(
          "Expected byte array match bytes <%s> but was <%s>",
          bytesAsHexString(expected, NO_WRAP), bytesAsHexString(actualBytes, NO_WRAP));
    }

    return this;
  }

  public BufferAssert hasBytes(byte[] expected) {
    return hasBytes(expected, 0);
  }

  public BufferAssert hasBytes(DirectBuffer buffer, int offset, int length) {
    final byte[] bytes = new byte[length];
    buffer.getBytes(offset, bytes);
    return hasBytes(bytes);
  }

  public BufferAssert hasBytes(DirectBuffer buffer) {
    return hasBytes(buffer, 0, buffer.capacity());
  }

  public BufferAssert hasCapacity(int expectedCapacity) {
    isNotNull();

    if (expectedCapacity != actual.capacity()) {
      failWithMessage("Expected capacity " + expectedCapacity + " but was " + actual.capacity());
    }

    return this;
  }

  public BufferAssert hasBytes(final BufferWriter writer) {
    final MutableDirectBuffer buffer = new UnsafeBuffer(new byte[writer.getLength()]);
    writer.write(buffer, 0);

    return hasBytes(buffer);
  }
}




© 2015 - 2025 Weber Informatics LLC | Privacy Policy