diff --git a/src/test/java/com/knecon/fforesight/databasetenantcommons/utils/TestTenantProvider.java b/src/test/java/com/knecon/fforesight/databasetenantcommons/utils/TestTenantProvider.java index 6855245..b5c4c0b 100644 --- a/src/test/java/com/knecon/fforesight/databasetenantcommons/utils/TestTenantProvider.java +++ b/src/test/java/com/knecon/fforesight/databasetenantcommons/utils/TestTenantProvider.java @@ -5,6 +5,7 @@ import java.util.List; import org.springframework.stereotype.Service; +import com.knecon.fforesight.tenantcommons.TenantApplicationType; import com.knecon.fforesight.tenantcommons.TenantProvider; import com.knecon.fforesight.tenantcommons.listener.ITenantEventHandler; import com.knecon.fforesight.tenantcommons.model.TenantCreatedEvent; @@ -26,7 +27,10 @@ public class TestTenantProvider implements TenantProvider { @Override public TenantResponse getTenant(String s) { - return tenants.stream().filter(t -> t.getTenantId().equals(s)).findFirst().orElse(null); + return tenants.stream() + .filter(t -> t.getTenantId().equals(s)) + .findFirst() + .orElse(null); } @@ -37,6 +41,17 @@ public class TestTenantProvider implements TenantProvider { } + @Override + public TenantApplicationType getTenantApplicationType(String s) { + + return tenants.stream() + .filter(t -> t.getTenantId().equals(s)) + .findFirst() + .map(TenantResponse::getApplicationType) + .orElse(TenantApplicationType.RedactManager); + } + + public void addTenant(TenantResponse tenantResponse) { this.tenants.add(tenantResponse);