btelman96/letsrobot-android

package tv.letsrobot.android.api.components;

import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import androidx.test.InstrumentationRegistry;
import androidx.test.runner.AndroidJUnit4;

/**
 * Specifically test just the robot control portion. No motors will move.
 *
 * This will connect to the LetsRobot api to ensure proper functionality and making sure that it
 * would send it to a motor
 *
 * To trigger events in the logs, please use the website controls on the robot page
 */
@RunWith(AndroidJUnit4.class)
public class ControlSocketComponentAndroidTests {
    @Test
    public void Init(){
        ControlSocketComponent controllerComponent = new ControlSocketComponent(InstrumentationRegistry.getTargetContext(),""); //TODO ROBOTID
        controllerComponent.enable();
        Assert.assertTrue(controllerComponent.getRunning().get());
        CountDownLatch latch = new CountDownLatch(1);
        try {
            latch.await(2, TimeUnit.MINUTES); //Wait for 2 Minutes
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        Assert.assertTrue(controllerComponent.getConnected()); //Make sure socket is actually connected
        controllerComponent.disable(); //Disable controller
        latch = new CountDownLatch(1);
        try {
            //wait a little bit to make sure it had time to disconnect
            latch.await(2, TimeUnit.SECONDS);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
        Assert.assertFalse(controllerComponent.getConnected()); //Make sure disable works
    }
}