Test cases for the datasource class.
Test cases for controller.
Merge pull request #2 from mnorbury/unit_tests

Unit tests