package rename for test libraries.
[utils] / test / enterprise / src / main / java / org / wamblee / test / jndi / StubInitialContext.java
1 /*
2  * Copyright 2005-2010 the original author or authors.
3  * 
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  * 
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  * 
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package org.wamblee.test.jndi;
17
18 import java.util.HashMap;
19 import java.util.Map;
20
21 import javax.naming.InitialContext;
22 import javax.naming.Name;
23 import javax.naming.NameNotFoundException;
24 import javax.naming.NamingException;
25
26 /**
27  * Initial context implementation. 
28  * 
29  * @author Erik Brakkee
30  */
31 class StubInitialContext extends InitialContext {
32     private Map<String, Object> bindings = new HashMap<String, Object>();
33
34     public StubInitialContext() throws NamingException {
35         super(true);
36     }
37
38     @Override
39     public void bind(String aName, Object aObj) throws NamingException {
40         bindings.put(aName, aObj);
41     }
42     
43     @Override
44     public void unbind(String aName) throws NamingException {
45         bindings.remove(aName);
46     }
47
48     @Override
49     public Object lookup(String aName) throws NamingException {
50         Object value = bindings.get(aName);
51         if (value == null) {
52             throw new NameNotFoundException(aName);
53         }
54         return value;
55     }
56     
57     @Override
58     public void bind(Name aName, Object aObj) throws NamingException {
59         bind(aName.toString(), aObj);
60     }
61     
62     @Override
63     public void unbind(Name aName) throws NamingException {
64         unbind(aName.toString());
65     }
66
67     @Override
68     public Object lookup(Name aName) throws NamingException {
69         return lookup(aName.toString());
70     }
71 }