Added FlowViewModel tests for loading messages