Add the rollback test in the data pipeline (#337)

Add the rollback test in the data pipeline and move the bug snapshot test to a pytest test

.taskcluster.yml

@@ -98,29 +98,6 @@ tasks:
           owner: ${user}@users.noreply.github.com
           source: ${repository}/raw/${head_rev}/.taskcluster.yml
 
-      - taskId: {$eval: as_slugid("rollback_test_task")}
-        created: {$fromNow: ''}
-        deadline: {$fromNow: '1 hour'}
-        provisionerId: aws-provisioner-v1
-        workerType: github-worker
-        payload:
-          maxRunTime: 3600
-          image: python
-          command:
-            - "/bin/bash"
-            - "-lcx"
-            - "git clone ${repository} &&
-               cd bugbug &&
-               git checkout ${head_rev} &&
-               pip install -r requirements.txt &&
-               python -c 'from bugbug import bugzilla, db; db.download()' &&
-               python -m bugbug.bug_snapshot"
-        metadata:
-          name: bugbug rollback test
-          description: bugbug rollback test
-          owner: ${user}@users.noreply.github.com
-          source: ${repository}/raw/${head_rev}/.taskcluster.yml
-
       - taskId: {$eval: as_slugid("packaging_test_task")}
         created: {$fromNow: ''}
         deadline: {$fromNow: '1 hour'}

infra/data-pipeline.yml

@@ -56,6 +56,30 @@ tasks:
       owner: release-mgmt-analysis@mozilla.com
       source: https://github.com/mozilla/bugbug/raw/master/data-pipeline.yml
 
+  - ID: rollback-test-task
+    created: {$fromNow: ''}
+    deadline: {$fromNow: '25 hours'}
+    provisionerId: aws-provisioner-v1
+    workerType: releng-svc
+    dependencies:
+      - bugs-retrieval
+    payload:
+      maxRunTime: 3600
+      image: mozilla/bugbug-base
+      command:
+        - "/bin/bash"
+        - "-lcx"
+        - "python -c 'from bugbug import bugzilla, db; db.download()' &&
+           python -m bugbug.bug_snapshot"
+    routes:
+      - notify.email.release-mgmt-analysis@mozilla.com.on-failed"
+      - notify.irc-channel.#bugbug.on-failed
+    metadata:
+      name: bugbug rollback test
+      description: bugbug rollback test
+      owner: release-mgmt-analysis@mozilla.com
+      source: https://github.com/mozilla/bugbug/raw/master/data-pipeline.yml
+
   - ID: train-component
     created: {$fromNow: ''}
     deadline: {$fromNow: '48 hours'}
@@ -170,6 +194,7 @@ tasks:
     provisionerId: aws-provisioner-v1
     workerType: releng-svc
     dependencies:
+      - rollback-test-task
       - train-tracking
       - train-regression
       - train-defect

tests/test_bug_snapshot.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at http://mozilla.org/MPL/2.0/.
+
+
+from bugbug import bugzilla
+from bugbug.bug_snapshot import rollback
+
+
+def test_bug_snapshot():
+    for i, bug in enumerate(bugzilla.get_bugs()):
+        print(bug["id"])
+        print(i)
+
+        rollback(bug, None, False)