1/**
2 * Copyright (C) 2014 Google, Inc.
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.google.inject.persist.jpa;
18
19import junit.framework.TestCase;
20import static org.hamcrest.Matchers.is;
21import static org.junit.Assert.assertThat;
22import static org.mockito.Mockito.doThrow;
23import static org.mockito.Mockito.mock;
24import static org.mockito.Mockito.when;
25
26import java.util.Properties;
27
28import javax.persistence.EntityManager;
29import javax.persistence.EntityManagerFactory;
30import javax.persistence.spi.PersistenceProvider;
31
32public class JpaPersistServiceTest extends TestCase {
33
34  private static final String PERSISTENCE_UNIT_NAME = "test_persistence_unit_name";
35  private static final Properties PERSISTENCE_PROPERTIES = new Properties();
36
37  private final JpaPersistService sut = new JpaPersistService(PERSISTENCE_UNIT_NAME, PERSISTENCE_PROPERTIES);
38  private final PersistenceProvider provider = mock(PersistenceProvider.class);
39  private final EntityManagerFactory factory = mock(EntityManagerFactory.class);
40  private final EntityManager entityManager = mock(EntityManager.class);
41
42  @Override
43  public void setUp() throws Exception {
44    when(provider.createEntityManagerFactory(PERSISTENCE_UNIT_NAME, PERSISTENCE_PROPERTIES)).thenReturn(factory);
45    when(factory.createEntityManager()).thenReturn(entityManager);
46  }
47
48  public void test_givenErrorOnEntityManagerClose_whenEndIsCalled_thenEntityManagerIsRemoved() {
49    sut.start(factory);
50    sut.begin();
51
52    // arrange an exception on sut.end(), which invokes entityManager.close()
53    doThrow(SimulatedException.class).when(entityManager).close();
54    try {
55      sut.end();
56      fail("Exception expected");
57    }
58    catch (SimulatedException expected) {
59      assertThat(sut.isWorking(), is(false));
60    }
61  }
62
63  private class SimulatedException extends RuntimeException {
64  }
65}
66