Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
G
gitlab-ce
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
Analytics
Analytics
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Commits
Issue Boards
Open sidebar
Léo-Paul Géneau
gitlab-ce
Commits
388abbd1
Commit
388abbd1
authored
Jul 11, 2017
by
Grzegorz Bizon
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Extract background migratons queue class method
parent
c17b1d5f
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
29 additions
and
21 deletions
+29
-21
lib/gitlab/background_migration.rb
lib/gitlab/background_migration.rb
+5
-2
spec/lib/gitlab/background_migration_spec.rb
spec/lib/gitlab/background_migration_spec.rb
+24
-19
No files found.
lib/gitlab/background_migration.rb
View file @
388abbd1
module
Gitlab
module
BackgroundMigration
def
self
.
queue
BackgroundMigrationWorker
.
sidekiq_options
[
'queue'
]
end
# Begins stealing jobs from the background migrations queue, blocking the
# caller until all jobs have been completed.
#
# steal_class - The name of the class for which to steal jobs.
def
self
.
steal
(
steal_class
)
queue
=
Sidekiq
::
Queue
.
new
(
BackgroundMigrationWorker
.
sidekiq_options
[
'queue'
])
queue
=
Sidekiq
::
Queue
.
new
(
self
.
queue
)
queue
.
each
do
|
job
|
migration_class
,
migration_args
=
job
.
args
...
...
spec/lib/gitlab/background_migration_spec.rb
View file @
388abbd1
require
'spec_helper'
describe
Gitlab
::
BackgroundMigration
do
describe
'.steal'
do
it
'steals jobs from a queue'
do
queue
=
[
double
(
:job
,
args:
[
'Foo'
,
[
10
,
20
]])]
allow
(
Sidekiq
::
Queue
).
to
receive
(
:new
)
.
with
(
BackgroundMigrationWorker
.
sidekiq_options
[
'queue'
])
.
and_return
(
queue
)
describe
'.queue'
do
it
'returns background migration worker queue'
do
expect
(
described_class
.
queue
)
.
to
eq
BackgroundMigrationWorker
.
sidekiq_options
[
'queue'
]
end
end
expect
(
queue
[
0
]).
to
receive
(
:delete
)
describe
'.steal'
do
context
'when there are enqueued jobs present'
do
let
(
:queue
)
{
[
double
(
:job
,
args:
[
'Foo'
,
[
10
,
20
]])]
}
expect
(
described_class
).
to
receive
(
:perform
).
with
(
'Foo'
,
[
10
,
20
])
before
do
allow
(
Sidekiq
::
Queue
).
to
receive
(
:new
)
.
with
(
described_class
.
queue
)
.
and_return
(
queue
)
end
described_class
.
steal
(
'Foo'
)
end
it
'steals jobs from a queue'
do
expect
(
queue
[
0
]).
to
receive
(
:delete
)
it
'does not steal jobs for a different migration'
do
queue
=
[
double
(
:job
,
args:
[
'Foo'
,
[
10
,
20
]])]
expect
(
described_class
).
to
receive
(
:perform
).
with
(
'Foo'
,
[
10
,
20
])
allow
(
Sidekiq
::
Queue
).
to
receive
(
:new
)
.
with
(
BackgroundMigrationWorker
.
sidekiq_options
[
'queue'
])
.
and_return
(
queue
)
described_class
.
steal
(
'Foo'
)
end
expect
(
described_class
).
not_to
receive
(
:perform
)
it
'does not steal jobs for a different migration'
do
expect
(
described_class
).
not_to
receive
(
:perform
)
expect
(
queue
[
0
]).
not_to
receive
(
:delete
)
expect
(
queue
[
0
]).
not_to
receive
(
:delete
)
described_class
.
steal
(
'Bar'
)
described_class
.
steal
(
'Bar'
)
end
end
end
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment