diff options
author | Marek Gradzki <mgradzki@cisco.com> | 2016-09-18 15:17:02 +0200 |
---|---|---|
committer | Maros Marsalek <mmarsale@cisco.com> | 2016-09-20 15:55:04 +0000 |
commit | 7a7a31ae0a390113399119835bdaa3a4ed694170 (patch) | |
tree | ea5c6bbf21f5d1829d4d6698dc50cc5f48bc096e /v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/test/ContextTestUtils.java | |
parent | 27eba3a20b98378a07178722606752960978ea5a (diff) |
HONEYCOMB-116: refactor v3po tests to use CTU.mockMapping
Change-Id: I72d3a545d66e1280e34beff72adcdf876dae86fd
Signed-off-by: Marek Gradzki <mgradzki@cisco.com>
Diffstat (limited to 'v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/test/ContextTestUtils.java')
-rw-r--r-- | v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/test/ContextTestUtils.java | 34 |
1 files changed, 25 insertions, 9 deletions
diff --git a/v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/test/ContextTestUtils.java b/v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/test/ContextTestUtils.java index 3b637a44c..9bfcd4d82 100644 --- a/v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/test/ContextTestUtils.java +++ b/v3po/v3po2vpp/src/test/java/io/fd/honeycomb/translate/v3po/test/ContextTestUtils.java @@ -45,14 +45,8 @@ public class ContextTestUtils { new NamingContextKey(namingContextName)).child(Mappings.class).child(Mapping.class, new MappingKey(name)); } - public static void mockMapping(final MappingContext mappingContext, final String name, final int id, - final String namingContextName) { - final InstanceIdentifier<Mappings> mappingsIid = - getMappingIid(name, namingContextName).firstIdentifierOf(Mappings.class); - - final Optional<Mapping> singleMapping = getMapping(name, id); + private static List<Mapping> getMappingList(final MappingContext mappingContext, final InstanceIdentifier<Mappings> mappingsIid) { final Optional<Mappings> previousMappings = mappingContext.read(mappingsIid); - final MappingsBuilder mappingsBuilder; if (previousMappings != null && previousMappings.isPresent()) { mappingsBuilder = new MappingsBuilder(previousMappings.get()); @@ -61,10 +55,32 @@ public class ContextTestUtils { mappingsBuilder.setMapping(Lists.newArrayList()); } - final List<Mapping> mappingList = mappingsBuilder.getMapping(); + return mappingsBuilder.getMapping(); + } + + public static void mockMapping(final MappingContext mappingContext, final String name, final int id, + final String namingContextName) { + final InstanceIdentifier<Mappings> mappingsIid = + getMappingIid(name, namingContextName).firstIdentifierOf(Mappings.class); + + final Optional<Mapping> singleMapping = getMapping(name, id); + final List<Mapping> mappingList = getMappingList(mappingContext, mappingsIid); mappingList.add(singleMapping.get()); - doReturn(Optional.of(mappingsBuilder.setMapping(mappingList).build())) + + doReturn(Optional.of(new MappingsBuilder().setMapping(mappingList).build())) .when(mappingContext).read(mappingsIid); doReturn(singleMapping).when(mappingContext).read(getMappingIid(name, namingContextName)); } + + public static void mockEmptyMapping(final MappingContext mappingContext, final String name, + final String namingContextName) { + final InstanceIdentifier<Mappings> mappingsIid = + getMappingIid(name, namingContextName).firstIdentifierOf(Mappings.class); + + final List<Mapping> mappingList = getMappingList(mappingContext, mappingsIid); + + doReturn(Optional.of(new MappingsBuilder().setMapping(mappingList).build())) + .when(mappingContext).read(mappingsIid); + doReturn(Optional.absent()).when(mappingContext).read(getMappingIid(name, namingContextName)); + } } |