runtime/jsoncontrol/src/test/java/org/apache/edgent/test/runtime/jsoncontrol/JsonControlServiceTest.java

/*
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements.  See the NOTICE file
distributed with this work for additional information
regarding copyright ownership.  The ASF licenses this file
to you under the Apache License, Version 2.0 (the
"License"); you may not use this file except in compliance
with the License.  You may obtain a copy of the License at

  http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing,
software distributed under the License is distributed on an
"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
KIND, either express or implied.  See the License for the
specific language governing permissions and limitations
under the License.
*/
package org.apache.edgent.test.runtime.jsoncontrol;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;

import java.lang.Thread.State;
import java.util.concurrent.TimeUnit;

import org.apache.edgent.execution.services.Controls;
import org.apache.edgent.runtime.jsoncontrol.JsonControlService;
import org.junit.Test;

import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import com.google.gson.JsonPrimitive;

public class JsonControlServiceTest {

    public static interface MyBean {
        void doIt();

        void doString(String s);

        void doInt(int i);

        void doLongDouble(long l, double d);
        
        void doBooleanEnums(boolean b, TimeUnit u, Thread.State ts);
    }

    public static class MyBeanImpl implements MyBean {

        private boolean doneIt;

        private String doneS;
        private int doneI;
        private long doneL;
        private double doneD;
        
        private boolean doneB;
        private TimeUnit doneU;
        private Thread.State doneTs;

        public String getDoneS() {
            return doneS;
        }

        public synchronized int getDoneI() {
            return doneI;
        }

        public synchronized long getDoneL() {
            return doneL;
        }

        public synchronized double getDoneD() {
            return doneD;
        }

        @Override
        public synchronized void doIt() {
            doneIt = true;
        }

        synchronized boolean isDoneIt() {
            return doneIt;
        }

        @Override
        public synchronized void doInt(int i) {
            doneI = i;
        }

        @Override
        public synchronized void doString(String s) {
            doneS = s;
        }

        @Override
        public synchronized void doLongDouble(long l, double d) {
            doneL = l;
            doneD = d;
        }
        @Override
        public synchronized void doBooleanEnums(boolean b, TimeUnit u, State ts) {
            doneB = b;
            doneU = u;
            doneTs = ts;
        }

        public synchronized boolean isDoneB() {
            return doneB;
        }

        public synchronized TimeUnit getDoneU() {
            return doneU;
        }

        public synchronized Thread.State getDoneTs() {
            return doneTs;
        }
    }
    
    @Test
    public void testMyBean() {
        assertTrue(Controls.isControlServiceMBean(MyBean.class));
        assertFalse(Controls.isControlServiceMBean(MyBeanImpl.class));
    }

    @Test
    public void testNoArg() throws Exception {
        JsonControlService control = new JsonControlService();

        MyBeanImpl cb1 = new MyBeanImpl();
        MyBeanImpl cb2 = new MyBeanImpl();

        assertFalse(cb1.isDoneIt());
        assertFalse(cb2.isDoneIt());

        control.registerControl("myb", "1", null, MyBean.class, cb1);
        control.registerControl("myb", "2", null, MyBean.class, cb2);

        assertFalse(cb1.isDoneIt());
        assertFalse(cb2.isDoneIt());

        JsonObject req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "1");
        req.addProperty(JsonControlService.OP_KEY, "doIt");
        control.controlRequest(req);

        assertTrue(cb1.isDoneIt());
        assertFalse(cb2.isDoneIt());

