1616// under the License.
1717package com .cloud .network .lb ;
1818
19+ import static org .mockito .ArgumentMatchers .any ;
20+ import static org .mockito .ArgumentMatchers .anyLong ;
21+ import static org .mockito .ArgumentMatchers .eq ;
22+ import static org .mockito .ArgumentMatchers .isNull ;
23+ import static org .mockito .ArgumentMatchers .nullable ;
24+ import static org .mockito .Mockito .when ;
25+
26+ import java .util .ArrayList ;
27+ import java .util .UUID ;
28+
29+ import org .apache .cloudstack .api .command .user .loadbalancer .UpdateLoadBalancerRuleCmd ;
30+ import org .apache .cloudstack .context .CallContext ;
31+ import org .junit .After ;
32+ import org .junit .Before ;
33+ import org .junit .Test ;
34+ import org .mockito .InOrder ;
35+ import org .mockito .Mockito ;
36+
1937import com .cloud .exception .InsufficientCapacityException ;
2038import com .cloud .exception .InvalidParameterValueException ;
2139import com .cloud .exception .ResourceAllocationException ;
3553import com .cloud .network .element .LoadBalancingServiceProvider ;
3654import com .cloud .offerings .dao .NetworkOfferingServiceMapDao ;
3755import com .cloud .user .Account ;
56+ import com .cloud .user .AccountManager ;
3857import com .cloud .user .AccountVO ;
39- import com .cloud .user .MockAccountManagerImpl ;
4058import com .cloud .user .User ;
4159import com .cloud .user .UserVO ;
42- import org .apache .cloudstack .api .command .user .loadbalancer .UpdateLoadBalancerRuleCmd ;
43- import org .apache .cloudstack .context .CallContext ;
44- import org .junit .After ;
45- import org .junit .Before ;
46- import org .junit .Test ;
47- import org .mockito .InOrder ;
48- import org .mockito .Mockito ;
49-
50- import java .util .ArrayList ;
51- import java .util .UUID ;
52-
53- import static org .mockito .ArgumentMatchers .isNull ;
54- import static org .mockito .ArgumentMatchers .any ;
55- import static org .mockito .ArgumentMatchers .anyLong ;
56- import static org .mockito .ArgumentMatchers .nullable ;
57- import static org .mockito .ArgumentMatchers .eq ;
58- import static org .mockito .Mockito .when ;
5960
6061public class UpdateLoadBalancerTest {
6162
@@ -73,7 +74,7 @@ public class UpdateLoadBalancerTest {
7374
7475 @ Before
7576 public void setUp () {
76- _lbMgr ._accountMgr = new MockAccountManagerImpl ( );
77+ _lbMgr ._accountMgr = Mockito . mock ( AccountManager . class );
7778 _lbMgr ._autoScaleVmGroupDao = Mockito .mock (AutoScaleVmGroupDao .class );
7879 _lbMgr ._networkDao = netDao ;
7980 _lbMgr ._networkOfferingServiceDao = ntwkOffServiceMapDao ;
0 commit comments