/*
 * Copyright 2015, TeamDev Ltd. All rights reserved.
 *
 * Redistribution and use in source and/or binary forms, with or without
 * modification, must retain the above copyright notice and the following
 * disclaimer.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package org.spine3.gae.datastore;

import com.google.appengine.tools.development.testing.LocalDatastoreServiceTestConfig;
import com.google.appengine.tools.development.testing.LocalServiceTestHelper;
import com.google.protobuf.Message;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.spine3.TypeName;
import org.spine3.base.CommandRequest;
import org.spine3.protobuf.JsonFormat;
import org.spine3.testutil.CommandRequestFactory;

import static org.junit.Assert.assertEquals;

@SuppressWarnings("InstanceMethodNamingConvention")
public class DataStoreHelperShould {

    private static final String SOME_RANDOM_STRING = "123qwe";

    private static final LocalServiceTestHelper testHelper =
            new LocalServiceTestHelper(new LocalDatastoreServiceTestConfig());
    private static DataStoreHelper dataStoreHelper;

    private final TypeName commandTypeName = TypeName.of(CommandRequest.getDescriptor());

    @BeforeClass
    public static void setUp() {
        dataStoreHelper = new DataStoreHelper();
        testHelper.setUp();
    }

    @AfterClass
    public static void tearDown() {
        testHelper.tearDown();
    }

    @Test
    public void write_correctly() {

        final CommandRequest commandRequest = CommandRequestFactory.create();
        final String stringId = JsonFormat.printToString(commandRequest.getContext().getCommandId());

        dataStoreHelper.put(Converters.convert(commandRequest));
        final Message readCommandRequest = dataStoreHelper.read(commandTypeName, stringId);
        assertEquals(commandRequest, readCommandRequest);
    }

    @Test(expected = MissingEntityException.class)
    public void fail_read_on_wrong_key() {
        dataStoreHelper.read(commandTypeName, SOME_RANDOM_STRING);
    }
}
Add tests