I ended up wrapping all my functions in the CreateInfrastructure function from the unit testing example. Added the CreateInfrastructure function to the main function for provisioning. Then for testing, the CreateInfrastructure returns a struct with every resource type. Now all my templates and tests are in sub folders. Not sure my folder structure the best, but open to suggestions.