scuttlebutt/src/test/java/org/apache/tuweni/scuttlebutt/InviteTest.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE
 * file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file
 * to You 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 org.apache.tuweni.scuttlebutt;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.fail;
import static org.junit.jupiter.api.Assumptions.assumeTrue;

import org.apache.tuweni.crypto.sodium.Signature;
import org.apache.tuweni.crypto.sodium.Sodium;

import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;

class InviteTest {

  @BeforeAll
  static void checkAvailable() {
    assumeTrue(Sodium.isAvailable(), "Sodium native library is not available");
  }

  @Test
  void invalidPort() {
    assertThrows(
        IllegalArgumentException.class,
        () -> new Invite("localhost", -1, Identity.random(), Signature.Seed.random()));
  }

  @Test
  void testToString() {
    Identity identity = Identity.random();
    Signature.Seed seed = Signature.Seed.random();

    Invite invite = new Invite("localhost", 8008, identity, seed);
    assertEquals(
        "localhost:8008:"
            + "@"
            + identity.publicKeyAsBase64String()
            + "."
            + identity.curveName()
            + "~"
            + seed.bytes().toBase64String(),
        invite.toString());
  }

  @Test
  void testParseFromCanonicalValid() {
    String testInvite =
        "fake.address.com:8009:@MS/HpeAess0EGruiZjfnc+x+FkPq7qoMqSD4SdvTCtM=.ed25519~IJubWEcZM6usWncF/Lu26CyI3ZiovcHjh9+kBI1hiKI=";

    try {
      Invite invite = Invite.fromCanonicalForm(testInvite);

      assertEquals(invite.host(), "fake.address.com");
      assertEquals(invite.port(), 8009);
      assertEquals(invite.identity().publicKeyAsBase64String(), "MS/HpeAess0EGruiZjfnc+x+FkPq7qoMqSD4SdvTCtM=");
      assertEquals(invite.seedKey().bytes().toBase64String(), "IJubWEcZM6usWncF/Lu26CyI3ZiovcHjh9+kBI1hiKI=");

      assertEquals(invite.toCanonicalForm(), testInvite);
    } catch (MalformedInviteCodeException malformedInviteCodeException) {
      fail("Exception while parsing into canonical form: " + malformedInviteCodeException.getMessage());
    }
  }

  @Test
  void testParseFromCanonicalMissingHost() {
    String testInvite =
        ":@MS/HpeAess0EGruiZjfnc+x+FkPq7qoMqSD4SdvTCtM=.ed25519~IJubWEcZM6usWncF/Lu26CyI3ZiovcHjh9+kBI1hiKI=";

    try {
      Invite.fromCanonicalForm(testInvite);
      fail("Exception expected when host missing from invite code.");
    } catch (MalformedInviteCodeException malformedInviteCodeException) {

    }
  }

  @Test
  void testParseFromCanonicalMissingSeed() {
    String testInvite = "fake.address.com:8009:@MS/HpeAess0EGruiZjfnc+x+FkPq7qoMqSD4SdvTCtM=.ed25519";

    try {
      Invite.fromCanonicalForm(testInvite);
      fail("Exception expected when seed missing from invite code.");
    } catch (MalformedInviteCodeException malformedInviteCodeException) {

    }

  }

}