QRunIO/qqq
2
0
Fork 0
mirror of https://github.com/Kingsrook/qqq.git synced 2025-07-17 20:50:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

CTLE-436: attempt to add more test coverage

Browse Source
This commit is contained in:
Tim Chamberlain
2023-06-29 16:03:54 -05:00
parent 905ac6d72a
commit 53b74fb61d
3 changed files with 107 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qqq-backend-module-api/src/main/java/com/kingsrook/qqq/backend/module/api/actions/BaseAPIActionUtil.java
View File

@ -787,7 +787,7 @@ public class BaseAPIActionUtil
** Helper method to create a value for an Authentication header, using just a
** username & password - encoded as Basic + base64(username:password)
*******************************************************************************/
protected String getBasicAuthenticationHeader(String username, String password)
public String getBasicAuthenticationHeader(String username, String password)
{
return "Basic " + Base64.getEncoder().encodeToString((username + ":" + password).getBytes());
}

25
qqq-backend-module-api/src/test/java/com/kingsrook/qqq/backend/module/api/TestUtils.java
View File

@ -66,6 +66,8 @@ public class TestUtils
qInstance.addBackend(defineEasypostBackend());
qInstance.addTable(defineTableEasypostTracker());
qInstance.addTable(defineVariant());
return (qInstance);
}
@ -151,6 +153,29 @@ public class TestUtils
/*******************************************************************************
**
*******************************************************************************/
private static QTableMetaData defineVariant()
{
return (new QTableMetaData()
.withName("variant")
.withBackendName(MEMORY_BACKEND_NAME)
.withField(new QFieldMetaData("id", QFieldType.INTEGER))
.withField(new QFieldMetaData("type", QFieldType.STRING))
.withField(new QFieldMetaData("apiKey", QFieldType.STRING))
.withField(new QFieldMetaData("username", QFieldType.STRING))
.withField(new QFieldMetaData("password", QFieldType.STRING))
.withPrimaryKeyField("id")
.withBackendDetails(new APITableBackendDetails()
.withTablePath("variant")
.withTableWrapperObjectName("variant")
)
);
}
/*******************************************************************************
**
*******************************************************************************/

82
qqq-backend-module-api/src/test/java/com/kingsrook/qqq/backend/module/api/actions/BaseAPIActionUtilTest.java
View File