        req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "2");
        req.addProperty(JsonControlService.OP_KEY, "doIt");
        control.controlRequest(req);

        assertTrue(cb1.isDoneIt());
        assertTrue(cb2.isDoneIt());
    }

    @Test
    public void test1ArgString() throws Exception {
        JsonControlService control = new JsonControlService();

        MyBeanImpl cb1 = new MyBeanImpl();
        MyBeanImpl cb2 = new MyBeanImpl();

        assertNull(cb1.getDoneS());
        assertNull(cb2.getDoneS());

        control.registerControl("myb", "1s", null, MyBean.class, cb1);
        control.registerControl("myb", "2s", null, MyBean.class, cb2);

        assertNull(cb1.getDoneS());
        assertNull(cb2.getDoneS());

        JsonObject req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "1s");
        req.addProperty(JsonControlService.OP_KEY, "doString");
        JsonArray args = new JsonArray();
        args.add(new JsonPrimitive("ABC"));       
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertEquals("ABC", cb1.getDoneS());
        assertNull(cb2.getDoneS());

        req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "2s");
        req.addProperty(JsonControlService.OP_KEY, "doString");
        args = new JsonArray();
        args.add(new JsonPrimitive("DEF"));       
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertEquals("ABC", cb1.getDoneS());
        assertEquals("DEF", cb2.getDoneS());
    }
    
    @Test
    public void test1ArgInt() throws Exception {
        JsonControlService control = new JsonControlService();

        MyBeanImpl cb1 = new MyBeanImpl();
        MyBeanImpl cb2 = new MyBeanImpl();

        assertEquals(0, cb1.getDoneI());
        assertEquals(0, cb2.getDoneI());

        control.registerControl("myb", "1i", null, MyBean.class, cb1);
        control.registerControl("myb", "2i", null, MyBean.class, cb2);

        assertEquals(0, cb1.getDoneI());
        assertEquals(0, cb2.getDoneI());

        JsonObject req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "1i");
        req.addProperty(JsonControlService.OP_KEY, "doInt");
        JsonArray args = new JsonArray();
        args.add(new JsonPrimitive(726));       
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertEquals(726, cb1.getDoneI());
        assertEquals(0, cb2.getDoneI());

        req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "2i");
        req.addProperty(JsonControlService.OP_KEY, "doInt");
        args = new JsonArray();
        args.add(new JsonPrimitive(2924));       
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertEquals(726, cb1.getDoneI());
        assertEquals(2924, cb2.getDoneI());
    }
    
    @Test
    public void test2ArgLongDouble() throws Exception {
        JsonControlService control = new JsonControlService();

        MyBeanImpl cb1 = new MyBeanImpl();
        MyBeanImpl cb2 = new MyBeanImpl();

        assertEquals(0.0, cb1.getDoneD(), 0.0);
        assertEquals(0L, cb1.getDoneL());
        assertEquals(0.0, cb2.getDoneD(), 0.0);
        assertEquals(0L, cb2.getDoneL());

        control.registerControl("myb", "1ld", null, MyBean.class, cb1);
        control.registerControl("myb", "2ld", null, MyBean.class, cb2);

        assertEquals(0.0, cb1.getDoneD(), 0.0);
        assertEquals(0L, cb1.getDoneL());
        assertEquals(0.0, cb2.getDoneD(), 0.0);
        assertEquals(0L, cb2.getDoneL());

        JsonObject req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "1ld");
        req.addProperty(JsonControlService.OP_KEY, "doLongDouble");
        JsonArray args = new JsonArray();
        args.add(new JsonPrimitive(9345L));   
        args.add(new JsonPrimitive(89.24));   
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertEquals(89.24, cb1.getDoneD(), 0.0);
        assertEquals(9345L, cb1.getDoneL());
        assertEquals(0.0, cb2.getDoneD(), 0.0);
        assertEquals(0L, cb2.getDoneL());


        req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "2ld");
        req.addProperty(JsonControlService.OP_KEY, "doLongDouble");
        args = new JsonArray();
        args.add(new JsonPrimitive(74737L));   
        args.add(new JsonPrimitive(-9235.232));   
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertEquals(89.24, cb1.getDoneD(), 0.0);
        assertEquals(9345L, cb1.getDoneL());
        assertEquals(-9235.232, cb2.getDoneD(), 0.0);
        assertEquals(74737L, cb2.getDoneL());
    }
    
    @Test
    public void test3ArgBooleanEnum() throws Exception {
        JsonControlService control = new JsonControlService();

        MyBeanImpl cb1 = new MyBeanImpl();
        MyBeanImpl cb2 = new MyBeanImpl();

        assertFalse(cb1.isDoneB());
        assertNull(cb1.getDoneU());
        assertNull(cb1.getDoneTs());
        assertFalse(cb2.isDoneB());
        assertNull(cb2.getDoneU());
        assertNull(cb2.getDoneTs());

        control.registerControl("myb", "1eb", null, MyBean.class, cb1);
        control.registerControl("myb", "2eb", null, MyBean.class, cb2);

        assertFalse(cb1.isDoneB());
        assertNull(cb1.getDoneU());
        assertNull(cb1.getDoneTs());
        assertFalse(cb2.isDoneB());
        assertNull(cb2.getDoneU());
        assertNull(cb2.getDoneTs());


        JsonObject req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "1eb");
        req.addProperty(JsonControlService.OP_KEY, "doBooleanEnums");
        JsonArray args = new JsonArray();
        args.add(new JsonPrimitive(true));   
        args.add(new JsonPrimitive(TimeUnit.DAYS.name()));
        args.add(new JsonPrimitive(Thread.State.NEW.name()));   
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertTrue(cb1.isDoneB());
        assertSame(TimeUnit.DAYS, cb1.getDoneU());
        assertSame(Thread.State.NEW, cb1.getDoneTs());
        assertFalse(cb2.isDoneB());
        assertNull(cb2.getDoneU());
        assertNull(cb2.getDoneTs());


        req = new JsonObject();
        req.addProperty(JsonControlService.TYPE_KEY, "myb");
        req.addProperty(JsonControlService.ALIAS_KEY, "2eb");
        req.addProperty(JsonControlService.OP_KEY, "doBooleanEnums");
        args = new JsonArray();
        args.add(new JsonPrimitive(false));   
        args.add(new JsonPrimitive(TimeUnit.HOURS.name()));
        args.add(new JsonPrimitive(Thread.State.BLOCKED.name()));   
        req.add(JsonControlService.ARGS_KEY, args);
        control.controlRequest(req);

        assertTrue(cb1.isDoneB());
        assertSame(TimeUnit.DAYS, cb1.getDoneU());
        assertSame(Thread.State.NEW, cb1.getDoneTs());
        assertFalse(cb2.isDoneB());
        assertSame(TimeUnit.HOURS, cb2.getDoneU());
        assertSame(Thread.State.BLOCKED, cb2.getDoneTs());
    }
}