Implement QueryAllBuildNumbersSource #131

flosell · Oct 22, 2016 · a11abd4 · a11abd4
1 parent bb191b0
commit a11abd4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/src/clj/lambdacd/internal/default_pipeline_state.clj b/src/clj/lambdacd/internal/default_pipeline_state.clj
@@ -65,7 +65,12 @@
     (get @state-atom build-number))
   protocols/NextBuildNumberSource
   (next-build-number [self]
-    (inc (most-recent-build-number-in-state @state-atom))))
+    (inc (most-recent-build-number-in-state @state-atom)))
+  protocols/QueryAllBuildNumbersSource
+  (all-build-numbers [self]
+    (-> @state-atom
+        (keys)
+        (sort))))
 
 (defn new-default-pipeline-state [config & {:keys [initial-state-for-testing]}]
   (let [home-dir   (:home-dir config)

diff --git a/test/clj/lambdacd/internal/default_pipeline_state_test.clj b/test/clj/lambdacd/internal/default_pipeline_state_test.clj
@@ -12,8 +12,15 @@
 
 (deftest next-build-number-test
   (testing "that the next buildnumber is the highest build-number currently in the pipeline-state"
-    (is (= 1 (next-build-number (new-default-pipeline-state {:home-dir (utils/create-temp-dir)}))))
-    (is (= 5 (next-build-number (new-default-pipeline-state {:home-dir (utils/create-temp-dir)} :initial-state-for-testing { 3 {} 4 {} 1 {}}))))))
+    (is (= 1 (protocols/next-build-number (new-default-pipeline-state {:home-dir (utils/create-temp-dir)}))))
+    (is (= 5 (protocols/next-build-number (new-default-pipeline-state {:home-dir (utils/create-temp-dir)}
+                                                                      :initial-state-for-testing {3 {} 4 {} 1 {}}))))))
+
+(deftest all-build-numbers-test
+  (testing "that we can get a list of all build numbers"
+    (is (= [] (protocols/all-build-numbers (new-default-pipeline-state {:home-dir (utils/create-temp-dir)}))))
+    (is (= [1 3 4] (protocols/all-build-numbers (new-default-pipeline-state {:home-dir (utils/create-temp-dir)}
+                                                                            :initial-state-for-testing (sorted-map-by > 3 {} 4 {} 1 {})))))))
 
 (deftest pipeline-structure-state-test
   (testing "that we can consume and retrieve pipeline-structure"