Backported getRecentBuildsets from buildbot 0.8.8 (to use it in RietveldPollerWithCache)

third_party/buildbot_8_4p1/buildbot/test/unit/test_db_buildsets.py

 # This file is part of Buildbot.  Buildbot is free software: you can
 # redistribute it and/or modify it under the terms of the GNU General Public
 # License as published by the Free Software Foundation, version 2.
 #
 # This program is distributed in the hope that it will be useful, but WITHOUT
 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 # FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 # details.
 #
 # You should have received a copy of the GNU General Public License along with
@@ skipping 423 matching lines @@
                 self.db.buildsets.completeBuildset(bsid=93, results=6,
                                                    _reactor=self.clock))
         def cb(_):
             self.fail("should not succeed")
         def eb(f):
             f.trap(KeyError)
             # pass
         d.addCallbacks(cb, eb)
         return d
 
+    def insert_test_getRecentBuildsets_data(self):
+        return self.insertTestData([
+            fakedb.Change(changeid=91, branch='branch_a', repository='repo_a'),
+            fakedb.SourceStamp(id=91, branch='branch_a', repository='repo_a'),
+            fakedb.SourceStampChange(sourcestampid=91, changeid=91),
+
+            fakedb.Buildset(id=91, sourcestampid=91, complete=0,
+                    complete_at=298297875, results=-1, submitted_at=266761875,
+                    external_idstring='extid', reason='rsn1'),
+            fakedb.Buildset(id=92, sourcestampid=91, complete=1,
+                    complete_at=298297876, results=7, submitted_at=266761876,
+                    external_idstring='extid', reason='rsn2'),
+
+            # buildset unrelated to the change
+            fakedb.Buildset(id=93, sourcestampid=1, complete=1,
+                    complete_at=298297877, results=7, submitted_at=266761877,
+                    external_idstring='extid', reason='rsn2'),
+        ])
+
+    def test_getRecentBuildsets_all(self):
+        d = self.insert_test_getRecentBuildsets_data()
+        d.addCallback(lambda _ :
+                self.db.buildsets.getRecentBuildsets(2, branch='branch_a',
+                                                     repository='repo_a'))
+        def check(bsdictlist):
+            self.assertEqual(bsdictlist, [
+              dict(external_idstring='extid', reason='rsn1', sourcestampid=91,
+                submitted_at=datetime.datetime(1978, 6, 15, 12, 31, 15,
+                                               tzinfo=UTC),
+                complete_at=datetime.datetime(1979, 6, 15, 12, 31, 15,
+                                               tzinfo=UTC),
+                complete=False, results=-1, bsid=91),
+              dict(external_idstring='extid', reason='rsn2', sourcestampid=91,
+                submitted_at=datetime.datetime(1978, 6, 15, 12, 31, 16,
+                                               tzinfo=UTC),
+                complete_at=datetime.datetime(1979, 6, 15, 12, 31, 16,
+                                               tzinfo=UTC),
+                complete=True, results=7, bsid=92),
+            ])
+        d.addCallback(check)
+        return d
+
+    def test_getRecentBuildsets_one(self):
+        d = self.insert_test_getRecentBuildsets_data()
+        d.addCallback(lambda _ :
+                self.db.buildsets.getRecentBuildsets(1, branch='branch_a',
+                                                     repository='repo_a'))
+        def check(bsdictlist):
+            self.assertEqual(bsdictlist, [
+              dict(external_idstring='extid', reason='rsn2', sourcestampid=91,
+                submitted_at=datetime.datetime(1978, 6, 15, 12, 31, 16,
+                                               tzinfo=UTC),
+                complete_at=datetime.datetime(1979, 6, 15, 12, 31, 16,
+                                               tzinfo=UTC),
+                complete=True, results=7, bsid=92),
+            ])
+        d.addCallback(check)
+        return d
+
+    def test_getRecentBuildsets_zero(self):
+        d = self.insert_test_getRecentBuildsets_data()
+        d.addCallback(lambda _ :
+                self.db.buildsets.getRecentBuildsets(0, branch='branch_a',
+                                                     repository='repo_a'))
+        def check(bsdictlist):
+            self.assertEqual(bsdictlist, [])
+        d.addCallback(check)
+        return d
+
+    def test_getRecentBuildsets_noBranchMatch(self):
+        d = self.insert_test_getRecentBuildsets_data()
+        d.addCallback(lambda _ :
+                self.db.buildsets.getRecentBuildsets(2, branch='bad_branch',
+                                                     repository='repo_a'))
+        def check(bsdictlist):
+            self.assertEqual(bsdictlist, [])
+        d.addCallback(check)
+        return d
+
+    def test_getRecentBuildsets_noRepoMatch(self):
+          d = self.insert_test_getRecentBuildsets_data()
+          d.addCallback(lambda _ :
+                  self.db.buildsets.getRecentBuildsets(2, branch='branch_a',
+                                                       repository='bad_repo'))
+          def check(bsdictlist):
+              self.assertEqual(bsdictlist, [])
+          d.addCallback(check)
+          return d