feat(search): implemented locale code search; improved tests

Author: s1moe2

src/index.js

@@ -1,9 +1,11 @@
 const {
   validateLocaleCode,
   findCountryLanguages,
+  findCountryLocales,
 } = require('./lib')
 
 module.exports = {
   validateLocaleCode,
   findCountryLanguages,
+  findCountryLocales,
 }

src/lib/index.js

@@ -10,11 +10,41 @@ const validateLocaleCode = (localeCode) => {
   return _.findIndex(data, (l) => l.code === localeCode) !== -1
 }
 
+/**
+ * Takes a country code and returns the list of languages valid for it
+ * @param {string} countryCode - Country code (e.g. "PT")
+ * @return {string[]} Languages list
+ */
 const findCountryLanguages = (countryCode) => {
-  return _.filter(data, (l) => l.countryCode === countryCode)
+  const countryEntries = _.filter(data, (l) => l.countryCode === countryCode)
+  const langSet = new Set()
+
+  for (let i = 0; i < countryEntries.length; i++) {
+    langSet.add(countryEntries[i].langCode)
+  }
+
+  return Array.from(langSet)
 }
 
+/**
+ * Takes a country code and returns the list of locales valid for it
+ * @param {string} countryCode - Country code (e.g. "PT")
+ * @return {string[]} Locales list
+ */
+const findCountryLocales = (countryCode) => {
+  const countryEntries = _.filter(data, (l) => l.countryCode === countryCode)
+  const localeSet = new Set()
+
+  for (let i = 0; i < countryEntries.length; i++) {
+    localeSet.add(countryEntries[i].code)
+  }
+
+  return Array.from(localeSet)
+}
+
+
 module.exports = {
   validateLocaleCode,
   findCountryLanguages,
+  findCountryLocales,
 }

test/index.js

@@ -4,6 +4,7 @@ const { assert, expect } = require('chai')
 const {
   validateLocaleCode,
   findCountryLanguages,
+  findCountryLocales,
 } = require('../src')
 
 describe('Validation tests.', () => {
@@ -24,11 +25,12 @@ describe('Validation tests.', () => {
   })
 })
 
-describe('Code search tests.', () => {
+describe('Language code by country search tests.', () => {
   it('should return a list of languages', (done) => {
     const res = findCountryLanguages('US')
     expect(res).to.be.an('array')
     expect(res).to.have.lengthOf(2)
+    expect(res).to.deep.equal(['en', 'es'])
     done()
   })
 
@@ -47,3 +49,50 @@ describe('Code search tests.', () => {
   })
 })
 
+describe('Locale code by country search tests.', () => {
+  it('should return a list of locales', (done) => {
+    const res = findCountryLocales('BE')
+    expect(res).to.be.an('array')
+    expect(res).to.have.lengthOf(4)
+    expect(res).to.deep.equal(['de-BE', 'en-BE', 'fr-BE', 'nl-BE'])
+    done()
+  })
+
+  it('should return an empty list', (done) => {
+    const res = findCountryLocales('YYY')
+    expect(res).to.be.an('array')
+    expect(res).to.have.lengthOf(0)
+    done()
+  })
+
+  it('should return an empty list again', (done) => {
+    const res = findCountryLocales(77236)
+    expect(res).to.be.an('array')
+    expect(res).to.have.lengthOf(0)
+    done()
+  })
+})
+
+describe('Locale code by country search tests.', () => {
+  it('should return a list of locales', (done) => {
+    const res = findCountryLocales('BE')
+    expect(res).to.be.an('array')
+    expect(res).to.have.lengthOf(4)
+    expect(res).to.deep.equal(['de-BE', 'en-BE', 'fr-BE', 'nl-BE'])
+    done()
+  })
+
+  it('should return an empty list', (done) => {
+    const res = findCountryLocales('YYY')
+    expect(res).to.be.an('array')
+    expect(res).to.have.lengthOf(0)
+    done()
+  })
+
+  it('should return an empty list again', (done) => {
+    const res = findCountryLocales(77236)
+    expect(res).to.be.an('array')
+    expect(res).to.have.lengthOf(0)
+    done()
+  })
+})