Package org.rhq.server.metrics.migrator.datasources

Examples of org.rhq.server.metrics.migrator.datasources.ScrollableDataSource

62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
    }

    protected ExistingDataSource getExistingDataSource(String query, Task task, DataMigratorConfiguration config) {
        if (Task.Migrate.equals(task)) {
            if (DatabaseType.Oracle.equals(config.getDatabaseType())) {
                return new ScrollableDataSource(config.getEntityManager(), config.getDatabaseType(), query);
            } else {
                if (!config.isExperimentalDataSource()) {
                    return new ScrollableDataSource(config.getEntityManager(), config.getDatabaseType(), query);
                } else {
                    return new ExistingPostgresDataBulkExportSource(config.getEntityManager(), query);
                }
            }
        } else if (Task.Estimate.equals(task)) {
            int limit = CallableMigrationWorker.MAX_RECORDS_TO_LOAD_FROM_SQL
                * (CallableMigrationWorker.NUMBER_OF_BATCHES_FOR_ESTIMATION + 1);

            if (DatabaseType.Oracle.equals(config.getDatabaseType())) {
                return new ScrollableDataSource(config.getEntityManager(), config.getDatabaseType(), query, limit);
            } else {
                if (!config.isExperimentalDataSource()) {
                    return new ScrollableDataSource(config.getEntityManager(), config.getDatabaseType(), query, limit);
                } else {
                    return new ExistingPostgresDataBulkExportSource(config.getEntityManager(), query, limit);
                }
            }
        }

        return new ScrollableDataSource(config.getEntityManager(), config.getDatabaseType(), query);
    }
View Full Code Here
87
88
89
90
91
92
93
94
95
96
97
98
99
100
        org.hibernate.SQLQuery mockQuery = mock(org.hibernate.SQLQuery.class);
        when(mockStatelessSession.createSQLQuery(any(String.class))).thenReturn(mockQuery);

        when(mockQuery.uniqueResult()).thenReturn("1000");

        ScrollableDataSource mockDataSource = mock(ScrollableDataSource.class);
        PowerMockito.whenNew(ScrollableDataSource.class)
            .withArguments(eq(mockEntityManager), eq(databaseType), any(), anyInt()).thenReturn(mockDataSource);
        when(mockDataSource.getData(eq(0), anyInt())).thenReturn(new ArrayList<Object[]>());

        //create object to test and inject required dependencies
        AggregateDataMigrator objectUnderTest = new AggregateDataMigrator(MigrationTable.ONE_HOUR, mockConfig);

        //run code under test
View Full Code Here
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
        org.hibernate.SQLQuery mockQuery = mock(org.hibernate.SQLQuery.class);
        when(mockStatelessSession.createSQLQuery(any(String.class))).thenReturn(mockQuery);

        when(mockQuery.uniqueResult()).thenReturn("1000");

        ScrollableDataSource mockDataSource = mock(ScrollableDataSource.class);
        PowerMockito.whenNew(ScrollableDataSource.class).withArguments(eq(mockEntityManager), eq(databaseType), any())
            .thenReturn(mockDataSource);

        List<Object[]> resultList = new ArrayList<Object[]>();
        resultList.add(new Object[] { 100, 100, 100, 100, 100 });
        resultList.add(new Object[] { 100, System.currentTimeMillis() - 100l, 100, 100, 100 });

        for (int index = 0; index < 15; index++) {
            when(mockDataSource.getData(eq(0), anyInt())).thenReturn(resultList);
            when(mockDataSource.getData(eq(2), anyInt())).thenReturn(new ArrayList<Object[]>());
        }

        ResultSetFuture mockResultSetFuture = mock(ResultSetFuture.class);
        when(mockCassandraSession.executeAsync(any(Query.class))).thenReturn(mockResultSetFuture);
View Full Code Here
86
87
88
89
90
91
92
93
94
95
96
97
98
99
        org.hibernate.SQLQuery mockQuery = mock(org.hibernate.SQLQuery.class);
        when(mockStatelessSession.createSQLQuery(any(String.class))).thenReturn(mockQuery);

        when(mockQuery.uniqueResult()).thenReturn("1000");

        ScrollableDataSource mockDataSource = mock(ScrollableDataSource.class);
        PowerMockito.whenNew(ScrollableDataSource.class)
            .withArguments(eq(mockEntityManager), eq(databaseType), any(), anyInt()).thenReturn(mockDataSource);
        when(mockDataSource.getData(eq(0), anyInt())).thenReturn(new ArrayList<Object[]>());

        //create object to test and inject required dependencies
        RawDataMigrator objectUnderTest = new RawDataMigrator(mockConfig);

        //run code under test
View Full Code Here
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
        org.hibernate.SQLQuery mockQuery = mock(org.hibernate.SQLQuery.class);
        when(mockStatelessSession.createSQLQuery(any(String.class))).thenReturn(mockQuery);

        when(mockQuery.uniqueResult()).thenReturn("1000");

        ScrollableDataSource mockDataSource = mock(ScrollableDataSource.class);
        PowerMockito.whenNew(ScrollableDataSource.class).withArguments(eq(mockEntityManager), eq(databaseType), any())
            .thenReturn(mockDataSource);

        List<Object[]> resultList = new ArrayList<Object[]>();
        resultList.add(new Object[] { 100, 100, 100 });
        resultList.add(new Object[] { 100, System.currentTimeMillis() - 100l, 100 });

        for (int index = 0; index < 15; index++) {
            when(mockDataSource.getData(eq(0), anyInt())).thenReturn(resultList);
            when(mockDataSource.getData(eq(2), anyInt())).thenReturn(new ArrayList<Object[]>());
        }

        ResultSetFuture mockResultSetFuture = mock(ResultSetFuture.class);
        when(mockCassandraSession.executeAsync(any(Query.class))).thenReturn(mockResultSetFuture);
View Full Code Here
TOP

Related Classes of org.rhq.server.metrics.migrator.datasources.ScrollableDataSource

  • org.hibernate.Query
  • org.rhq.server.metrics.migrator.workers.AbstractMigrationWorker
  • org.rhq.server.metrics.migrator.workers.AggregateDataMigratorTest
  • org.rhq.server.metrics.migrator.workers.RawDataMigratorTest
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.