[Jboss-cvs] JBossAS SVN: r56191 - trunk/testsuite/src/main/org/jboss/test/security/test

[jboss-cvs-commits at lists.jboss.org]

Author: anil.saldhana at jboss.com
Date: 2006-08-23 14:55:34 -0400 (Wed, 23 Aug 2006)
New Revision: 56191

Modified:
   trunk/testsuite/src/main/org/jboss/test/security/test/DynamicLoginConfigServiceUnitTestCase.java
Log:
Use ServiceController to deal with start/stop service

Modified: trunk/testsuite/src/main/org/jboss/test/security/test/DynamicLoginConfigServiceUnitTestCase.java
===================================================================
--- trunk/testsuite/src/main/org/jboss/test/security/test/DynamicLoginConfigServiceUnitTestCase.java	2006-08-23 18:50:43 UTC (rev 56190)
+++ trunk/testsuite/src/main/org/jboss/test/security/test/DynamicLoginConfigServiceUnitTestCase.java	2006-08-23 18:55:34 UTC (rev 56191)
@@ -23,15 +23,16 @@
 
 import java.net.URL;
 
-import javax.management.Attribute;
-import javax.management.MBeanServerConnection;
+import javax.management.Attribute; 
+import javax.management.MBeanServerConnection; 
 import javax.management.ObjectName;
 
 import junit.framework.Test;
 import junit.framework.TestSuite;
-
+  
 import org.jboss.test.JBossTestCase;
 import org.jboss.test.JBossTestSetup; 
+import org.jboss.test.util.ServiceControllerUtil;
 
 //$Id$
 
@@ -48,9 +49,18 @@
                 "<login-module code='org.jboss.security.auth.spi.UsersRolesLoginModule'" +
                 " flag = 'required' /> </authentication></application-policy></policy>";
    
