You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

score.spec.js 4.7KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161
  1. 'use strict'
  2. const test = require('ava')
  3. const { stub } = require('sinon')
  4. const Hapi = require('@hapi/hapi')
  5. const plugin = require('../lib/plugins/profile')
  6. const Profile = require('../lib/models/profile')
  7. const ZipCode = require('../lib/models/zip-code')
  8. const MatchQueue = require('../lib/models/matchqueue')
  9. const Aspect = require('../lib/models/aspect')
  10. const AspectLabel = require('../lib/models/aspect_label')
  11. // !: Must match the key set in servives/profile.js
  12. const zipcodeKey = 7
  13. /**
  14. * Route parameters
  15. */
  16. const params = {
  17. profile_id: 1,
  18. max_distance: 1000,
  19. }
  20. const mockReturn = {
  21. user: [
  22. {
  23. profile_id: 1,
  24. user: { is_poster: 1 },
  25. responses: [
  26. { response_key_id: 1, val: '100' },
  27. { response_key_id: 2, val: '200' },
  28. { response_key_id: zipcodeKey, val: '90065' },
  29. ],
  30. },
  31. {
  32. profile_id: 2,
  33. user: { is_poster: 0 },
  34. responses: [
  35. { response_key_id: 1, val: '100' },
  36. { response_key_id: 2, val: '200' },
  37. { response_key_id: zipcodeKey, val: '97002' },
  38. ],
  39. },
  40. {
  41. profile_id: 3,
  42. user: { is_poster: 0 },
  43. responses: [
  44. { response_key_id: 1, val: '200' },
  45. { response_key_id: 2, val: '200' },
  46. { response_key_id: zipcodeKey, val: '96741' },
  47. ],
  48. },
  49. ],
  50. labels: [
  51. { aspect_id: 1, a: 100, b: 100 },
  52. { aspect_id: 2, a: 100, b: 200 },
  53. { aspect_id: 3, a: 200, b: 200 },
  54. { aspect_id: 4, a: 200, b: 100 },
  55. ],
  56. aspects: [
  57. { aspect_id: 1, 1: 111, 2: 112, 3: 113, 4: 114 },
  58. { aspect_id: 2, 1: 222, 2: 222, 3: 333, 4: 222 },
  59. { aspect_id: 3, 1: 111, 2: 112, 3: 113, 4: 114 },
  60. { aspect_id: 4, 1: 111, 2: 111, 3: 111, 4: 111 },
  61. ],
  62. }
  63. const pathToTest = {
  64. method: 'GET',
  65. url: `/${params.profile_id}/score?max_distance=${params.max_distance}`,
  66. }
  67. test(`path ${pathToTest.url} should return ok on GET`, async t => {
  68. /**
  69. * Create a new server and register services,
  70. * models and routes for testing
  71. * -
  72. * NOTE: We use a mocked registerModel() and register
  73. * models manually. Normally this is handled by
  74. * Schwifty at runtime.
  75. */
  76. const server = Hapi.server()
  77. /**
  78. * Overload so we don't register any models
  79. * using the plugin call (see plugins/profile.js)
  80. * and Manually load the model we need for the test
  81. */
  82. server.registerModel = () => {}
  83. server.models = () => ({
  84. Aspect,
  85. AspectLabel,
  86. MatchQueue,
  87. Profile,
  88. ZipCode,
  89. })
  90. server.registrations = {
  91. 'main-app-plugin': {
  92. options: {},
  93. },
  94. }
  95. /**
  96. * Register Routes and Services as usual
  97. */
  98. await plugin.register(server)
  99. /**
  100. * Replace Objection model methods with our own mock functions
  101. * !: Janky - might be better to temp knex sqlite instance
  102. */
  103. stub(server.models()['Profile'], 'query').returns({
  104. // Mocked for pathToTest(), scoreProfilesFor()
  105. findOne: () => ({
  106. withGraphFetched: () => ({
  107. withGraphFetched: () => mockReturn.user[0],
  108. }),
  109. }),
  110. // Mocked for scoreProfilesFor()
  111. withGraphFetched: () => ({
  112. withGraphFetched: () => [mockReturn.user[1], mockReturn.user[2]],
  113. }),
  114. // Mocked for _getProfileIdsForUserId()
  115. where: () => ({}),
  116. // Mocked for getCompleteProfilesFor(), getProfilesFor()
  117. whereIn: () => ({
  118. withGraphFetched: () => ({}),
  119. }),
  120. // Mocked for deleteProfile()
  121. delete: () => ({
  122. where: () => ({}),
  123. }),
  124. })
  125. stub(server.models()['AspectLabel'], 'query').returns(mockReturn.labels)
  126. stub(server.models()['Aspect'], 'query').returns(mockReturn.aspects)
  127. stub(server.models()['ZipCode'], 'query').returns({
  128. // Mocked for _latLonForZip()
  129. findOne: () => ({
  130. latitude: 38.0 + Math.random(),
  131. longitude: -121.0 + Math.random(),
  132. }),
  133. })
  134. stub(server.models()['MatchQueue'], 'query').returns({
  135. patch: () => ({
  136. where: () => ({}),
  137. }),
  138. insert: () => ({}),
  139. where: () => ({
  140. where: () => {
  141. return { profile_id: 99 }
  142. },
  143. }),
  144. })
  145. /**
  146. * Test the server with registered models and services
  147. */
  148. const { payload } = await server.inject(pathToTest)
  149. const res = JSON.parse(payload)
  150. t.deepEqual(res.ok, true)
  151. t.is(res.data.length, 2)
  152. t.is(res.data[0].profile_id, 3)
  153. server.stop()
  154. })