import os
from pathlib import Path
import sys

def main():
    workdir, task = sys.argv[1:]
    src = Path(workdir) / "src"
    for srcfile in src.iterdir():
        if srcfile.name == "heredoc":
            assert(not srcfile.is_symlink())
            with open(srcfile, "r") as f:
                local_contents = f.readline()
                assert(local_contents.strip() == 'log foo')
        else:
            assert(srcfile.is_symlink() == (task == "link"))

if __name__ == "__main__":
    main()