// SPDX-FileCopyrightText: 2022 Paul Schaub <vanitasvitae@fsfe.org>
//
// SPDX-License-Identifier: Apache-2.0

package pgp.wkd;

import org.junit.jupiter.api.Test;
import pgp.wkd.discovery.WKDPolicy;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.nio.charset.StandardCharsets;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;

public class WKDPolicyTest {

    @Test
    public void parseEmptyPolicy() throws IOException {
        ByteArrayInputStream empty = new ByteArrayInputStream(new byte[0]);
        WKDPolicy policy = WKDPolicy.fromInputStream(empty);

        assertFalse(policy.isMailboxOnly());
        assertFalse(policy.isDaneOnly());
        assertFalse(policy.isAuthSubmit());
        assertNull(policy.getProtocolVersion());
        assertNull(policy.getSubmissionAddress());
    }

    @Test
    public void parseWhitespaceOnlyPolicy() throws IOException {
        ByteArrayInputStream whitespaceOnly = new ByteArrayInputStream(
                ("\n" +
                        "\r\n" +
                        "     \n" +
                        "\t\n" +
                        "\n").getBytes(StandardCharsets.UTF_8));
        WKDPolicy policy = WKDPolicy.fromInputStream(whitespaceOnly);

        assertFalse(policy.isMailboxOnly());
        assertFalse(policy.isDaneOnly());
        assertFalse(policy.isAuthSubmit());
        assertNull(policy.getProtocolVersion());
        assertNull(policy.getSubmissionAddress());
    }

    @Test
    public void parseSparsePolicy() throws IOException {
        ByteArrayInputStream sparse = new ByteArrayInputStream(
                "protocol-version: 13\n".getBytes(StandardCharsets.UTF_8));
        WKDPolicy policy = WKDPolicy.fromInputStream(sparse);

        assertFalse(policy.isMailboxOnly());
        assertFalse(policy.isDaneOnly());
        assertFalse(policy.isAuthSubmit());
        assertEquals(13, policy.getProtocolVersion());
        assertNull(policy.getSubmissionAddress());
    }

    @Test
    public void parseFullPolicy() throws IOException {
        ByteArrayInputStream full = new ByteArrayInputStream(
                ("mailbox-only\n" +
                        "dane-only\n" +
                        "auth-submit\n" +
                        "protocol-version: 12\n" +
                        "submission-address: key-submission-example.org@directory.example.org")
                        .getBytes(StandardCharsets.UTF_8));
        WKDPolicy policy = WKDPolicy.fromInputStream(full);

        assertTrue(policy.isMailboxOnly());
        assertTrue(policy.isDaneOnly());
        assertTrue(policy.isAuthSubmit());

        assertEquals(12, policy.getProtocolVersion());
        assertEquals("key-submission-example.org@directory.example.org", policy.getSubmissionAddress());
    }

    @Test
    public void malformedInput_MissingSpaces() throws IOException {
        ByteArrayInputStream malformed = new ByteArrayInputStream(
                ("submission-address submit-keys-here@directory.example.org\n" +
                        "protocol-version:12").getBytes(StandardCharsets.UTF_8));

        WKDPolicy policy = WKDPolicy.fromInputStream(malformed);
        assertNull(policy.getProtocolVersion());
    }

    @Test
    public void malformedInput_missingColon() throws IOException {
        ByteArrayInputStream malformed = new ByteArrayInputStream(
                ("submission-address submit-keys-here@directory.example.org\n" +
                        "protocol-version 13").getBytes(StandardCharsets.UTF_8));
        WKDPolicy policy = WKDPolicy.fromInputStream(malformed);
        assertNull(policy.getSubmissionAddress());
        assertNull(policy.getProtocolVersion());
    }

    @Test
    public void malformedInput_NotANumber() throws IOException {
        ByteArrayInputStream malformed = new ByteArrayInputStream(
                "protocol-version: ABC".getBytes(StandardCharsets.UTF_8));

        WKDPolicy policy = WKDPolicy.fromInputStream(malformed);
        assertNull(policy.getProtocolVersion());
    }
}