Function test_disabling_cache

src/secrets/src/cache.rs:260–297

()

    #[mz_ore::test(tokio::test)]
    async fn test_disabling_cache() {
        let controller = InMemorySecretsController::new();
        let testing_reader = TestingSecretsReader::new(controller.reader());
        let caching_reader = CachingSecretsReader::new(Arc::new(testing_reader.clone()));

        let secret = [42, 42, 42, 42];
        let id = CatalogItemId::User(1);

        // Store a value.
        controller.ensure(id, &secret).await.expect("success");

        // Read twice, which should hit the cache the second time.
        caching_reader.read(id).await.expect("success");
        caching_reader.read(id).await.expect("success");

        let reads = testing_reader.drain();
        assert_eq!(reads.len(), 1);

        // Disable caching.
        caching_reader.disable_caching();

        // Read twice again, both should hit the source.
        caching_reader.read(id).await.expect("success");
        caching_reader.read(id).await.expect("success");

        let reads = testing_reader.drain();
        assert_eq!(reads.len(), 2);

        // Re-enable caching.
        caching_reader.enable_caching();

        // Read twice again, first should populate the cache, second should hit it.
        caching_reader.read(id).await.expect("success");
        caching_reader.read(id).await.expect("success");

        let reads = testing_reader.drain();
        assert_eq!(reads.len(), 1);
    }

    #[mz_ore::test(tokio::test)]
    async fn updating_cache_values() {