/* * Copyright 2017 TWO SIGMA OPEN SOURCE, LLC * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.twosigma.beaker.jvm.serialization; import com.fasterxml.jackson.databind.JsonNode; import com.twosigma.beaker.KernelTest; import com.twosigma.beaker.jupyter.KernelManager; import com.twosigma.beaker.jvm.object.SimpleLayoutManager; import org.assertj.core.api.Assertions; import org.junit.After; import org.junit.Before; import org.junit.BeforeClass; import org.junit.Test; import java.io.IOException; public class SimpleLayoutManagerSerializerTest { private SimpleLayoutManager layoutManager; private static SimpleLayoutManagerSerializer serializer; private static SerializationTestHelper<SimpleLayoutManagerSerializer, SimpleLayoutManager> helper; @BeforeClass public static void setUpClass() throws IOException { serializer = new SimpleLayoutManagerSerializer(() -> { return new BasicObjectSerializer(); } ); helper = new SerializationTestHelper<>(serializer); } @Before public void setUp() throws Exception { KernelManager.register(new KernelTest()); layoutManager = new SimpleLayoutManager(); } @After public void tearDown() throws Exception { KernelManager.register(null); } @Test public void serializeSimpleLayoutManager_resultJsonHasType() throws IOException { //when JsonNode actualObj = helper.serializeObject(layoutManager); //then Assertions.assertThat(actualObj.has("type")).isTrue(); Assertions.assertThat(actualObj.get("type").asText()).isEqualTo("SimpleLayoutManager"); } @Test public void serializeBorderDisplayed_resultJsonHasBorderDisplayed() throws IOException { //given layoutManager.setBorderDisplayed(true); //when JsonNode actualObj = helper.serializeObject(layoutManager); //then Assertions.assertThat(actualObj.has("borderDisplayed")).isTrue(); Assertions.assertThat(actualObj.get("borderDisplayed").asBoolean()).isTrue(); } }