CommandsNlpServiceTest.java

/**
 *
 * Copyright (c) 2009-2016 Freedomotic team http://freedomotic.com
 *
 * This file is part of Freedomotic
 *
 * This Program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software
 * Foundation; either version 2, or (at your option) any later version.
 *
 * This Program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License along with
 * Freedomotic; see the file COPYING. If not, see
 * <http://www.gnu.org/licenses/>.
 */
package com.freedomotic.core;

import com.freedomotic.nlp.Nlp;
import com.freedomotic.nlp.NlpCommand;
import com.freedomotic.reactions.Command;
import com.freedomotic.reactions.CommandRepository;
import com.freedomotic.testutils.FreedomoticTestsInjector;
import com.freedomotic.testutils.GuiceJUnitRunner;
import java.util.List;
import org.slf4j.LoggerFactory;
import javax.inject.Inject;
import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.*;
import org.junit.runner.RunWith;
import org.slf4j.Logger;

/**
 *
 * @author Enrico Nicoletti
 */
@RunWith(GuiceJUnitRunner.class)
@GuiceJUnitRunner.GuiceInjectors({FreedomoticTestsInjector.class})
public class CommandsNlpServiceTest {

    private static final Logger LOG = LoggerFactory.getLogger(CommandsNlpServiceTest.class.getName());

    @Inject
    CommandRepository commandRepository;
    @Inject
    NlpCommand nlpCommand;

    public CommandsNlpServiceTest() {
    }

    @BeforeClass
    public static void setUpClass() {
    }

    @AfterClass
    public static void tearDownClass() {
    }

    @Before
    public void setUp() {
        // Fix for failing test:
        commandRepository.deleteAll();


        Command command = new Command();
        command.setName("Turn on kitchen light");
        command.setReceiver("app.events.sensors.behavior.request.objects");
        command.setProperty("object", "Kitchen Light");
        command.setProperty("behavior", "powered");
        command.setProperty("value", "true");
        commandRepository.create(command);
    }

    @After
    public void tearDown() {
        // DeleteAll  is done on setup
        //commandRepository.deleteAll();
    }

    /**
     * Test of findMostSimilarCommand method, of class
     * FreeFormCommandsInterpreter.
     *
     * @throws java.lang.Exception
     */
    @Test
    public void testFindMostSimilarCommand() throws Exception {
        LOG.info("Find the most similar command using NLP");
        String phrase = "kitccchen LiGhTT Power ON";
        // Compute the commands ranking
        List<Nlp.Rank<Command>> ranking = nlpCommand.computeSimilarity(phrase, 10);
        assertEquals("Should find exacly one similar command", ranking.size(), 1);
    }

    @Test
    public void testNoSimilarCommand() throws Exception {
        LOG.info("Test for an unexistent command");
        String phrase = "asdasd tretert gbffdg uyututy mnbb";
        // Compute the commands ranking
        List<Nlp.Rank<Command>> ranking = nlpCommand.computeSimilarity(phrase, 10);
      
        assertEquals("Should find a command anyway, because zero similarity is allowed", 1, ranking.size());
        assertEquals("The command should be totally different from anyone else in the repository", 0, ranking.get(0).getSimilarity());
    }

}