1 /*
2  *  Make.org Core API
3  *  Copyright (C) 2018 Make.org
4  *
5  * This program is free software: you can redistribute it and/or modify
6  *  it under the terms of the GNU Affero General Public License as
7  *  published by the Free Software Foundation, either version 3 of the
8  *  License, or (at your option) any later version.
9  *
10  *  This program is distributed in the hope that it will be useful,
11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  *  GNU Affero General Public License for more details.
14  *
15  *  You should have received a copy of the GNU Affero General Public License
16  *  along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  *
18  */
19 
20 package org.make.api.technical.generator
21 
22 import cats.data.{NonEmptyList => Nel}
23 import enumeratum.values.scalacheck._
24 import eu.timepit.refined.refineV
25 import eu.timepit.refined.auto._
26 import eu.timepit.refined.collection._
27 import org.make.api.operation.CreateOperationOfQuestion
28 import org.make.api.organisation.OrganisationRegisterData
29 import org.make.api.partner.CreatePartnerRequest
30 import org.make.api.technical.{DefaultIdGeneratorComponent, IdGeneratorComponent}
31 import org.make.api.user.UserRegisterData
32 import org.make.core.operation.OperationId
33 import org.make.core.partner.PartnerKind
34 import org.make.core.profile.{Gender, SocioProfessionalCategory}
35 import org.make.core.question.QuestionId
36 import org.make.core.reference.{Country, Language}
37 import org.make.core.technical.generator.{
38   CustomGenerators,
39   DateGenerators,
40   RichGenerators,
41   EntitiesGen => CoreEntitiesGen
42 }
43 import org.make.core.user.User
44 import org.scalacheck.Arbitrary.arbitrary
45 import org.scalacheck.Gen
46 
47 import java.time.Period
48 
49 trait EntitiesGen extends CoreEntitiesGen with DateGenerators { self: IdGeneratorComponent =>
50 
51   def genCreateOperationOfQuestion(
52     operationId: OperationId,
53     forceCountries: Option[Nel[Country]],
54     forceLanguage: Option[Language]
55   ): Gen[CreateOperationOfQuestion] = {
56     for {
57       startDate <- genDateWithOffset(lowerOffset = Period.ofYears(-3), upperOffset = Period.ofYears(1))
58       endDate <- genDateWithOffset(
59         lowerOffset = Period.ofMonths(1),
60         upperOffset = Period.ofMonths(6),
61         fromDate = startDate
62       )
63       slug                   <- CustomGenerators.LoremIpsumGen.slug(maxLength = Some(30))
64       (countries, languages) <- genCountriesLanguage(forceCountries, forceLanguage)
65       defaultLanguage = languages.head
66       operationTitle <- CustomGenerators.LoremIpsumGen.sentence(maxLength = Some(150))
67       question       <- CustomGenerators.LoremIpsumGen.sentence(maxLength = Some(150))
68       proposalPrefix <- CustomGenerators.LoremIpsumGen.sentence(maxLength = Some(15))
69       featured       <- arbitrary[Boolean]
70     } yield CreateOperationOfQuestion(
71       operationId = operationId,
72       startDate = startDate,
73       endDate = endDate,
74       operationTitle = operationTitle,
75       slug = slug,
76       countries = countries,
77       defaultLanguage = defaultLanguage,
78       question = refineV[NonEmpty].unsafeFrom(question),
79       proposalPrefix = proposalPrefix,
80       featured = featured
81     )
82   }
83 
84   def genUserRegisterData(questionId: Option[QuestionId]): Gen[UserRegisterData] =
85     for {
86       email                     <- CustomGenerators.Mail.gen()
87       firstName                 <- CustomGenerators.FirstName.gen.asOption
88       dateOfBirth               <- genBirthDate.asOption
89       profession                <- CustomGenerators.LoremIpsumGen.sentence(maxLength = Some(15)).asOption
90       postalCode                <- CustomGenerators.PostalCode.gen.asOption
91       gender                    <- arbitrary[Option[Gender]]
92       socioProfessionalCategory <- arbitrary[Option[SocioProfessionalCategory]]
93       (country, language)       <- genCountryLanguage
94       optIn                     <- arbitrary[Option[Boolean]]
95       optInPartner              <- arbitrary[Option[Boolean]]
96       roles                     <- genRoles
97       politicalParty            <- CustomGenerators.LoremIpsumGen.word.asOption
98       publicProfile             <- arbitrary[Boolean]
99     } yield UserRegisterData(
100       email = email,
101       firstName = firstName,
102       lastName = None,
103       password = Some(email),
104       lastIp = None,
105       dateOfBirth = dateOfBirth,
106       profession = profession,
107       postalCode = postalCode,
108       gender = gender,
109       socioProfessionalCategory = socioProfessionalCategory,
110       country = country,
111       language = language,
112       crmCountry = country,
113       crmLanguage = language,
114       questionId = questionId,
115       optIn = optIn,
116       optInPartner = optInPartner,
117       roles = roles,
118       availableQuestions = questionId.toList,
119       politicalParty = politicalParty,
120       website = None,
121       publicProfile = publicProfile,
122       privacyPolicyApprovalDate = None
123     )
124 
125   def genOrganisationRegisterData: Gen[OrganisationRegisterData] =
126     for {
127       name                <- CustomGenerators.LoremIpsumGen.word
128       email               <- CustomGenerators.Mail.gen(Some("organisation"))
129       avatar              <- CustomGenerators.ImageUrl.gen(width = 100, height = 100)
130       description         <- Gen.option(CustomGenerators.LoremIpsumGen.sentence(maxLength = Some(300)))
131       (country, language) <- genCountryLanguage
132       website             <- Gen.option(CustomGenerators.URL.gen)
133     } yield OrganisationRegisterData(
134       name = name,
135       email = email,
136       password = Some(email),
137       avatar = Some(avatar),
138       description = description,
139       country = country,
140       language = language,
141       website = website
142     )
143 
144   def genCreatePartnerRequest(orga: Option[User], questionId: QuestionId): Gen[CreatePartnerRequest] =
145     for {
146       name <- CustomGenerators.LoremIpsumGen.word
147       logo <- CustomGenerators.ImageUrl.gen(width = 100, height = 100).asOption
148       link <- CustomGenerators.URL.gen.asOption
149       partnerKind <- Gen.frequency(
150         (2, PartnerKind.Media),
151         (7, PartnerKind.ActionPartner),
152         (1, PartnerKind.Founder),
153         (4, PartnerKind.Actor)
154       )
155       weight <- Gen.posNum[Float]
156     } yield CreatePartnerRequest(
157       name = orga.flatMap(_.displayName).getOrElse(name),
158       logo = logo,
159       link = link,
160       organisationId = orga.map(_.userId),
161       partnerKind = partnerKind,
162       questionId = questionId,
163       weight = weight
164     )
165 
166 }
167 
168 object EntitiesGen extends EntitiesGen with DefaultIdGeneratorComponent