Add test for token store