PdfPig/src/UglyToad.PdfPig.Tests/TestObjectLocationProvider.cs

namespace UglyToad.PdfPig.Tests
{
    using System.Collections.Generic;
    using PdfPig.Tokenization.Scanner;
    using PdfPig.Tokens;

    internal class TestObjectLocationProvider : IObjectLocationProvider
    {
        public Dictionary<IndirectReference, long> Offsets { get; } = new Dictionary<IndirectReference, long>();

        public bool TryGetOffset(IndirectReference reference, out long offset)
        {
            return Offsets.TryGetValue(reference, out offset);
        }

        public void UpdateOffset(IndirectReference reference, long offset)
        {
            Offsets[reference] = offset;
        }

        public bool TryGetCached(IndirectReference reference, out ObjectToken objectToken)
        {
            objectToken = null;
            return false;
        }

        public void Cache(ObjectToken objectToken)
        {
        }
    }
}