Mirror/MarginaliaSearch
1
0
Fork 0
mirror of https://github.com/MarginaliaSearch/MarginaliaSearch.git synced 2025-02-23 21:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

(search) Imrpove search-service paperdoll by mocking suggestions and news

Browse Source
This commit is contained in:
Viktor Lofgren 2024-05-06 19:52:13 +02:00
parent bcbb9afac0
commit 29a4d3df23
1 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
code/services-application/search-service/test/nu/marginalia/search/paperdoll/SearchServicePaperDoll.java
View File

@ -1,5 +1,6 @@
package nu.marginalia.search.paperdoll;
import com.google.gson.Gson;
import com.google.inject.AbstractModule;
import com.google.inject.Guice;
import com.zaxxer.hikari.HikariConfig;
@ -17,6 +18,7 @@ import nu.marginalia.index.query.limit.QueryStrategy;
import nu.marginalia.index.query.limit.SpecificationLimit;
import nu.marginalia.model.EdgeUrl;
import nu.marginalia.model.crawl.HtmlFeature;
import nu.marginalia.model.gson.GsonFactory;
import nu.marginalia.search.SearchModule;
import nu.marginalia.search.SearchService;
import nu.marginalia.service.ServiceId;
@ -31,8 +33,10 @@ import org.mockito.Mockito;
import org.testcontainers.containers.MariaDBContainer;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.junit.jupiter.Testcontainers;
import spark.Spark;
import java.net.URISyntaxException;
import java.sql.SQLException;
import java.util.*;
import static org.mockito.ArgumentMatchers.any;
@ -66,6 +70,7 @@ public class SearchServicePaperDoll extends AbstractModule {
private static List<DecoratedSearchResultItem> results = new ArrayList<>();
private static QueryResponse searchResponse;
private static final Gson gson = GsonFactory.get();
@SneakyThrows
void registerSearchResult(
@ -112,6 +117,33 @@ public class SearchServicePaperDoll extends AbstractModule {
System.setProperty("service-name", "search");
System.setProperty("search.websiteUrl", "http://localhost:9999/");
try (var conn = dataSource.getConnection();
var ps = conn.prepareStatement("""
INSERT INTO SEARCH_NEWS_FEED(TITLE, LINK, SOURCE, LIST_DATE)
VALUES (?, ?, ?, ?)
""")) {
ps.setString(1, "Lex Luthor elected president");
ps.setString(2, "https://www.example.com/foo");
ps.setString(3, "Daily Planet");
ps.setDate(4, new java.sql.Date(System.currentTimeMillis()));
ps.execute();
ps.setString(1, "Besieged Alesian onlookers confused as Caesar builds a wall around his wall around the city walls");
ps.setString(2, "https://www.example2.com/bar");
ps.setString(3, "The Gaulish Observer");
ps.setDate(4, new java.sql.Date(System.currentTimeMillis()));
ps.execute();
ps.setString(1, "Marginalia acquires Google");
ps.setString(2, "https://www.example3.com/baz");
ps.setString(3, "The Dependent");
ps.setDate(4, new java.sql.Date(System.currentTimeMillis()));
ps.execute();
} catch (SQLException e) {
e.printStackTrace();
}
searchResponse = new QueryResponse(
new SearchSpecification(new SearchQuery(), List.of(), "", "test",
SpecificationLimit.none(),
@ -142,6 +174,13 @@ public class SearchServicePaperDoll extends AbstractModule {
injector.getInstance(SearchService.class);
List<String> suggestions = List.of("foo", "bar", "baz");
Spark.get("/suggest/", (rq, rsp) -> {
rsp.type("application/json");
return gson.toJson(suggestions);
});
registerSearchResult("https://www.example.com/foo", "Foo", "Lorem ipsum dolor sit amet", Set.of(), 0.5, 0.5, ~0L);
registerSearchResult("https://www.example2.com/bar", "Bar", "Some text goes here", Set.of(), 0.5, 0.5, 1L);
registerSearchResult("https://www.example3.com/baz", "All HTML Features", "This one's got every feature", EnumSet.allOf(HtmlFeature.class), 0.5, 0.5, 1L);