56 lines
1.5 KiB
C#
56 lines
1.5 KiB
C#
using Bot.Rasa;
|
|
using Bot.Rasa.Agents;
|
|
using Bot.Rasa.Console;
|
|
using EntityFrameworkCore.BootKit;
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
|
|
namespace Bot.UnitTest
|
|
{
|
|
[TestClass]
|
|
public class AgentTest : TestEssential
|
|
{
|
|
public static String PIZZA_BOT_ID = "2b6a288e-d891-40c6-96ce-6a0cf324545c";
|
|
|
|
[TestMethod]
|
|
public void CreateAgent()
|
|
{
|
|
var rasa = new RasaConsole(dc);
|
|
|
|
var agent = rasa.RestoreAgent(PIZZA_BOT_ID);
|
|
|
|
int row = dc.DbTran(() => rasa.CreateAgent(agent));
|
|
|
|
if(row > 0)
|
|
{
|
|
//var result = rasa.Train(dc, agent.Id);
|
|
}
|
|
|
|
var loadedAgent = rasa.LoadAgent(agent.Id);
|
|
Assert.IsTrue(loadedAgent.Intents.Count == agent.Intents.Count);
|
|
|
|
var response = rasa.TextRequest(agent.Id, "weather in Chicago tomorrow");
|
|
Assert.IsTrue(response.Intent.Name == "weather");
|
|
}
|
|
|
|
[TestMethod]
|
|
public void TextRequest()
|
|
{
|
|
var rasa = new RasaConsole(dc);
|
|
var response = rasa.TextRequest(PIZZA_BOT_ID, "how old are you");
|
|
response = rasa.TextRequest(PIZZA_BOT_ID, "where do you come from");
|
|
response = rasa.TextRequest(PIZZA_BOT_ID, "would you like some cookie");
|
|
}
|
|
|
|
[TestMethod]
|
|
public void Train()
|
|
{
|
|
var rasa = new RasaConsole(dc);
|
|
rasa.Train(dc, PIZZA_BOT_ID);
|
|
}
|
|
}
|
|
}
|