diff options
author | Aleksandr <ognelisar@gmail.com> | 2017-09-28 14:32:21 +0700 |
---|---|---|
committer | Aleksandr <ognelisar@gmail.com> | 2017-09-28 14:32:21 +0700 |
commit | 1f569ac1a31f88334c25976d94e7c495a7bbde80 (patch) | |
tree | 584162d344db1ea767adb72dc6a67164a1e89e68 /src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala | |
parent | d8e71e0a9ee7db58032384d059403bc227a35138 (diff) | |
download | rest-query-1f569ac1a31f88334c25976d94e7c495a7bbde80.tar.gz rest-query-1f569ac1a31f88334c25976d94e7c495a7bbde80.tar.bz2 rest-query-1f569ac1a31f88334c25976d94e7c495a7bbde80.zip |
Implemented all generator for REP's entities
Diffstat (limited to 'src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala')
-rw-r--r-- | src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala | 35 |
1 files changed, 30 insertions, 5 deletions
diff --git a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala index 48eaf79..512e324 100644 --- a/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala +++ b/src/main/scala/xyz/driver/pdsuidomain/fakes/entities/rep/ExtractedDataGen.scala @@ -1,12 +1,17 @@ package xyz.driver.pdsuidomain.fakes.entities.rep + import xyz.driver.core.generators._ -import xyz.driver.pdsuidomain.entities.{Document, ExtractedData} +import xyz.driver.pdsuicommon.domain.{LongId, TextJson} +import xyz.driver.pdsuidomain.entities._ import xyz.driver.pdsuidomain.entities.ExtractedData.Meta import xyz.driver.pdsuidomain.fakes.entities.common._ +import xyz.driver.pdsuidomain.services.ExtractedDataService.RichExtractedData object ExtractedDataGen { + private val maxItemsInCollectionNumber = 50 + private val maxPageNumber = 100 private val maxIndexNumber = 100 private val maxOffsetNumber = 10 @@ -62,13 +67,33 @@ object ExtractedDataGen { ) } - def nextExtractedData() = { - ExtractedData.apply( + def nextExtractedData(documentId: LongId[Document]): ExtractedData = { + ExtractedData( id = nextLongId[ExtractedData], - documentId = nextLongId[Document], + documentId = documentId, keywordId = nextOption(nextLongId[xyz.driver.pdsuidomain.entities.Keyword]), evidenceText = nextOption(nextString()), - ??? + meta = nextOption(TextJson(nextExtractedDataMeta())) + ) + } + + + def nextExtractedDataLabel(): ExtractedDataLabel = { + ExtractedDataLabel( + id = nextLongId[ExtractedDataLabel], + dataId = nextLongId[ExtractedData], + labelId = nextOption(nextLongId[Label]), + categoryId = nextOption(nextLongId[Category]), + value = nextOption(Common.nextFuzzyValue()) + ) + } + + def nextRichExtractedData(documentId: LongId[Document]): RichExtractedData = { + RichExtractedData( + extractedData = nextExtractedData(documentId), + labels = List.fill( + nextInt(maxItemsInCollectionNumber, minValue = 0) + )(nextExtractedDataLabel()) ) } } |