Skip to main content
summaryrefslogtreecommitdiffstats
blob: 05073d8c90a9825c9ad4e586e5b44ab00f186693 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
//
//  ========================================================================
//  Copyright (c) 1995-2014 Mort Bay Consulting Pty. Ltd.
//  ------------------------------------------------------------------------
//  All rights reserved. This program and the accompanying materials
//  are made available under the terms of the Eclipse Public License v1.0
//  and Apache License v2.0 which accompanies this distribution.
//
//      The Eclipse Public License is available at
//      http://www.eclipse.org/legal/epl-v10.html
//
//      The Apache License v2.0 is available at
//      http://www.opensource.org/licenses/apache2.0.php
//
//  You may elect to redistribute this code under either of these licenses.
//  ========================================================================
//

package org.eclipse.jetty.server;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import java.io.InputStreamReader;
import java.io.LineNumberReader;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.Socket;

import org.junit.Test;

/**
 * ShutdownMonitorTest
 *
 *
 *
 */
public class ShutdownMonitorTest
{


    @Test
    public void testShutdown ()
    throws Exception
    {

        //test port and key assignment
        ShutdownMonitor.getInstance().setPort(0);
        ShutdownMonitor.getInstance().setExitVm(false);
        ShutdownMonitor.getInstance().start();
        String key = ShutdownMonitor.getInstance().getKey();
        int port = ShutdownMonitor.getInstance().getPort();
        
        //try starting a 2nd time (should be ignored)
        ShutdownMonitor.getInstance().start();   
      
        
        stop(port,key,true);
        assertTrue(!ShutdownMonitor.getInstance().isAlive());
      
        //should be able to change port and key because it is stopped
        ShutdownMonitor.getInstance().setPort(0);
        ShutdownMonitor.getInstance().setKey("foo");        
        ShutdownMonitor.getInstance().start();
        
        key = ShutdownMonitor.getInstance().getKey();
        port = ShutdownMonitor.getInstance().getPort();
        assertTrue(ShutdownMonitor.getInstance().isAlive());

        stop(port,key,true);
        assertTrue(!ShutdownMonitor.getInstance().isAlive());
    }


    public void stop (int port, String key, boolean check)
    throws Exception
    {
        Socket s = null;

        try
        {
            //send stop command
            s = new Socket(InetAddress.getByName("127.0.0.1"),port);

            OutputStream out = s.getOutputStream();
            out.write((key + "\r\nstop\r\n").getBytes());
            out.flush();

            if (check)
            {
                //wait a little
                Thread.currentThread().sleep(600);

                //check for stop confirmation
                LineNumberReader lin = new LineNumberReader(new InputStreamReader(s.getInputStream()));
                String response;
                if ((response = lin.readLine()) != null)
                {
                    assertEquals("Stopped", response);
                }
                else
                    throw new IllegalStateException("No stop confirmation");
            }
        }
        finally
        {
            if (s != null) s.close();
        }
    }

}

Back to the top