@node-dlc/messaging/__tests__/messages/FundingInput.spec.ts

DLC Messaging Protocol

import { Sequence, Tx } from '@node-dlc/bitcoin';
import { StreamReader } from '@node-dlc/bufio';
import { expect } from 'chai';

import { DlcInput } from '../../lib/messages/DlcInput';
import { FundingInput } from '../../lib/messages/FundingInput';

describe('FundingInput', () => {
  describe('serialization/deserialization', () => {
    it('should deserialize and serialize FundingInput message', () => {
      const fundingInput = FundingInput.fromJSON({
        inputSerialId: 1,
        prevTx:
          '0200000001f58f85b356ad5bb5b6d0ef3eb863be8a6cb95e08e1e9e92885b4b22e7e51eb9d0000000000ffffffff01008c8647000000002200201234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef00000000',
        prevTxVout: 0,
        sequence: 4294967295,
        maxWitnessLen: 108,
        redeemScript: '',
      });

      const serialized = fundingInput.serialize();
      const deserialized = FundingInput.deserialize(serialized);

      expect(deserialized.inputSerialId).to.equal(fundingInput.inputSerialId);
      expect(deserialized.prevTxVout).to.equal(fundingInput.prevTxVout);
      expect(deserialized.sequence.value).to.equal(fundingInput.sequence.value);
      expect(deserialized.maxWitnessLen).to.equal(fundingInput.maxWitnessLen);
      expect(deserialized.redeemScript).to.deep.equal(
        fundingInput.redeemScript,
      );
    });

    it('should handle FundingInput with optional DlcInput', () => {
      const fundingInputWithDlc = FundingInput.fromJSON({
        inputSerialId: 1,
        prevTx:
          '0200000001f58f85b356ad5bb5b6d0ef3eb863be8a6cb95e08e1e9e92885b4b22e7e51eb9d0000000000ffffffff01008c8647000000002200201234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef00000000',
        prevTxVout: 0,
        sequence: 4294967295,
        maxWitnessLen: 108,
        redeemScript: '',
        dlcInput: {
          localFundPubkey:
            '023da092f6980e58d2c037173180e9a465476026ee50f96695963e8efe436f54eb',
          remoteFundPubkey:
            '025476c2e83188368da1ff3e292e7acafcdb3566bb0ad253f62fc70f07aeee6357',
          contractId:
            '1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef',
        },
      });

      // Test full serialize/deserialize
      const serialized = fundingInputWithDlc.serialize();
      const deserialized = FundingInput.deserialize(serialized);

      expect(deserialized.dlcInput).to.not.be.undefined;
      expect(deserialized.dlcInput!.localFundPubkey).to.deep.equal(
        fundingInputWithDlc.dlcInput!.localFundPubkey,
      );
      expect(deserialized.dlcInput!.remoteFundPubkey).to.deep.equal(
        fundingInputWithDlc.dlcInput!.remoteFundPubkey,
      );
      expect(deserialized.dlcInput!.contractId).to.deep.equal(
        fundingInputWithDlc.dlcInput!.contractId,
      );

      // Test body serialize/deserialize (used in DlcOffer)
      const serializedBody = fundingInputWithDlc.serializeBody();
      const deserializedBody = FundingInput.deserializeBody(serializedBody);

      expect(deserializedBody.dlcInput).to.not.be.undefined;
      expect(deserializedBody.dlcInput!.localFundPubkey).to.deep.equal(
        fundingInputWithDlc.dlcInput!.localFundPubkey,
      );
      expect(deserializedBody.dlcInput!.remoteFundPubkey).to.deep.equal(
        fundingInputWithDlc.dlcInput!.remoteFundPubkey,
      );
      expect(deserializedBody.dlcInput!.contractId).to.deep.equal(
        fundingInputWithDlc.dlcInput!.contractId,
      );
    });

    it('should handle FundingInput without DlcInput', () => {
      const fundingInputWithoutDlc = FundingInput.fromJSON({
        inputSerialId: 1,
        prevTx:
          '0200000001f58f85b356ad5bb5b6d0ef3eb863be8a6cb95e08e1e9e92885b4b22e7e51eb9d0000000000ffffffff01008c8647000000002200201234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef00000000',
        prevTxVout: 0,
        sequence: 4294967295,
        maxWitnessLen: 108,
        redeemScript: '',
      });

      // Test full serialize/deserialize
      const serialized = fundingInputWithoutDlc.serialize();
      const deserialized = FundingInput.deserialize(serialized);

      expect(deserialized.dlcInput).to.be.undefined;

      // Test body serialize/deserialize (used in DlcOffer)
      const serializedBody = fundingInputWithoutDlc.serializeBody();
      const deserializedBody = FundingInput.deserializeBody(serializedBody);

      expect(deserializedBody.dlcInput).to.be.undefined;
    });

    it('should use rust-dlc optional format (0x00/0x01 prefix)', () => {
      // Test without DlcInput - should have 0x00 byte at the end
      const fundingInputWithoutDlc = FundingInput.fromJSON({
        inputSerialId: 1,
        prevTx:
          '0200000001f58f85b356ad5bb5b6d0ef3eb863be8a6cb95e08e1e9e92885b4b22e7e51eb9d0000000000ffffffff01008c8647000000002200201234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef00000000',
        prevTxVout: 0,
        sequence: 4294967295,
        maxWitnessLen: 108,
        redeemScript: '',
      });

      const serializedBody = fundingInputWithoutDlc.serializeBody();
      // Last byte should be 0x00 indicating no DLC input
      expect(serializedBody[serializedBody.length - 1]).to.equal(0x00);

      // Test with DlcInput - should have 0x01 byte followed by DLC input data
      const fundingInputWithDlc = FundingInput.fromJSON({
        inputSerialId: 1,
        prevTx:
          '0200000001f58f85b356ad5bb5b6d0ef3eb863be8a6cb95e08e1e9e92885b4b22e7e51eb9d0000000000ffffffff01008c8647000000002200201234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef00000000',
        prevTxVout: 0,
        sequence: 4294967295,
        maxWitnessLen: 108,
        redeemScript: '',
        dlcInput: {
          localFundPubkey:
            '023da092f6980e58d2c037173180e9a465476026ee50f96695963e8efe436f54eb',
          remoteFundPubkey:
            '025476c2e83188368da1ff3e292e7acafcdb3566bb0ad253f62fc70f07aeee6357',
          contractId:
            '1234567890abcdef1234567890abcdef1234567890abcdef1234567890abcdef',
        },
      });

      const serializedBodyWithDlc = fundingInputWithDlc.serializeBody();

      // Find where the DLC input optional marker should be
      // It should be at: 8 (inputSerialId) + 1-9 (prevTx length) + prevTx.length + 4 (prevTxVout) + 4 (sequence) + 2 (maxWitnessLen) + 2 (redeemScript.length) + redeemScript.length
      const dlcInputMarkerIndex =
        serializedBodyWithDlc.length - (33 + 33 + 32 + 1); // DLC input data + marker (contractId is 32 bytes)
      expect(serializedBodyWithDlc[dlcInputMarkerIndex]).to.equal(0x01);
    });
  });

  describe('validate', () => {
    it('should ensure inputs are segwit', () => {
      const fundingInput = FundingInput.fromJSON({
        inputSerialId: 1,
        prevTx:
          '02000000000100c2eb0b00000000160014e70dcc9ffa7ff84c889c9e79b218708bae3bc95800000000', // has no inputs
        prevTxVout: 0,
        sequence: 4294967295,
        maxWitnessLen: 108,
        redeemScript: '',
      });

      expect(function () {
        fundingInput.validate();
      }).to.throw(Error);
    });
  });
});