def test_load_by_dir(self): """Test DeviceModel.load_by_dir""" device = DeviceModel().load_by_dir('marklar') self.assertIsNone(device.id()) device = DeviceModel().load_by_dir('fixture') self.assertIsNone(device.id()) device = DeviceModel().load_by_dir('fixture/device') self.assertEqual(device.id(), 1)
def index_all_devices(self): """Index files for every folder that qualifies as a device within the current directory """ os.chdir(self._basedir) for node in os.listdir(self._basedir): device = DeviceModel().load_by_dir(node) if device.id() is not None: self.index_device(device)
def test_get_transfer_dirs(self): """Test DeviceModel.get_transfer_dirs""" device = DeviceModel() device.directory('marklar') self.assertEqual(len(device.get_transfer_dirs()), 0) device.directory('fixture/device') dirs = device.get_transfer_dirs() self.assertTrue(len(device.get_transfer_dirs()) > 0) for d in dirs: self.assertTrue(os.path.isdir(d))
def post(self): """ Uses the houseguard db to make a device :return: """ parser.add_argument('allowed', type=int, required=True) parser.add_argument('blocked', type=int, required=True) parser.add_argument('unknown', type=int, required=True) args = parser.parse_args() # Adding device to db model = DeviceModel() model.create_device(args['allowed'], args['blocked'], args['unknown']) return 'Complete', 200
def get(self): model = DeviceModel() device = model.get_device() return device, 200