@ -22,12 +22,14 @@
package com.kingsrook.qqq.backend.module.api.actions;
import java.io.Serializable;
import java.nio.charset.StandardCharsets;
import java.util.Base64;
import java.util.List;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import com.kingsrook.qqq.backend.core.actions.tables.CountAction;
import com.kingsrook.qqq.backend.core.actions.tables.DeleteAction;
import com.kingsrook.qqq.backend.core.actions.tables.GetAction;
import com.kingsrook.qqq.backend.core.actions.tables.InsertAction;
import com.kingsrook.qqq.backend.core.actions.tables.QueryAction;
@ -36,6 +38,7 @@ import com.kingsrook.qqq.backend.core.context.QContext;
import com.kingsrook.qqq.backend.core.exceptions.QException;
import com.kingsrook.qqq.backend.core.model.actions.tables.count.CountInput;
import com.kingsrook.qqq.backend.core.model.actions.tables.count.CountOutput;
import com.kingsrook.qqq.backend.core.model.actions.tables.delete.DeleteInput;
import com.kingsrook.qqq.backend.core.model.actions.tables.get.GetInput;
import com.kingsrook.qqq.backend.core.model.actions.tables.get.GetOutput;
import com.kingsrook.qqq.backend.core.model.actions.tables.insert.InsertInput;
@ -59,6 +62,7 @@ import com.kingsrook.qqq.backend.module.api.model.OutboundAPILog;
import com.kingsrook.qqq.backend.module.api.model.OutboundAPILogMetaDataProvider;
import com.kingsrook.qqq.backend.module.api.model.metadata.APIBackendMetaData;
import org.apache.http.Header;
import org.apache.http.client.methods.HttpGet;
import org.json.JSONArray;
import org.json.JSONObject;
import org.junit.jupiter.api.BeforeEach;
@ -81,11 +85,21 @@ class BaseAPIActionUtilTest extends BaseTest
**
*******************************************************************************/
@BeforeEach
void beforeEach()
void beforeEach() throws QException
{
mockApiUtilsHelper = new MockApiUtilsHelper();
mockApiUtilsHelper.setUseMock(true);
MockApiActionUtils.mockApiUtilsHelper = mockApiUtilsHelper;
QueryInput queryInput = new QueryInput();
queryInput.setTableName("variant");
QueryOutput output = new QueryAction().execute(queryInput);
List<Serializable> ids = output.getRecords().stream().map(r -> r.getValue("id")).toList();
DeleteInput deleteInput = new DeleteInput();
deleteInput.setTableName("variant");
deleteInput.setPrimaryKeys(ids);
new DeleteAction().execute(deleteInput);
}
@ -603,6 +617,72 @@ class BaseAPIActionUtilTest extends BaseTest
/*******************************************************************************
**
*******************************************************************************/
@Test
void testBackendWithVariantsApiKey() throws QException
{
APIBackendMetaData backend = (APIBackendMetaData) QContext.getQInstance().getBackend(TestUtils.MOCK_BACKEND_NAME);
backend.setAuthorizationType(AuthorizationType.API_KEY_HEADER);
backend.setUsesVariants(true);
backend.setVariantOptionsTableName("variant");
backend.setVariantOptionsTableIdField("id");
backend.setVariantOptionsTableApiKeyField("apiKey");
backend.setVariantOptionsTableTypeValue("API_KEY_TYPE");
InsertInput insertInput = new InsertInput();
insertInput.setTableName("variant");
insertInput.setRecords(List.of(new QRecord()
.withValue("type", "API_KEY_TYPE")
.withValue("apiKey", "abcdefg1234567")));
InsertOutput insertOutput = new InsertAction().execute(insertInput);
QContext.getQSession().setBackendVariants(Map.of("API_KEY_TYPE", insertOutput.getRecords().get(0).getValue("id")));
HttpGet httpGet = new HttpGet();
BaseAPIActionUtil util = new BaseAPIActionUtil();
util.setBackendMetaData(backend);
util.setupAuthorizationInRequest(httpGet);
Header authHeader = httpGet.getFirstHeader("API-Key");
assertTrue(authHeader.getValue().startsWith("abcde"));
}
/*******************************************************************************
**
*******************************************************************************/
@Test
void testBackendWithVariantsUsernamePassword() throws QException
{
APIBackendMetaData backend = (APIBackendMetaData) QContext.getQInstance().getBackend(TestUtils.MOCK_BACKEND_NAME);
backend.setAuthorizationType(AuthorizationType.BASIC_AUTH_USERNAME_PASSWORD);
backend.setUsesVariants(true);
backend.setVariantOptionsTableName("variant");
backend.setVariantOptionsTableIdField("id");
backend.setVariantOptionsTableUsernameField("username");
backend.setVariantOptionsTablePasswordField("password");
backend.setVariantOptionsTableTypeValue("USER_PASS");
InsertInput insertInput = new InsertInput();
insertInput.setTableName("variant");
insertInput.setRecords(List.of(new QRecord()
.withValue("type", "USER_PASS")
.withValue("username", "user")
.withValue("password", "pass")));
InsertOutput insertOutput = new InsertAction().execute(insertInput);
QContext.getQSession().setBackendVariants(Map.of("USER_PASS", insertOutput.getRecords().get(0).getValue("id")));
HttpGet httpGet = new HttpGet();
BaseAPIActionUtil util = new BaseAPIActionUtil();
util.setBackendMetaData(backend);
util.setupAuthorizationInRequest(httpGet);
Header authHeader = httpGet.getFirstHeader("Authorization");
assertTrue(authHeader.getValue().equals(util.getBasicAuthenticationHeader("user", "pass")));
}
/*******************************************************************************
**
*******************************************************************************/