trying to figure out jest issue.
This commit is contained in:
@ -9,21 +9,21 @@ describe('Conversation', () => {
|
||||
|
||||
beforeEach(() => {
|
||||
const translator = new Translator("en", "es");
|
||||
const s1: Speaker = { language: "en", id: "s1" };
|
||||
const s2: Speaker = { id: "s2", language: "es" }
|
||||
const s1: Speaker = { language: "en", id: "host" };
|
||||
const s2: Speaker = { id: "guest", language: "es" }
|
||||
conversation = new Conversation(translator, s1, s2);
|
||||
});
|
||||
|
||||
it('should add a message to the conversation', () => {
|
||||
conversation.addMessage({ id: "s1", language: "en" }, "Hello");
|
||||
conversation.addMessage({ id: "host", language: "en" }, "Hello");
|
||||
expect(conversation.length).toBe(1);
|
||||
expect(conversation[0].speaker.id).toEqual("s1");
|
||||
expect(conversation[0].speaker.id).toEqual("host");
|
||||
expect(conversation[0].text).toEqual("Hello");
|
||||
});
|
||||
|
||||
it('should translate the last message in the conversation', async () => {
|
||||
|
||||
conversation.addMessage({ id: "s1", language: "en" }, "Hello");
|
||||
conversation.addMessage({ id: "host", language: "en" }, "Hello");
|
||||
|
||||
await conversation.translateLast();
|
||||
expect(conversation[conversation.length - 1].translation).toEqual("Hola");
|
||||
|
@ -5,33 +5,54 @@ export type Speaker = {
|
||||
language : string,
|
||||
}
|
||||
|
||||
|
||||
export class Message {
|
||||
public translation? : string;
|
||||
constructor (public text : string, public speaker : Speaker) {}
|
||||
public onTextUpdate?: (message: Message) => void;
|
||||
public onTextDone?: (message: Message) => Promise<void>;
|
||||
public onTranslationDone?: (message: Message) => void;
|
||||
|
||||
constructor (public conversation : Conversation, public speaker : Speaker, public text? : string) {}
|
||||
|
||||
public async translate(translator : Translator, language? : string) {
|
||||
if (!this.text) throw new Error("No text")
|
||||
this.translation = await translator.translate(this.text, language);
|
||||
}
|
||||
|
||||
get otherSpeaker() {
|
||||
return this.speaker.id === this.conversation.host.id ? this.conversation.guest : this.conversation.host
|
||||
}
|
||||
|
||||
get otherLanguage() {
|
||||
return this.otherSpeaker.language
|
||||
}
|
||||
}
|
||||
|
||||
export class Conversation extends Array<Message> {
|
||||
|
||||
public onAddMessage? : (conversation : Conversation) => Promise<any>;
|
||||
public onTranslationDone? : (conversation : Conversation) => Promise<any>;
|
||||
|
||||
constructor (
|
||||
private translator : Translator,
|
||||
public s1 : Speaker,
|
||||
public s2 : Speaker,
|
||||
public host : Speaker,
|
||||
public guest : Speaker,
|
||||
) {
|
||||
super();
|
||||
}
|
||||
|
||||
public addMessage(speaker : Speaker, text : string) {
|
||||
this.push(new Message(text, speaker));
|
||||
public addMessage(speaker : Speaker, text? : string) {
|
||||
this.push(new Message(this, speaker, text));
|
||||
}
|
||||
|
||||
public async translateMessage(i : number) {
|
||||
if (!this[i]) throw new Error(`${i} is not a valid message number`);
|
||||
const otherSpeaker = this[i].speaker.id === this.s1.id ? this.s2 : this.s1
|
||||
console.log(`Translating sentence to %s: %s`, otherSpeaker.language, this[i].text)
|
||||
await this[i].translate(this.translator, otherSpeaker.language);
|
||||
console.log(`Translating sentence to %s: %s`, this[i].otherLanguage, this[i].text)
|
||||
await this[i].translate(this.translator, this[i].otherLanguage);
|
||||
}
|
||||
|
||||
get lastMessage() {
|
||||
return this[this.length-1]
|
||||
}
|
||||
|
||||
public async translateLast() {
|
||||
|
Reference in New Issue
Block a user