use of com.amplifyframework.auth.AuthCategory in project amplify-android by aws-amplify.
the class MultiAuthSyncEngineInstrumentationTest method configure.
/**
* Method used to configure each scenario.
* @param modelType The model type.
* @param signInToCognito Does the test scenario require the user to be logged in with user pools.
* @param signInWithOidc Does the test scenario require the user to be logged in with an OIDC provider.
* @param expectedAuthType The auth type that should succeed for the test.
* @throws AmplifyException No expected.
* @throws IOException Not expected.
*/
private void configure(Class<? extends Model> modelType, boolean signInToCognito, boolean signInWithOidc, AuthorizationType expectedAuthType) throws AmplifyException, IOException {
Amplify.addPlugin(new AndroidLoggingPlugin(LogLevel.VERBOSE));
String tag = modelType.getSimpleName();
MultiAuthTestModelProvider modelProvider = MultiAuthTestModelProvider.getInstance(Collections.singletonList(modelType));
SchemaRegistry schemaRegistry = SchemaRegistry.instance();
ModelSchema modelSchema = ModelSchema.fromModelClass(modelType);
schemaRegistry.register(modelType.getSimpleName(), modelSchema);
StrictMode.enable();
Context context = getApplicationContext();
@RawRes int configResourceId = Resources.getRawResourceId(context, "amplifyconfiguration");
AmplifyConfiguration amplifyConfiguration = AmplifyConfiguration.fromConfigFile(context, configResourceId);
readCredsFromConfig(context);
// Setup an auth plugin
CategoryConfiguration authCategoryConfiguration = amplifyConfiguration.forCategoryType(CategoryType.AUTH);
// Turn off persistence so the mobile client's state for one test does not interfere with the others.
try {
authCategoryConfiguration.getPluginConfig("awsCognitoAuthPlugin").getJSONObject("Auth").getJSONObject("Default").put("Persistence", false);
} catch (JSONException exception) {
exception.printStackTrace();
fail();
return;
}
AuthCategory authCategory = new AuthCategory();
AWSCognitoAuthPlugin authPlugin = new AWSCognitoAuthPlugin();
authCategory.addPlugin(authPlugin);
authCategory.configure(authCategoryConfiguration, context);
auth = SynchronousAuth.delegatingTo(authCategory);
if (signInToCognito) {
Log.v(tag, "Test requires signIn.");
AuthSignInResult authSignInResult = auth.signIn(cognitoUser, cognitoPassword);
if (!authSignInResult.isSignInComplete()) {
fail("Unable to complete initial sign-in");
}
}
if (signInWithOidc) {
oidcLogin();
if (token.get() == null) {
fail("Unable to autenticate with OIDC provider");
}
}
// Setup an API
DefaultCognitoUserPoolsAuthProvider cognitoProvider = new DefaultCognitoUserPoolsAuthProvider(authPlugin.getEscapeHatch());
CategoryConfiguration apiCategoryConfiguration = amplifyConfiguration.forCategoryType(CategoryType.API);
ApiAuthProviders apiAuthProviders = ApiAuthProviders.builder().cognitoUserPoolsAuthProvider(cognitoProvider).awsCredentialsProvider(authPlugin.getEscapeHatch()).oidcAuthProvider(token::get).build();
ApiCategory apiCategory = new ApiCategory();
requestInterceptor = new HttpRequestInterceptor(expectedAuthType);
apiCategory.addPlugin(AWSApiPlugin.builder().configureClient("DataStoreIntegTestsApi", okHttpClientBuilder -> okHttpClientBuilder.addInterceptor(requestInterceptor)).apiAuthProviders(apiAuthProviders).build());
apiCategory.configure(apiCategoryConfiguration, context);
api = SynchronousApi.delegatingTo(apiCategory);
// Setup DataStore
DataStoreConfiguration dsConfig = DataStoreConfiguration.builder().errorHandler(exception -> Log.e(tag, "DataStore error handler received an error.", exception)).syncExpression(modelSchema.getName(), () -> Where.id("FAKE_ID").getQueryPredicate()).build();
CategoryConfiguration dataStoreCategoryConfiguration = AmplifyConfiguration.fromConfigFile(context, configResourceId).forCategoryType(CategoryType.DATASTORE);
String databaseName = "IntegTest" + modelType.getSimpleName() + ".db";
SQLiteStorageAdapter sqLiteStorageAdapter = TestStorageAdapter.create(schemaRegistry, modelProvider, databaseName);
AWSDataStorePlugin awsDataStorePlugin = AWSDataStorePlugin.builder().storageAdapter(sqLiteStorageAdapter).modelProvider(modelProvider).apiCategory(apiCategory).authModeStrategy(AuthModeStrategyType.MULTIAUTH).schemaRegistry(schemaRegistry).dataStoreConfiguration(dsConfig).build();
DataStoreCategory dataStoreCategory = new DataStoreCategory();
dataStoreCategory.addPlugin(awsDataStorePlugin);
dataStoreCategory.configure(dataStoreCategoryConfiguration, context);
dataStoreCategory.initialize(context);
dataStore = SynchronousDataStore.delegatingTo(dataStoreCategory);
}
use of com.amplifyframework.auth.AuthCategory in project amplify-android by aws-amplify.
the class AuthComponentConfigureTest method setup.
/**
* Get all setup for the future tests with mocks and a standard response for tokens.
* @throws AmplifyException If add plugin fails
*/
@Before
public void setup() throws AmplifyException {
mobileClient = mock(AWSMobileClient.class);
authCategory = new AuthCategory();
authCategory.addPlugin(new AWSCognitoAuthPlugin(mobileClient, USER_SUB));
}
use of com.amplifyframework.auth.AuthCategory in project amplify-android by aws-amplify.
the class AuthComponentTest method setup.
/**
* Get all setup for the future tests with mocks and a standard response for tokens.
* @throws AmplifyException If add plugin fails
* @throws JSONException has to be declared as part of creating a test JSON object
*/
@Before
public void setup() throws AmplifyException, JSONException {
mobileClient = mock(AWSMobileClient.class);
authCategory = new AuthCategory();
authCategory.addPlugin(new AWSCognitoAuthPlugin(mobileClient, USER_SUB));
Map<String, String> additionalInfoMap = Collections.singletonMap("token", ACCESS_TOKEN);
UserStateDetails userStateDetails = new UserStateDetails(UserState.SIGNED_IN, additionalInfoMap);
Context context = getApplicationContext();
JSONObject pluginConfig = new JSONObject().put("TestKey", "TestVal");
pluginConfig.put("UserAgentOverride", UserAgent.string());
JSONObject json = new JSONObject().put("plugins", new JSONObject().put(PLUGIN_KEY, pluginConfig));
AuthCategoryConfiguration authConfig = new AuthCategoryConfiguration();
authConfig.populateFromJSON(json);
doAnswer(invocation -> {
Callback<UserStateDetails> callback = invocation.getArgument(2);
callback.onResult(userStateDetails);
return null;
}).when(mobileClient).initialize(any(), any(), any());
authCategory.configure(authConfig, context);
authCategory.initialize(context);
synchronousAuth = SynchronousAuth.delegatingTo(authCategory);
}
use of com.amplifyframework.auth.AuthCategory in project amplify-android by aws-amplify.
the class TestCategory method forPlugin.
/**
* Creates a test category to be used for testing. A category containing given
* plugin will be used to register and configure the plugin. The category
* instance will have the same Amplify category-type as registered plugin,
* and it should be cast appropriately before usage.
*
* <pre>
* AuthCategory auth = (AuthCategory) TestCategory.forPlugin(plugin);
* </pre>
*
* Resulting category object should not be used in production.
*
* @param plugin Plugin to register in the category. The resulting category
* will share the category type with this plugin.
* @param <P> Type of plugin being registered.
* @return An Amplify category object with the given plugin registered and
* configured. The category will be of same type as registered plugin
* and it should be cast to correct type before usage.
* @throws AmplifyException if category fails to configure given plugin.
*/
public static <P extends Plugin<?>> Category<?> forPlugin(P plugin) throws AmplifyException {
CategoryType categoryType = plugin.getCategoryType();
Category<Plugin<?>> category = (Category<Plugin<?>>) fromCategoryType(categoryType);
category.addPlugin(plugin);
CategoryConfiguration config = AmplifyConfiguration.fromConfigFile(context()).forCategoryType(categoryType);
category.configure(config, context());
category.initialize(context());
return category;
}
Aggregations