+   private ServiceControllerUtil sutil = null;
    public DynamicLoginConfigServiceUnitTestCase(String name)
    {
       super(name); 
+      try
+      {
+         sutil = new ServiceControllerUtil(this.getServer());
+      }
+      catch (Exception e)
+      {
+         fail("Exception thrown in creating util class"+e.getLocalizedMessage());
+      }
    }
    
    /**
@@ -65,27 +75,30 @@
             new Object[]{"test-dyn",login_config},
             new String[] {"java.lang.String", "java.lang.String"});
       assertNotNull(" config url != null",confURL); 
-      ObjectName serviceOName = new ObjectName("jboss:service=TestDynamicLoginConfig");
-      if(server.isRegistered(serviceOName))
-         server.unregisterMBean(serviceOName);
-      prepareTestDynamicLoginConfig(server,serviceOName,confURL.toExternalForm());   
-      server.invoke(serviceOName,"create", new Object[0], new String[0]); 
-      server.invoke(serviceOName,"start", new Object[0], new String[0]);
+      ObjectName serviceName = new ObjectName("jboss:service=TestDynamicLoginConfig");
+      if(server.isRegistered(serviceName))
+         server.unregisterMBean(serviceName);
+      prepareTestDynamicLoginConfig(server,serviceName,confURL.toExternalForm()); 
+      sutil.createAService(serviceName);
+      assertTrue("Created?", sutil.isCreated(serviceName));
+      sutil.startAService(serviceName);
+      assertTrue("Started?", sutil.isStarted(serviceName)); 
+      //Restart the service 
+      sutil.stopAService(serviceName);
+      assertTrue("Stopped?", sutil.isStopped(serviceName)); 
+      sutil.startAService(serviceName);
+      assertTrue("Started?", sutil.isStarted(serviceName));  
       
-      //Restart the service
-      server.invoke(serviceOName,"stop", new Object[0], new String[0]);
-      server.invoke(serviceOName,"start", new Object[0], new String[0]);
-      
-      //Ensure that the service has been started
-      String startedStr = (String)server.getAttribute(serviceOName, "StateString");
-      assertTrue("Test Dynamic Login Config Service started?", startedStr.equalsIgnoreCase("Started"));
-      
-      String authConfig = (String)server.getAttribute(serviceOName,"AuthConfig");
+      String authConfig = (String)server.getAttribute(serviceName,"AuthConfig");
       assertEquals(confURL + "matches", confURL.toExternalForm(), authConfig); 
-      server.invoke(serviceOName,"stop", new Object[0], new String[0]);
-      server.invoke(serviceOName,"destroy", new Object[0], new String[0]);
-      if(server.isRegistered(serviceOName))
-         server.unregisterMBean(serviceOName);
+      sutil.stopAService(serviceName);
+      assertTrue("Stopped?", sutil.isStopped(serviceName)); 
+      sutil.startAService(serviceName);
+      assertTrue("Started?", sutil.isStarted(serviceName));
+      sutil.destroyAService(serviceName);
+      assertTrue("Destroyed?", sutil.isDestroyed(serviceName));
+      if(server.isRegistered(serviceName))
+         server.unregisterMBean(serviceName);
    } 
    
    /**
@@ -95,30 +108,44 @@
    public void testAuthConf() throws Exception
    {
       MBeanServerConnection server = getServer(); 
-      ObjectName serviceOName = new ObjectName("jboss:service=TestDynamicLoginConfig");
-      if(server.isRegistered(serviceOName))
-         server.unregisterMBean(serviceOName);
+      ObjectName serviceName = new ObjectName("jboss:service=TestDynamicLoginConfig");
+      if(server.isRegistered(serviceName))
+         server.unregisterMBean(serviceName);
       prepareTestDynamicLoginConfig(server, 
-            new ObjectName("jboss:service=TestDynamicLoginConfig"), null);
+            new ObjectName("jboss:service=TestDynamicLoginConfig"), null); 
       try
-      {
-         server.invoke(serviceOName,"create", new Object[0], new String[0]); 
-         server.invoke(serviceOName,"start", new Object[0], new String[0]);
-         fail("Service should not have started");
+      { 
+         sutil.createAService(serviceName);
+         assertTrue("Created?", sutil.isCreated(serviceName)); 
+         sutil.startAService(serviceName);
+         assertFalse("Should not Start", sutil.isStarted(serviceName));  
       }
       catch(Throwable t)
       {
-         log.debug("Service has rightly disagreed to start",t); 
+         fail("Exception thrown:"+t.getLocalizedMessage()); 
       }
       finally
       {
-         server.invoke(serviceOName,"stop", new Object[0], new String[0]); 
+         if(sutil.isStarted(serviceName)) 
+         {
+            sutil.stopAService(serviceName);
+            assertTrue("Stopped?", sutil.isStopped(serviceName));
+         } 
       }
-      server.setAttribute(serviceOName,new Attribute("AuthConfig","login-config.xml"));  
       try
+      {
+         server.setAttribute(serviceName,
+               new Attribute("AuthConfig","login-config.xml"));
+      }
+      catch(Throwable t)
+      {
+         fail("Not able to set attribute:"+ t.getLocalizedMessage());
+      }
+        
+      try
       { 
-         server.invoke(serviceOName,"start", new Object[0], new String[0]);
-         fail("Service should not have started");
+         sutil.startAService(serviceName);
+         assertFalse("Should not Start?", sutil.isStarted(serviceName)); 
       }
       catch(Throwable t)
       {
@@ -126,10 +153,15 @@
       }
       finally
       {
-         server.invoke(serviceOName,"stop", new Object[0], new String[0]);
-         server.invoke(serviceOName,"destroy", new Object[0], new String[0]);
-         if(server.isRegistered(serviceOName))
-            server.unregisterMBean(serviceOName);
+         if(sutil.isStarted(serviceName)) 
+         {
+            sutil.stopAService(serviceName);
+            assertTrue("Stopped?", sutil.isStopped(serviceName));
+            sutil.destroyAService(serviceName);
+            assertTrue("Destroyed?", sutil.isDestroyed(serviceName));
+         }  
+         if(server.isRegistered(serviceName))
+            server.unregisterMBean(serviceName);
       }
    }
    
@@ -147,7 +179,7 @@
       ObjectName lcs = new ObjectName("jboss.security:service=XMLLoginConfig");
       Attribute attrLCS = new Attribute("LoginConfigService", lcs);
       server.setAttribute(serviceOName,attrLCS); 
-   }
+   } 
    
    public static Test suite()
    throws Exception