Replace Sone mock methods with a mock builder.

[Sone.git] / src / test / java / net / pterodactylus / sone / fcp / GetLocalSonesCommandTest.java

diff --git a/src/test/java/net/pterodactylus/sone/fcp/GetLocalSonesCommandTest.java b/src/test/java/net/pterodactylus/sone/fcp/GetLocalSonesCommandTest.java
index 002ca9b..0ed5204 100644 (file)
--- a/src/test/java/net/pterodactylus/sone/fcp/GetLocalSonesCommandTest.java
+++ b/src/test/java/net/pterodactylus/sone/fcp/GetLocalSonesCommandTest.java
@@ -53,9 +53,13 @@ public class GetLocalSonesCommandTest {
 
        @Test
        public void verifyThatOnlyLocalSonesAreReturned() throws FSParseException {
-               Collection<Sone> localSones = asList(mocks.mockLocalSone("LSone1"), mocks.mockLocalSone("LSone2"), mocks.mockLocalSone("LSone3"));
-               mocks.mockRemoteSone("RSone1");
-               mocks.mockRemoteSone("RSone2");
+               Collection<Sone> localSones = asList(
+                               mocks.mockSone("LSone1").local().create(),
+                               mocks.mockSone("LSone2").local().create(),
+                               mocks.mockSone("LSone3").local().create()
+               );
+               mocks.mockSone("RSone1").create();
+               mocks.mockSone("RSone2").create();
                SimpleFieldSet getLocalSonesFieldSet = new SimpleFieldSetBuilder().put("Message", "GetLocalSones").get();
                Response response = getLocalSonesCommand.execute(getLocalSonesFieldSet, null, DIRECT);
                assertThat(response, notNullValue());