Skip to content

Commit

Permalink
Add Redis to the universal integration tests (#1784)
Browse files Browse the repository at this point in the history
* Add Redis to the universal integration tests

Signed-off-by: Achal Shah <achals@gmail.com>

* Static config

Signed-off-by: Achal Shah <achals@gmail.com>
  • Loading branch information
achals authored Aug 17, 2021
1 parent c0fe653 commit 90559fc
Showing 1 changed file with 24 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -36,17 +36,37 @@ class TestRepoConfig:
infer_event_timestamp_col: bool = True


def ds_creator_path(cls: str):
return f"tests.integration.feature_repos.universal.data_sources.{cls}"


DYNAMO_CONFIG = {"type": "dynamodb", "region": "us-west-2"}
REDIS_CONFIG = {"type": "redis", "connection_string": "localhost:6379,db=0"}
FULL_REPO_CONFIGS: List[TestRepoConfig] = [
TestRepoConfig(), # Local
# Local configurations
TestRepoConfig(),
TestRepoConfig(online_store=REDIS_CONFIG),
# GCP configurations
TestRepoConfig(
provider="gcp",
offline_store_creator="tests.integration.feature_repos.universal.data_sources.bigquery.BigQueryDataSourceCreator",
offline_store_creator=ds_creator_path("bigquery.BigQueryDataSourceCreator"),
online_store="datastore",
),
TestRepoConfig(
provider="gcp",
offline_store_creator=ds_creator_path("bigquery.BigQueryDataSourceCreator"),
online_store=REDIS_CONFIG,
),
# AWS configurations
TestRepoConfig(
provider="aws",
offline_store_creator=ds_creator_path("redshift.RedshiftDataSourceCreator"),
online_store=DYNAMO_CONFIG,
),
TestRepoConfig(
provider="aws",
offline_store_creator="tests.integration.feature_repos.universal.data_sources.redshift.RedshiftDataSourceCreator",
online_store={"type": "dynamodb", "region": "us-west-2"},
offline_store_creator=ds_creator_path("redshift.RedshiftDataSourceCreator"),
online_store=REDIS_CONFIG,
),
]

Expand Down

0 comments on commit 90559fc

Please sign in to comment.