Add unittest for data handlers