@Test
public void testGetTrafficPolicyVersion() {

    ListTrafficPolicyInstancesResult listTrafficPolicyInstancesResult = PowerMockito.mock(ListTrafficPolicyInstancesResult.class);
    Mockito.when(mockAmazonRoute53.listTrafficPolicyInstances()).thenReturn(listTrafficPolicyInstancesResult);
    TrafficPolicyInstance trafficPolicyInstance = new TrafficPolicyInstance();
    trafficPolicyInstance.setName("testPolicy");
    trafficPolicyInstance.setTrafficPolicyVersion(1);
    trafficPolicyInstance.setTrafficPolicyId("abc");
    ListTrafficPoliciesResult listTrafficPoliciesResult = PowerMockito.mock(ListTrafficPoliciesResult.class);
    Mockito.when(mockAmazonRoute53.listTrafficPolicies()).thenReturn(listTrafficPoliciesResult);
    TrafficPolicySummary trafficPolicySummary = new TrafficPolicySummary();
    trafficPolicySummary.setName("testPolicy");
    trafficPolicySummary.setId("abc");
    Mockito.when(listTrafficPoliciesResult.getTrafficPolicySummaries()).thenReturn(Arrays.asList(trafficPolicySummary));
    Mockito.when(listTrafficPolicyInstancesResult.getTrafficPolicyInstances()).thenReturn(Arrays.asList(trafficPolicyInstance));
    int policyVersion = route53RaiderUnderTest.getTrafficPolicyVersion("testPolicy");
    Assert.assertEquals(policyVersion,1);
}