import jcifs.dcerpc.DcerpcMessage;

@DisplayName("MsrpcSamrConnect2 Tests")
public class MsrpcSamrConnect2Test {

    private SamrPolicyHandle mockPolicyHandle;
    private String testServer;
    private int testAccess;

    @BeforeEach
    void setUp() {
        mockPolicyHandle = mock(SamrPolicyHandle.class);
        testServer = "testServer";
        testAccess = 0x02000000; // SAM_SERVER_CONNECT access right
    }

    @Test

     * @param access the desired access rights
     * @param policyHandle the policy handle to be populated
     */
    public MsrpcSamrConnect2(final String server, final int access, final SamrPolicyHandle policyHandle) {
        super(server, access, policyHandle);
        this.ptype = 0;
        this.flags = DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG;
    }

     * @param access the desired access rights
     * @param policyHandle the policy handle to be populated
     */
    public MsrpcSamrConnect4(final String server, final int access, final SamrPolicyHandle policyHandle) {
        super(server, 2, access, policyHandle);
        ptype = 0;
        flags = DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG;
    }

     * @param sid the security identifier of the domain
     * @throws IOException if an I/O error occurs during handle creation
     */
    public SamrDomainHandle(final DcerpcHandle handle, final SamrPolicyHandle policyHandle, final int access, final rpc.sid_t sid)
            throws IOException {
        this.handle = handle;
        final MsrpcSamrOpenDomain rpc = new MsrpcSamrOpenDomain(policyHandle, access, sid, this);

     * @param access the desired access rights
     * @param sid the security identifier of the domain
     * @param domainHandle the domain handle to be populated
     */
    public MsrpcSamrOpenDomain(final SamrPolicyHandle handle, final int access, final rpc.sid_t sid, final SamrDomainHandle domainHandle) {
        super(handle, access, sid, domainHandle);
        ptype = 0;
        flags = DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG;
    }

        DcerpcHandle handle = null;
        SamrPolicyHandle policyHandle = null;
        SamrDomainHandle domainHandle = null;
        final SID domsid = getDomainSid();

        synchronized (sid_cache) {
            try {
                handle = DcerpcHandle.getHandle("ncacn_np:" + authorityServerName + "[\\PIPE\\samr]", auth);
                policyHandle = new SamrPolicyHandle(handle, authorityServerName, 0x00000030);

     * @param sid the security identifier of the domain
     * @throws IOException if an I/O error occurs during handle creation
     */
    public SamrDomainHandle(final DcerpcHandle handle, final SamrPolicyHandle policyHandle, final int access, final rpc.sid_t sid)
            throws IOException {
        this.handle = handle;
        final MsrpcSamrOpenDomain rpc = new MsrpcSamrOpenDomain(policyHandle, access, sid, this);

import jcifs.dcerpc.rpc;
import jcifs.smb.SmbException;

@ExtendWith(MockitoExtension.class)
class SamrDomainHandleTest {

    @Mock
    private DcerpcHandle mockDcerpcHandle;
    @Mock
    private SamrPolicyHandle mockPolicyHandle;
    @Mock
    private rpc.sid_t mockSid;

    // Use ArgumentCaptor to capture the MsrpcSamrOpenDomain instance passed to sendrecv
    private ArgumentCaptor<MsrpcSamrOpenDomain> openDomainCaptor;