[test] Add moar unit tests!