adapters/mso-openstack-adapters/src/test/java/db/migration/CloudConfigMigrationTest.java

   1 package db.migration;
   2
   3 import org.junit.Assert;
   4 import org.junit.Before;
   5 import org.junit.Test;
   6 import org.onap.so.adapters.vnf.BaseRestTestUtils;
   7 import org.springframework.beans.factory.annotation.Autowired;
   8 import org.springframework.beans.factory.annotation.Qualifier;
   9
  10 import javax.sql.DataSource;
  11 import java.sql.Connection;
  12 import java.sql.ResultSet;
  13 import java.sql.Statement;
  14
  15 public class CloudConfigMigrationTest extends BaseRestTestUtils {
  16
  17     @Qualifier("dataSource")
  18     @Autowired
  19     DataSource dataSource;
  20
  21     R__CloudConfigMigration cloudConfigMigration;
  22
  23     @Before
  24     public void setup() {
  25         cloudConfigMigration = new R__CloudConfigMigration();
  26     }
  27
  28     @Test
  29     public void testMigrate() throws Exception {
  30         System.setProperty("spring.profiles.active", "test");
  31         cloudConfigMigration.migrate(dataSource.getConnection());
  32         assertMigratedIdentityServiceData();
  33         assertMigratedCloudSiteData();
  34         assertMigratedCloudManagerData();
  35     }
  36
  37     @Test
  38     public void testMigrateNoData() throws Exception {
  39         System.setProperty("spring.profiles.active", "nomigrate");
  40         int identityCount = getDataCount("identity_services");
  41         int cloudSiteCount = getDataCount("cloud_sites");
  42         int cloudManagerCount = getDataCount("cloudify_managers");
  43
  44         cloudConfigMigration.migrate(dataSource.getConnection());
  45
  46         Assert.assertEquals(identityCount, getDataCount("identity_services"));
  47         Assert.assertEquals(cloudSiteCount, getDataCount("cloud_sites"));
  48         Assert.assertEquals(cloudManagerCount, getDataCount("cloudify_managers"));
  49     }
  50
  51
  52     private int getDataCount(String tableName) throws Exception {
  53         try (Connection con = dataSource.getConnection(); Statement stmt = con.createStatement(); ResultSet rs = stmt.executeQuery("select count(1) from " + tableName)) {
  54             while (rs.next()) {
  55                 return rs.getInt(1);
  56             }
  57         }
  58         return 0;
  59     }
  60
  61     private void assertMigratedIdentityServiceData() throws Exception {
  62         try (Connection con = dataSource.getConnection(); Statement stmt = con.createStatement(); ResultSet rs = stmt.executeQuery("select * from identity_services where id='MTKEYSTONE'")) {
  63             boolean dataAvailable = false;
  64             while (rs.next()) {
  65                 dataAvailable = true;
  66                 Assert.assertEquals("MTKEYSTONE", rs.getString("id"));
  67                 Assert.assertEquals("http://localhost:5000/v2.0", rs.getString("identity_url"));
  68                 Assert.assertEquals("john", rs.getString("mso_id"));
  69                 Assert.assertEquals("313DECE408AF7759D442D7B06DD9A6AA", rs.getString("mso_pass"));
  70                 Assert.assertEquals("admin", rs.getString("admin_tenant"));
  71                 Assert.assertEquals("_member_", rs.getString("member_role"));
  72                 Assert.assertEquals("KEYSTONE", rs.getString("identity_server_type"));
  73                 Assert.assertEquals("USERNAME_PASSWORD", rs.getString("identity_authentication_type"));
  74             }
  75             Assert.assertTrue("Expected data in identity_services table post migration but didnt find any!!!", dataAvailable);
  76         }
  77     }
  78
  79     private void assertMigratedCloudSiteData() throws Exception {
  80         try (Connection con = dataSource.getConnection(); Statement stmt = con.createStatement(); ResultSet rs = stmt.executeQuery("select * from cloud_sites where id='regionOne'")) {
  81             boolean dataAvailable = false;
  82             while (rs.next()) {
  83                 dataAvailable = true;
  84                 Assert.assertEquals("regionOne", rs.getString("id"));
  85                 Assert.assertEquals("regionOne", rs.getString("region_id"));
  86                 Assert.assertEquals("MT2", rs.getString("clli"));
  87                 Assert.assertEquals("2.5", rs.getString("cloud_version"));
  88                 Assert.assertEquals("MTKEYSTONE", rs.getString("identity_service_id"));
  89             }
  90             Assert.assertTrue("Expected data in identity_services table post migration but didnt find any!!!", dataAvailable);
  91         }
  92     }
  93
  94     private void assertMigratedCloudManagerData() throws Exception {
  95         try (Connection con = dataSource.getConnection(); Statement stmt = con.createStatement(); ResultSet rs = stmt.executeQuery("select * from cloudify_managers where id='manager'")) {
  96             boolean dataAvailable = false;
  97             while (rs.next()) {
  98                 dataAvailable = true;
  99                 Assert.assertEquals("http://localhost:8080", rs.getString("cloudify_url"));
 100                 Assert.assertEquals("user", rs.getString("username"));
 101                 Assert.assertEquals("password", rs.getString("password"));
 102                 Assert.assertEquals("2.0", rs.getString("version"));
 103             }
 104             Assert.assertTrue("Expected data in identity_services table post migration but didnt find any!!!", dataAvailable);
 105         }
 106     }
 107 }