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
