Add method for getting badge claim codes.

mozilla · Apr 23, 2013 · 4e47ef5 · 4e47ef5
1 parent a4c2e33
commit 4e47ef5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/models/badge.js b/models/badge.js
@@ -293,12 +293,6 @@ Badge.prototype.generateClaimCodes = function generateClaimCodes(options, callba
   });
 };
 
-/**
- * Find an claim code by name
- *
- * @param {String} code
- * @return {Object|Null}
- */
 
 Badge.prototype.getClaimCode = function getClaimCode(code) {
   const codes = this.claimCodes;
@@ -311,6 +305,13 @@ Badge.prototype.getClaimCode = function getClaimCode(code) {
   return null;
 };
 
+Badge.prototype.getClaimCodes = function getClaimCodes() {
+  const codes = this.claimCodes;
+  return codes.map(function (entry) {
+    return entry.code;
+  });
+};
+
 /**
  * Whether or not an claim code is claimed
  *

diff --git a/tests/badge-model.test.js b/tests/badge-model.test.js
@@ -231,14 +231,21 @@ test.applyFixtures(fixtures, function () {
     });
   });
 
-  test('Badge.hasClaimCode', function (t) {
+  test('Badge#hasClaimCode', function (t) {
     const badge = fixtures['offline-badge'];
     const code = 'will-claim';
     t.same(badge.hasClaimCode(code), true);
     t.same(badge.hasClaimCode('nopeniopenope'), false);
     t.end();
   });
 
+  test('Badge#getClaimCodes', function (t) {
+    const badge = fixtures['offline-badge'];
+    const expect = [ 'already-claimed', 'never-claim', 'will-claim', 'remove-claim' ];
+    t.same(badge.getClaimCodes(), expect);
+    t.end();
+  });
+
   test('Badge#addClaimCodes', function (t) {
     const badge = fixtures['offline-badge'];
     const codes = ['lethargic-hummingbird', 'woeful-turtle'];