def test_get_table_metadata_full(self):
self.mox.StubOutWithMock(queuer.db, 'shard_connection')
conn = self.mox.CreateMockAnything()
queuer.db.shard_connection('src', read=True).AndReturn(self.context(conn))
self.mox.StubOutWithMock(queuer.db, 'cursor')
cur = self.mox.CreateMockAnything()
conn.get_current_timestamp().AndReturn(1)
self.tt.update()
self.mox.StubOutWithMock(queuer.QueueChunksWorker, 'get_column_metadata')
queuer.QueueChunksWorker.get_column_metadata(conn)
db.cursor(conn).AndReturn(self.context(cur))
cur.execute(mox.IgnoreArg(), mox.IgnoreArg())
cur.fetchone().AndReturn((1, 2, 3))
self.mox.ReplayAll()
self.config.migration_type = orm.MigrationType.FULL
worker = queuer.QueueChunksWorker(self.config, self.redis)
worker.get_table_metadata()
def test_get_table_metadata_delta(self):
self.config.delta_start = 12345
self.mox.StubOutWithMock(queuer.db, 'shard_connection')
conn = self.mox.CreateMockAnything()
queuer.db.shard_connection('src', read=True).AndReturn(self.context(conn))
self.mox.StubOutWithMock(queuer.db, 'cursor')
cur = self.mox.CreateMockAnything()
conn.get_current_timestamp().AndReturn(1)
self.tt.update()
self.mox.StubOutWithMock(queuer.QueueChunksWorker, 'get_column_metadata')
queuer.QueueChunksWorker.get_column_metadata(conn)
conn.from_unixtime_value(12345).AndReturn('2042T')
db.cursor(conn).AndReturn(self.context(cur))
cur.execute(mox.IgnoreArg(), mox.IgnoreArg())
cur.fetchone().AndReturn((1, 2, 3))
self.mox.ReplayAll()
self.config.migration_type = orm.MigrationType.DELTA
worker = queuer.QueueChunksWorker(self.config, self.redis)
worker.column_map = {'a': {}, 'b': {}, 'c': {}}
worker.get_table_metadata()
