Make popout-square fall back to flat-square

badges · Jul 22, 2019 · 702c8e6 · 702c8e6
1 parent 774f86a
commit 702c8e6
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/gh-badges/lib/make-badge.js b/gh-badges/lib/make-badge.js
@@ -131,7 +131,11 @@ module.exports = function makeBadge({
   }
 
   if (!(template in templates)) {
-    template = 'flat'
+    if (template && template.endsWith('-square')) {
+      template = 'flat-square'
+    } else {
+      template = 'flat'
+    }
   }
   if (template === 'social') {
     left = capitalize(left)

diff --git a/gh-badges/lib/make-badge.spec.js b/gh-badges/lib/make-badge.spec.js
@@ -122,6 +122,22 @@ describe('The badge generator', function() {
         .to.equal(jsonBadgeWithDefaultStyle)
         .and.to.satisfy(isSvg)
     })
+
+    it('should replace "popout-square" svg template with "flat-square"', function() {
+      const jsonBadgeWithUnknownStyle = makeBadge({
+        text: ['name', 'Bob'],
+        format: 'svg',
+        template: 'popout-square',
+      })
+      const jsonBadgeWithDefaultStyle = makeBadge({
+        text: ['name', 'Bob'],
+        format: 'svg',
+        template: 'flat-square',
+      })
+      expect(jsonBadgeWithUnknownStyle)
+        .to.equal(jsonBadgeWithDefaultStyle)
+        .and.to.satisfy(isSvg)
+    })
   })
 
   describe('"for-the-badge" template badge generation', function() {