From 8c3d850f54855771b962e30139888f065e316ee0 Mon Sep 17 00:00:00 2001 From: "Erik C. Thauvin" Date: Fri, 13 Jul 2018 00:29:01 -0700 Subject: [PATCH] Fixed whois test. --- .../net/thauvin/erik/mobibot/modules/LookupTest.java | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/test/java/net/thauvin/erik/mobibot/modules/LookupTest.java b/src/test/java/net/thauvin/erik/mobibot/modules/LookupTest.java index 18c3972..4b14261 100644 --- a/src/test/java/net/thauvin/erik/mobibot/modules/LookupTest.java +++ b/src/test/java/net/thauvin/erik/mobibot/modules/LookupTest.java @@ -33,6 +33,8 @@ package net.thauvin.erik.mobibot.modules; import org.testng.annotations.Test; +import java.util.Arrays; + import static org.assertj.core.api.Assertions.assertThat; /** @@ -46,13 +48,14 @@ public class LookupTest { @Test public void testLookup() throws Exception { final String result = Lookup.lookup("erik.thauvin.net"); - assertThat(result).as("lookup(erik.thauvin.net)").contains("104.31.77.12"); + assertThat(result).as("lookup(erik.thauvin.net/104.31.77.12)").contains("104.31.77.12"); } @Test public void testWhois() throws Exception { final String[] result = Lookup.whois("17.178.96.59", Lookup.WHOIS_HOST); - assertThat(result).as("whois(17.178.96.59)") - .contains("Apple Inc. APPLE-WWNET (NET-17-0-0-0-1) 17.0.0.0 - 17.255.255.255"); + + assertThat(Arrays.asList(result).stream().anyMatch(m -> m.contains("Apple Inc."))) + .as("whois(17.178.96.59/Apple Inc.").isTrue(); } -} \ No newline at end of file +}