16
16
// under the License.
17
17
package com .cloud .network .lb ;
18
18
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
+
19
37
import com .cloud .exception .InsufficientCapacityException ;
20
38
import com .cloud .exception .InvalidParameterValueException ;
21
39
import com .cloud .exception .ResourceAllocationException ;
35
53
import com .cloud .network .element .LoadBalancingServiceProvider ;
36
54
import com .cloud .offerings .dao .NetworkOfferingServiceMapDao ;
37
55
import com .cloud .user .Account ;
56
+ import com .cloud .user .AccountManager ;
38
57
import com .cloud .user .AccountVO ;
39
- import com .cloud .user .MockAccountManagerImpl ;
40
58
import com .cloud .user .User ;
41
59
import 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 ;
59
60
60
61
public class UpdateLoadBalancerTest {
61
62
@@ -73,7 +74,7 @@ public class UpdateLoadBalancerTest {
73
74
74
75
@ Before
75
76
public void setUp () {
76
- _lbMgr ._accountMgr = new MockAccountManagerImpl ( );
77
+ _lbMgr ._accountMgr = Mockito . mock ( AccountManager . class );
77
78
_lbMgr ._autoScaleVmGroupDao = Mockito .mock (AutoScaleVmGroupDao .class );
78
79
_lbMgr ._networkDao = netDao ;
79
80
_lbMgr ._networkOfferingServiceDao = ntwkOffServiceMapDao ;
0 commit comments