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.

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