Add more tests in framework module