void testWriteSetupWireFormat() {
        // Given
        trans2GetDfsReferral = new Trans2GetDfsReferral(mockConfig, "test");
        byte[] dst = new byte[10];
        int dstIndex = 0;

        // When
        int bytesWritten = trans2GetDfsReferral.writeSetupWireFormat(dst, dstIndex);

        // Then
        assertEquals(2, bytesWritten);
        assertEquals(SmbComTransaction.TRANS2_GET_DFS_REFERRAL, dst[0]);

        this.took = took;
    }

    public int getTotalShards() {
        return totalShards;
    }

    public void setTotalShards(int totalShards) {
        this.totalShards = totalShards;
    }

    public int getSuccessfulShards() {
        return successfulShards;
    }

    public void setSuccessfulShards(int successfulShards) {
        this.successfulShards = successfulShards;
    }

	type Toy struct {
		ID        int
		Name      string
		OwnerID   int
		OwnerType string
	}
	type User struct {
		ID  int
		Cat struct {
			Name string
			Toy  Toy   `gorm:"polymorphic:Owner;"`
			Toys []Toy `gorm:"polymorphic:Owner;"`
		} `gorm:"embedded;embeddedPrefix:cat_"`
		Dog struct {
			ID     int
			Name   string
			UserID int
			Toy    Toy   `gorm:"polymorphic:Owner;"`

        // Given
        int sid = 0x1234;
        int resumeKey = 0x89ABCDEF; // LE expected: EF CD AB 89
        String name = "file.txt";
        Trans2FindNext2 next = new Trans2FindNext2(sid, resumeKey, name);

        int expectedLen = 2 + 2 + 2 + 4 + 2 + name.length() + 1;
        byte[] dst = new byte[expectedLen];

        // When
        int len = next.writeParametersWireFormat(dst, 0);

        final int length = string.length();
        if (length == 0) {
            return new byte[0];
        }
        final int pad = string.charAt(length - 2) == '=' ? 2 : string.charAt(length - 1) == '=' ? 1 : 0;
        final int size = length * 3 / 4 - pad;
        final byte[] buffer = new byte[size];
        int block;
        int i = 0;
        int index = 0;
        while (i < length) {

pkg p3, func BadHop(int, int, int) (bool, bool, *ThirdBase, *ThirdBase, error)
pkg p3, method (*ThirdBase) GoodPlayer() (int, int, int)

 */

public class NbtSocket extends Socket {

    private static final int SSN_SRVC_PORT = 139;
    private static final int BUFFER_SIZE = 512;
    private static final int DEFAULT_SO_TIMEOUT = 5000;

    private static LogStream log = LogStream.getInstance();

    private NbtAddress address;
    private Name calledName;
    private int soTimeout;

    /**
     * Constructs an unconnected NbtSocket.
     */

            public int getActiveShards() {
                return 20;
            }

            @Override
            public int getRelocatingShards() {
                return 0;
            }

            @Override
            public int getInitializingShards() {
                return 0;
            }

            @Override
            public int getUnassignedShards() {
                return 0;

  }

  @Benchmark
  void longEqualJava(int reps) {
    for (int i = 0; i < reps; ++i) {
      if (javaImpl.compare(ba1, ba2) != 0) {
        throw new Error(); // deoptimization
      }
    }
  }

  @Benchmark
  void longEqualUnsafe(int reps) {
    for (int i = 0; i < reps; ++i) {
      if (unsafeImpl.compare(ba1, ba2) != 0) {

        Field[] fields = SecurityInfo.class.getDeclaredFields();
        Set<Integer> values = new HashSet<>();

        for (Field field : fields) {
            if (field.getType() == int.class) {
                int value = field.getInt(null);
                assertTrue(values.add(value), "Duplicate value found for field: " + field.getName());
            }
        }
    }

    @Test