1. /*
2.  * Copyright 2017, Pearson Education, Learning Technology Group
3.  *
4.  * TableCreationServiceTest.java
5.  */
6.  
7. package com.pearson.ltg.ole.event.tracker.service;
8.  
9. import static org.junit.Assert.assertEquals;
10. import static org.mockito.Mockito.verify;
11. import static org.mockito.Mockito.when;
12.  
13. import org.junit.Before;
14. import org.junit.Test;
15. import org.mockito.InjectMocks;
16. import org.mockito.Mock;
17. import org.mockito.MockitoAnnotations;
18. import org.springframework.test.util.ReflectionTestUtils;
19.  
20. import com.pearson.ltg.ole.event.tracker.dbsession.DynamoDBOperations;
21. import com.pearson.ltg.ole.event.tracker.entity.EventTrackerMapping;
22.  
23. public class TableCreationServiceTest {
24.    
25.     @InjectMocks TableCreationService tablecreationservice;
26.     @Mock DynamoDBOperations mockDynamoDBOperations;
27.  
28.     @Before
29.     public void setup() {
30.         MockitoAnnotations.initMocks(this);
31.         ReflectionTestUtils.setField(tablecreationservice, "environment", "dev");
32.         ReflectionTestUtils.setField(tablecreationservice, "eventTrackerTablename", "_realize_webapp_event-tracker");
33.     }
34.  
35.     @Test
36.     public void testConstructEventTrackerTable() {
37.         String expectedTableName = "dev_realize_webapp_event-tracker_Feb_2017";
38.         when(mockDynamoDBOperations.createTable(EventTrackerMapping.class, expectedTableName)).thenReturn(expectedTableName);
39.         String suffixReturnValue = tablecreationservice.constructEventTrackerTable();
40.         assertEquals(expectedTableName, suffixReturnValue);
41.         verify(mockDynamoDBOperations).createTable(EventTrackerMapping.class, expectedTableName);
42.     }
43. }