devp2p/src/test/kotlin/org/apache/tuweni/devp2p/v5/packet/UdpMessageTest.kt

/*
 * 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.devp2p.v5.packet

import org.apache.tuweni.bytes.Bytes
import org.junit.jupiter.api.Test
import org.junit.jupiter.api.assertThrows

class UdpMessageTest {

  @Test
  fun magicCreatesSha256OfDestNodeIdAndConstantString() {
    val destId = Bytes.fromHexString("0xA5CFE10E0EFC543CBE023560B2900E2243D798FAFD0EA46267DDD20D283CE13C")
    val expected = Bytes.fromHexString("0x98EB6D611291FA21F6169BFF382B9369C33D997FE4DC93410987E27796360640")

    val result = UdpMessage.magic(destId)

    assert(expected == result)
  }

  @Test
  fun tagHashesSourceAndDestNodeIdCorrectly() {
    val srcId = Bytes.fromHexString("0x98EB6D611291FA21F6169BFF382B9369C33D997FE4DC93410987E27796360640")
    val destId = Bytes.fromHexString("0xA5CFE10E0EFC543CBE023560B2900E2243D798FAFD0EA46267DDD20D283CE13C")
    val expected = Bytes.fromHexString("0xB7A0D7CA8BD37611315DA0882FF479DE14B442FD30AE0EFBE6FC6344D55DC632")

    val result = UdpMessage.tag(srcId, destId)

    assert(expected == result)
  }

  @Test
  fun getSourceFromTagFetchesSrcNodeId() {
    val srcId = Bytes.fromHexString("0x98EB6D611291FA21F6169BFF382B9369C33D997FE4DC93410987E27796360640")
    val destId = Bytes.fromHexString("0xA5CFE10E0EFC543CBE023560B2900E2243D798FAFD0EA46267DDD20D283CE13C")
    val tag = UdpMessage.tag(srcId, destId)

    val result = UdpMessage.getSourceFromTag(tag, destId)

    assert(srcId == result)
  }

  @Test
  fun authTagGivesRandom12Bytes() {
    val firstResult = UdpMessage.authTag()

    assert(UdpMessage.AUTH_TAG_LENGTH == firstResult.size())

    val secondResult = UdpMessage.authTag()

    assert(secondResult != firstResult)
  }

  @Test
  fun randomDataGivesRandom44Bytes() {
    val firstResult = UdpMessage.randomData()

    assert(UdpMessage.RANDOM_DATA_LENGTH == firstResult.size())

    val secondResult = UdpMessage.randomData()

    assert(secondResult != firstResult)
  }

  @Test
  fun idNonceGivesRandom32Bytes() {
    val firstResult = UdpMessage.idNonce()

    assert(UdpMessage.ID_NONCE_LENGTH == firstResult.size())

    val secondResult = UdpMessage.idNonce()

    assert(secondResult != firstResult)
  }

  @Test
  fun getMessageTypeThrowsExceptionByDefault() {
    assertThrows<UnsupportedOperationException> {
      RandomMessage().getMessageType()
    }
  }
}