io.zeebe.test.util.BufferAssert Maven / Gradle / Ivy
/*
* 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