def test_get_os(self):
self.fail_if_not_testing_env()
self.clean_monkey_db()
linux_monkey = Monkey(
guid=str(uuid.uuid4()),
description=
"Linux shay-Virtual-Machine 4.15.0-50-generic #54-Ubuntu SMP Mon May 6 18:46:08 UTC 2019 x86_64 x86_64"
)
windows_monkey = Monkey(guid=str(uuid.uuid4()),
description="Windows bla bla bla")
unknown_monkey = Monkey(guid=str(uuid.uuid4()),
description="bla bla bla")
linux_monkey.save()
windows_monkey.save()
unknown_monkey.save()
self.assertEquals(
1, len(filter(lambda m: m.get_os() == "windows",
Monkey.objects())))
self.assertEquals(
1, len(filter(lambda m: m.get_os() == "linux", Monkey.objects())))
self.assertEquals(
1, len(filter(lambda m: m.get_os() == "unknown",
Monkey.objects())))
