Move StateCapture stuff into JUnit
Add shading for our project and custom MANIFEST for agent
Use StateCapture when running tests