diff --git a/classes/board.php b/classes/board.php
index 56098a4..43ec001 100644
--- a/classes/board.php
+++ b/classes/board.php
@@ -559,8 +559,9 @@ public static function store_note_file($noteid, $attachment) {
      * @return bool
      */
     public static function valid_for_upload($attachment) {
-        $fileextension = strtolower(array_pop(explode('.', basename($attachment['filename']))));
-        if (!in_array($fileextension, explode(',', self::ACCEPTED_FILE_EXTENSIONS))) {
+        $fileparts = explode('.', basename($attachment['filename']));
+        $fileextension = strtolower(array_pop($fileparts));
+        if (!in_array($fileextension, explode(',', ACCEPTED_FILE_EXTENSIONS))) {
             return false;
         }
         $filelength = strlen($attachment['filecontents']);
diff --git a/tests/board_test.php b/tests/board_test.php
new file mode 100644
index 0000000..fae24f6
--- /dev/null
+++ b/tests/board_test.php
@@ -0,0 +1,533 @@
+<?php
+// This file is part of Moodle - http://moodle.org/
+//
+// Moodle is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// Moodle is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
+
+namespace mod_board;
+
+use mod_board\board;
+
+/**
+ * Class board_test
+ *
+ * @package    mod_board
+ * @copyright  2020 onward: Brickfield Education Labs, www.brickfield.ie
+ * @author     Jay Churchward (jay@brickfieldlabs.ie)
+ * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
+ * @group mod_boards
+ */
+
+ class board_test extends \advanced_testcase {
+    public function test_coursemodule_for_board() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+
+        $result = board::coursemodule_for_board($board);
+        $this->assertEquals($result->instance, $board->id);
+    }
+
+    public function test_get_board() {
+        $this->resetAfterTest();
+        $board = self::add_board(2);
+        $output = board::get_board($board->id);
+
+        $this->assertEquals($board->id, $output->id);
+    }
+
+    public function test_get_column() {
+        $this->resetAfterTest();
+        $board = self::add_board(2);
+        $column = self::add_column($board->id);
+        $output = board::get_column($column->id);
+
+        $this->assertEquals($column->id, $output->id);
+    }
+
+    public function test_get_note() {
+        $this->resetAfterTest();
+        $board = self::add_board(2);
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+        $output = board::get_note($note->id);
+
+        $this->assertEquals($note->id, $output->id);
+    }
+
+    public function test_get_note_rating() {
+        $this->resetAfterTest();
+        $board = self::add_board(2);
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+        self::add_note_rating($note->id, 2);
+        $output = board::get_note_rating($note->id);
+
+        $this->assertEquals(1, $output);
+
+        self::add_note_rating($note->id, 3);
+        $output = board::get_note_rating($note->id);
+
+        $this->assertEquals(2, $output);
+    }
+
+    public function test_context_for_board() {
+        $this->resetAfterTest();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $output = board::context_for_board($board->id);
+
+        $this->assertEquals($board->cmid, $output->instanceid);
+    }
+
+    public function test_context_for_column() {
+        global $DB;
+        $this->resetAfterTest();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $output = board::context_for_column($column->id);
+
+        $this->assertEquals($board->cmid, $output->instanceid);
+    }
+
+    public function test_clear_history() {
+        global $DB;
+        $this->resetAfterTest();
+        $board = self::add_board(1);
+        $record = array(
+            'id' => 1,
+            'boardid' => $board->id,
+            'groupid' => 1,
+            'action' => 'action',
+            'userid' => 1,
+            'content' => 'content',
+            'timecreated' => 0
+        );
+        $DB->insert_record('board_history', $record);
+
+        $record = $DB->get_record('board_history', array('boardid' => $board->id));
+        $this->assertEquals($record->content, 'content');
+
+        board::clear_history();
+        $record = $DB->get_record('board_history', array('boardid' => $board->id));
+        $this->assertFalse($record);
+    }
+
+    public function test_board_hide_headers() {
+        $this->resetAfterTest();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+
+        $result = board::board_hide_headers($board->id);
+        $this->assertFalse($result);
+
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id, 'hideheaders' => 1));
+        $result = board::board_hide_headers($board->id);
+        $this->assertTrue($result);
+
+        $this->setAdminUser();
+        $result = board::board_hide_headers($board->id);
+        $this->assertFalse($result);
+    }
+
+    public function test_board_get() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+
+        $result = board::board_get($board->id);
+        $this->assertEquals($result[$column->id]->name, 'New Heading');
+    }
+
+    public function test_board_history() {
+        global $DB;
+
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $record = array(
+            'id' => 1,
+            'boardid' => $board->id,
+            'groupid' => 1,
+            'action' => 'action',
+            'userid' => 1,
+            'content' => 'content',
+            'timecreated' => 101010101010
+        );
+
+        $DB->insert_record('board_history', $record);
+        $record = $DB->get_record('board_history', array('action' => 'action'));
+
+        $result = board::board_history($board->id, 1);
+        $this->assertEquals($result[$record->id]->boardid, $board->id);
+    }
+
+    public function test_board_add_column() {
+        global $DB;
+
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $result = board::board_add_column($board->id, 'Test column');
+
+        $column = $DB->get_record('board_columns', array('name' => 'Test column'));
+
+        $this->assertIsArray($result);
+        $this->assertEquals($result['id'], $column->id);
+    }
+
+    public function test_board_update_column() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $result = board::board_update_column($column->id, 'Test column');
+
+        $this->assertIsArray($result);
+        $this->assertTrue($result['status']);
+    }
+
+    public function test_board_delete_column() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $result = board::board_delete_column($column->id);
+
+        $this->assertIsArray($result);
+        $this->assertTrue($result['status']);
+    }
+
+     // Undefined index: media -  media never gets set anywhere
+    public function test_get_note_file() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+
+        $result = board::get_note_file($note->id);
+        $this->assertNull($result);
+
+        $note = self::add_note_file($column->id, 'www.google.com');
+        $result = board::get_note_file($note->id);
+        $this->assertFalse($result);
+
+        $attachment = array(
+            'type' => 2,
+            'info' => '',
+            'url' => '',
+            'filename' => 'testimage.png',
+            'filecontents' => 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAABpAAAAQaCAIAhEUgAABpAAAAQaCAIAAADL9awBAAAACXBIWXMAA'
+        );
+
+        $note = board::board_add_note($column->id, 'heading', 'content', $attachment);
+        $result = board::get_note_file($note['note']->id);
+        $this->assertFalse($result);
+    }
+
+    public function test_store_note_file() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+
+        $attachment = array(
+            'type' => 2,
+            'info' => '',
+            'url' => '',
+            'filename' => 'testimage.png',
+            'filecontents' => '�PNG'
+        );
+
+        $result = board::store_note_file($note->id, $attachment);
+        $this->assertContains('mod_board', $result);
+        $this->assertContains('testimage.png', $result);
+    }
+
+    public function test_valid_for_upload() {
+        $attachment = array(
+            'type' => 2,
+            'info' => '',
+            'url' => '',
+            'filename' => 'testimage.png',
+            'filecontents' => '�PNGdata:image/png;base64,iVBORw0KGgoAAAANSUhEUgAABpAAAAQaCAIAhEUgAABpAAAAQaCAIAAADL9awBAAAACXBIWXMA'
+        );
+        $result = board::valid_for_upload($attachment);
+        $this->assertTrue($result);
+
+        $attachment = array(
+            'type' => 2,
+            'info' => '',
+            'url' => '',
+            'filename' => 'testimage.hhhh',
+            'filecontents' => '�PNGdata:image/png;base64,iVBORw0KGgoAAAANSUhEUgAABpAAAAQaCAIAhEUgAABpAAAAQaCAIAAADL9awBAAAACXBIWXMA'
+        );
+        $result = board::valid_for_upload($attachment);
+        $this->assertFalse($result);
+
+        $attachment = array(
+            'type' => 2,
+            'info' => '',
+            'url' => '',
+            'filename' => 'testimage.png',
+            'filecontents' => '�PNGdata:image/png;base64,AQAADL9awBAAAACXBIWXMA'
+        );
+        $result = board::valid_for_upload($attachment);
+        $this->assertFalse($result);
+    }
+
+    public function test_board_note_update_attachment() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+
+        $attachment = array(
+            'type' => 2,
+            'info' => 'test info',
+            'url' => 'test url',
+            'filename' => 'testimage.png',
+            'filecontents' => 'data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAABpAAAAQaCAIAhEUgAABpAAAAQaCAIAAADL9awBAAAACXBIWXMAASAS'
+        );
+
+        $result = board::board_note_update_attachment($note->id, $attachment);
+        $this->assertEquals($result['info'], $attachment['info']);
+        $this->assertEquals($result['url'], $attachment['url']);
+    }
+
+     // Undefined index: media -  media never gets set anywhere
+    public function test_board_add_note() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $attachment =  array(
+            'type' => 0,
+            'info' => '',
+            'url' => '',
+        );
+        $result = board::board_add_note($column->id, 'Test heading', 'Test content', $attachment);
+
+        $this->assertIsArray($result);
+    }
+
+    // Undefined index: media -  media never gets set anywhere
+    public function test_board_update_note() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+        $attachment =  array(
+            'type' => 0,
+            'info' => '',
+            'url' => '',
+        );
+        $result = board::board_update_note($note->id, 'update heading', 'update content', $attachment);
+
+        $this->assertIsArray($result);
+    }
+
+    public function test_board_delete_note() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+        $result = board::board_delete_note($note->id);
+
+        $this->assertIsArray($result);
+        $this->assertTrue($result['status']);
+    }
+
+    public function test_board_move_note() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+        $column2 = self::add_column($board->id, 'New column');
+        $result = board::board_move_note($note->id, $column2->id);
+
+        $this->assertIsArray($result);
+        $this->assertTrue($result['status']);
+    }
+
+    public function test_board_can_rate_note() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id, 'addrating' => 3));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+        $result = board::board_can_rate_note($note->id);
+
+        $this->assertTrue($result);
+    }
+
+    public function test_board_rating_enabled() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id));
+
+        $result = board::board_rating_enabled($board->id);
+        $this->assertFalse($result);
+
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id, 'addrating' => 3));
+        $result = board::board_rating_enabled($board->id);
+        $this->assertTrue($result);
+    }
+
+     // Undefined variable 'rate', it never gets defined if a valid note is passed in.
+    public function test_board_rate_note() {
+        $this->resetAfterTest();
+        $this->setAdminUser();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id, 'addrating' => 3));
+        $column = self::add_column($board->id);
+        $note = self::add_note($column->id);
+        $result = board::board_rate_note($note->id);
+
+        $this->assertIsArray($result);
+        $this->assertTrue($result['status']);
+    }
+
+    public function test_board_is_editor() {
+        $this->resetAfterTest();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id, 'addrating' => 3));
+
+        $result = board::board_is_editor($board->id);
+        $this->assertFalse($result);
+
+        $this->setAdminUser();
+        $result = board::board_is_editor($board->id);
+        $this->assertTrue($result);
+    }
+
+    public function test_board_readonly() {
+        $this->resetAfterTest();
+        $course = $this->getDataGenerator()->create_course();
+        $board = $this->getDataGenerator()->create_module('board', array('course' => $course->id, 'addrating' => 3));
+
+        $result = board::board_readonly($board->id);
+        $this->assertFalse($result);
+
+        $user = $this->getDataGenerator()->create_user();
+        $this->setUser($user);
+        $group = $this->getDataGenerator()->create_group(array('courseid' => $course->id));
+        $this->getDataGenerator()->create_group_member(array('userid' => $user->id, 'groupid' => $group->id));
+        $result = board::board_readonly($board->id);
+        $this->assertFalse($result);
+    }
+
+    private static function add_board($courseid) {
+         global $DB;
+
+         $record = array(
+            'id' => 1,
+            'course' => $courseid,
+            'name' => 'test',
+            'timemodified' => 0,
+            'intro' => '',
+            'historyid' => 3,
+            'background_color' => null,
+            'addrating' => 3,
+            'hideheaders' => 0,
+            'sortby' => 2,
+            'postby' => 0
+        );
+
+        $DB->insert_record('board', $record);
+        return $DB->get_record('board', array('name' => 'test'));
+    }
+
+    private static function add_column($boardid, $name = 'New Heading') {
+        global $DB;
+
+        $record = array(
+            'id' => 1,
+            'boardid' => $boardid,
+            'name' => $name,
+        );
+
+        $DB->insert_record('board_columns', $record);
+        return $DB->get_record('board_columns', array('name' => $name));
+    }
+
+    private static function add_note($columnid) {
+        global $DB;
+
+        $record = array(
+            'id' => 1,
+            'columnid' => $columnid,
+            'userid' => '2',
+            'groupid' => null,
+            'content' => 'Test content',
+            'heading' => 'Test heading',
+        );
+
+        $DB->insert_record('board_notes', $record);
+        return $DB->get_record('board_notes', array('columnid' => $columnid));
+    }
+
+    private static function add_note_file($columnid, $url) {
+        global $DB;
+
+        $record = array(
+            'id' => 1,
+            'columnid' => $columnid,
+            'userid' => '2',
+            'groupid' => null,
+            'content' => 'Test content',
+            'heading' => 'Test heading',
+            'url' => $url,
+        );
+
+        $DB->insert_record('board_notes', $record);
+        return $DB->get_record('board_notes', array('url' => $url));
+    }
+
+    private static function add_note_rating($noteid, $userid) {
+        global $DB;
+
+        $record = array(
+            'id' => 1,
+            'noteid' => $noteid,
+            'userid' => $userid,
+        );
+
+        $DB->insert_record('board_note_ratings', $record);
+        return $DB->get_record('board_note_ratings', array('userid' => $userid));
+    }
+}
diff --git a/tests/generator/lib.php b/tests/generator/lib.php
new file mode 100644
index 0000000..ffdf158
--- /dev/null
+++ b/tests/generator/lib.php
@@ -0,0 +1,27 @@
+<?php
+// This file is part of Moodle - http://moodle.org/
+//
+// Moodle is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// Moodle is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with Moodle.  If not, see <http://www.gnu.org/licenses/>.
+
+/**
+ * Board test generator.
+ *
+ * @package    mod_board
+ * @copyright  2020 onward: Brickfield Education Labs, www.brickfield.ie
+ * @author     Jay Churchward (jay@brickfieldlabs.ie)
+ * @license    http://www.gnu.org/copyleft/gpl.html GNU GPL v3 or later
+ */
+class mod_board_generator extends testing_module_generator {
+
+}
\ No newline at end of file