Mətnlərin tərcüməsi

package academy.study.usecases;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;

public final class ProtectedTranslation {
    public enum Locale { EN, AZ }
    public enum Kind { PROSE, VERBATIM }
    public record Part(String id, Kind kind, String text) {}
    public record TextPart(String id, String text) {}
    public record Request(Locale sourceLocale, Locale targetLocale, List<TextPart> parts) {}
    public record Draft(Locale sourceLocale, Locale targetLocale, List<TextPart> parts) {}
    private final StructuredWork work;
    public ProtectedTranslation(StructuredWork work) { this.work = work; }

    public List<Part> translate(Locale source, Locale target, List<Part> parts) {
        if (source == null || target == null || parts == null || parts.isEmpty() || parts.size() > 8) {
            throw StructuredWork.invalid();
        }
        var ids = new HashSet<String>();
        int length = 0;
        for (var part : parts) {
            if (part == null || part.id() == null || !part.id().matches("p[1-8]") || !ids.add(part.id())
                    || part.kind() == null || part.text() == null || part.text().isBlank()
                    || part.text().length() > (part.kind() == Kind.PROSE ? 600 : 1200)) {
                throw StructuredWork.invalid();
            }
            length += part.text().length();
        }
        if (length > 3000) throw StructuredWork.invalid();
        parts = List.copyOf(parts);
        var prose = parts.stream().filter(p -> p.kind() == Kind.PROSE)
                .map(p -> new TextPart(p.id(), p.text())).toList();
        if (source == target || prose.isEmpty()) return parts;
        var draft = work.run("""
                Translate every prose block into targetLocale, preserving its meaning and qualifications.
                Return sourceLocale, targetLocale and translated parts with the same IDs and order.
                EN means English; AZ means Azerbaijani. Do not invent or add blocks.
                """, new Request(source, target, prose), Draft.class);
        if (draft.sourceLocale() != source || draft.targetLocale() != target || draft.parts() == null
                || draft.parts().size() != prose.size()) throw StructuredWork.invalid();
        for (int i = 0; i < prose.size(); i++) {
            var translated = draft.parts().get(i);
            if (translated == null || !prose.get(i).id().equals(translated.id()) || translated.text() == null
                    || translated.text().isBlank() || translated.text().length() > 1200) {
                throw StructuredWork.invalid();
            }
        }
        var result = new ArrayList<Part>();
        int next = 0;
        for (var part : parts) {
            result.add(part.kind() == Kind.VERBATIM ? part
                    : new Part(part.id(), part.kind(), draft.parts().get(next++).text()));
        }
        return List.copyOf(result);
    }
}

package academy.study.usecases;

import java.util.List;
import org.junit.jupiter.api.Test;
import static academy.study.usecases.ProtectedTranslation.*;
import static org.assertj.core.api.Assertions.*;

class TranslationTest {
    static final String CODE = "var planId = \"p-17\";\nservice.savePlan(planId);";
    static final List<Part> SOURCE = List.of(new Part("p1", Kind.PROSE, "Save only after confirmation."),
            new Part("p2", Kind.VERBATIM, CODE), new Part("p3", Kind.PROSE, "The example uses a stable identifier."));
    static final String VALID = """
            {"sourceLocale":"EN","targetLocale":"AZ","parts":[
            {"id":"p1","text":"Yalnız təsdiqdən sonra saxlayın."},
            {"id":"p3","text":"Nümunə sabit identifikatordan istifadə edir."}]}
            """;
    ProtectedTranslation service(ScriptedModel model) { return new ProtectedTranslation(new StructuredWork(model)); }
    @Test void protectedCodeNeverEntersTheModelAndIsRestoredUnchanged() {
        var model = new ScriptedModel(VALID);
        var result = service(model).translate(Locale.EN, Locale.AZ, SOURCE);
        assertThat(result.get(1)).isEqualTo(SOURCE.get(1));
        assertThat(result.get(0).text()).isEqualTo("Yalnız təsdiqdən sonra saxlayın.");
        assertThat(model.prompts.getFirst().getContents()).doesNotContain("p-17", "savePlan", "planId");
    }
    @Test void wrongLocaleMissingBlockAndInventedProtectedBlockAreRejected() {
        for (String output : List.of(VALID.replace("\"targetLocale\":\"AZ\"", "\"targetLocale\":\"EN\""),
                VALID.replace("\"id\":\"p3\"", "\"id\":\"p2\""),
                "{\"sourceLocale\":\"EN\",\"targetLocale\":\"AZ\",\"parts\":[]}")) {
            assertThatThrownBy(() -> service(new ScriptedModel(output)).translate(Locale.EN, Locale.AZ, SOURCE))
                    .isInstanceOf(IllegalArgumentException.class);
        }
    }
    @Test void sameLocaleAndCodeOnlyInputsDoNotCallModel() {
        var model = new ScriptedModel(VALID);
        assertThat(service(model).translate(Locale.EN, Locale.EN, SOURCE)).isEqualTo(SOURCE);
        assertThat(service(model).translate(Locale.EN, Locale.AZ, List.of(SOURCE.get(1))))
                .containsExactly(SOURCE.get(1));
        assertThat(model.prompts).isEmpty();
    }
    @Test void duplicateInputIdsAreRejectedBeforeModelCall() {
        var model = new ScriptedModel(VALID);
        assertThatThrownBy(() -> service(model).translate(Locale.EN, Locale.AZ,
                List.of(SOURCE.get(0), SOURCE.get(0)))).isInstanceOf(IllegalArgumentException.class);
        assertThat(model.prompts).isEmpty();
    }
    @Test void localeLabelAloneDoesNotProveTranslationQuality() {
        var model = new ScriptedModel(VALID.replace("Yalnız təsdiqdən sonra saxlayın.", "Save only after confirmation."));
        var result = service(model).translate(Locale.EN, Locale.AZ, SOURCE);
        assertThat(result.getFirst().text()).isEqualTo("Save only after confirmation.");
    }
}

mvn -B -Dtest=TranslationTest test