def test_get_all_keys(self): client = SysAdminClient("127.0.0.1", 4000) client.set("network.dhcp.startip", "192.168.1.1") client.set("network.dhcp.endip", "192.168.1.100") client.set("network.dhcp.interfaces", ["eth0", "eth1"]) client.set("my.service.restarter", True) client.set("network.breakfast", "waffles") client.set("network.lease", 1440) client.commit() resp = set(map(lambda x: x.key, client.get("network.dhcp.*").get.kvs)) self.assertEqual( set([ "network.dhcp.startip", "network.dhcp.endip", "network.dhcp.interfaces" ]), resp) resp = map(lambda x: UnpackFromProto(x.value), client.get("network.dhcp.*").get.kvs) self.assertEqual(3, len(resp)) self.assertTrue("192.168.1.1" in resp) self.assertTrue("192.168.1.100" in resp) self.assertTrue(["eth0", "eth1"] in resp) resp = set(map(lambda x: x.key, client.get("network.*").get.kvs)) self.assertEqual( set([ "network.dhcp.startip", "network.dhcp.endip", "network.dhcp.interfaces", "network.breakfast", "network.lease" ]), resp) resp = map(lambda x: UnpackFromProto(x.value), client.get("network.*").get.kvs) self.assertEqual(5, len(resp)) self.assertTrue("192.168.1.1" in resp) self.assertTrue("192.168.1.100" in resp) self.assertTrue(["eth0", "eth1"] in resp) self.assertTrue("waffles" in resp) self.assertTrue(1440 in resp)
def test_new_key_migration(self): client = SysAdminClient("127.0.0.1", 4000) client.set("network.dhcp.startip", "192.168.1.1") client.set("network.dhcp.endip", "192.168.1.100") client.set("network.dhcp.interfaces", ["eth0", "eth1"]) client.set("my.service.restarter", True) client.set("network.breakfast", "waffles") client.set("network.lease", 1440) client.commit() m = SysAdminMigrator(client) m.new_keys({"new.key": "value"}) client.commit() self.assertEqual( "value", UnpackFromProto(client.get("new.key").get.kvs[0].value)) m.change_values({"new.key": "4"}) client.commit() self.assertEqual( "4", UnpackFromProto(client.get("new.key").get.kvs[0].value)) m.change_types({"new.key": "int32"}) client.commit() self.assertEqual( 4, UnpackFromProto(client.get("new.key").get.kvs[0].value)) m.rename_keys([{"new.key": "renamed.key"}]) client.commit() self.assertEqual( 4, UnpackFromProto(client.get("renamed.key").get.kvs[0].value)) m.remove_keys(["new.key"]) client.commit() self.assertEqual( 4, UnpackFromProto(client.get("renamed.key").get.kvs[0].value))
def test_basic_drop(self): client = SysAdminClient("127.0.0.1", 4000) client.set("wireless.wl0.psk", "password") client.set("wireless.wl0.channel", 40) client.drop() self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, client.get("wireless.wl0.psk").status) self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, client.get("wireless.wl0.channel").status)
def test_new_key_rollback(self): #make a commit with a new key and value, ensure that a rollback can remove the key client = SysAdminClient("127.0.0.1", 4000) client.set("cheese", 42) self.assertEqual(1, client.commit().commit.commit_id) self.assertEqual( 42, UnpackFromProto(client.get("cheese").get.kvs[0].value)) self.assertEqual(sysadminctl_pb2.SUCCESS, client.rollback(1).status) self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, client.get("cheese").status)
def test_change_types(self): client = SysAdminClient("127.0.0.1", 4000) m = SysAdminMigrator(client) m.new_keys({"gonna.be.list": "value"}) client.commit() m.change_types({"gonna.be.list": "stringlist"}) client.commit() self.assertEqual(["value"], UnpackFromProto( client.get("gonna.be.list").get.kvs[0].value)) m.change_types({"gonna.be.list": "str"}) client.commit() self.assertEqual( "value", UnpackFromProto(client.get("gonna.be.list").get.kvs[0].value)) m.new_keys({"gonna.be.intlist": "4"}) client.commit() m.change_types({"gonna.be.intlist": "int32list"}) client.commit() self.assertEqual([4], UnpackFromProto( client.get("gonna.be.intlist").get.kvs[0].value)) m.change_types({"gonna.be.intlist": "str"}) client.commit() self.assertEqual( "4", UnpackFromProto(client.get("gonna.be.intlist").get.kvs[0].value)) m.new_keys({"gonna.be.boollist": "true"}) client.commit() m.change_types({"gonna.be.boollist": "boollist"}) client.commit() self.assertEqual([True], UnpackFromProto( client.get("gonna.be.boollist").get.kvs[0].value)) m.change_types({"gonna.be.boollist": "str"}) client.commit() self.assertEqual( "True", UnpackFromProto(client.get("gonna.be.boollist").get.kvs[0].value)) m.new_keys({"long.list": ["stringlist", "4", "5", "6"]}) client.commit() m.change_types({"long.list": "int32list"}) client.commit() self.assertEqual([4, 5, 6], UnpackFromProto( client.get("long.list").get.kvs[0].value)) m.change_types({"long.list": "int32list"}) client.commit() self.assertEqual([4, 5, 6], UnpackFromProto( client.get("long.list").get.kvs[0].value)) m.change_types({"long.list": "stringlist"}) client.commit() self.assertEqual(["4", "5", "6"], UnpackFromProto( client.get("long.list").get.kvs[0].value))
def test_new_key_migration(self): client = SysAdminClient("127.0.0.1", 4000) client.set("network.dhcp.startip", "192.168.1.1") client.set("network.dhcp.endip", "192.168.1.100") client.set("network.dhcp.interfaces", ["eth0", "eth1"]) client.set("my.service.restarter", True) client.set("network.breakfast", "eggs") client.set("network.lease", 1440) client.set("default.network.breakfast", "waffles") client.set("default.network.lease", 1440) client.commit() client.reset() resp = client.get("network.dhcp.startip") self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, resp.status) self.assertEqual( "waffles", UnpackFromProto(client.get("network.breakfast").get.kvs[0].value))
def test_erase(self): client = SysAdminClient("127.0.0.1", 4000) client.set("wireless.wl0.psk", "password") client.set("wireless.wl0.channel", 40) client.set("wireless.wl0.enabled", True) client.set("wireless.wl0.frequencies", [1, 2, 3]) client.commit() self.assertEqual( "password", UnpackFromProto(client.get("wireless.wl0.psk").get.kvs[0].value)) client.erase("wireless.wl0.psk") self.assertEqual( "password", UnpackFromProto(client.get("wireless.wl0.psk").get.kvs[0].value)) client.erase("not.a.key") client.commit() self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, client.get("wireless.wl0.psk").status)
def FetchAllValues(keys): client = SysAdminClient("127.0.0.1", 4000) output = {} for arg in keys: resp = client.get(arg) if resp.status == sysadminctl_pb2.SUCCESS: for value in resp.get.kvs: output[value.key] = UnpackFromProto(value.value) elif "last." not in arg: raise Exception("Could not find required key. Key %s, Error: %s" % (arg, resp.status)) return condense(output)
def test_overridden_xid(self): # This client gets the regular sysadmin assigned id client = SysAdminClient("127.0.0.1", 4000) client.set("network.dhcp.startip", "192.168.1.1") client.set("network.dhcp.endip", "192.168.1.100") # This client has a overridden id 0 otherclient = SysAdminClient("127.0.0.1", 4000, 0) # This client can't commit the original clients changes otherclient.commit() self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, client.get("network.dhcp.startip").status) self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, client.get("network.dhcp.endip").status) client.commit() self.assertEqual( "192.168.1.1", UnpackFromProto( client.get("network.dhcp.startip").get.kvs[0].value)) self.assertEqual( "192.168.1.100", UnpackFromProto(client.get("network.dhcp.endip").get.kvs[0].value))
def test_type_checking(self): client = SysAdminClient("127.0.0.1", 4000) client.set("wireless.wl0.psk", "password") resp = client.set("wireless.wl0.psk", 5) self.assertEqual(sysadminctl_pb2.SUCCESS, resp.status) client.commit() resp = client.set("wireless.wl0.psk", "password") self.assertEqual(sysadminctl_pb2.TYPE_MISMATCH, resp.status) client.commit() self.assertEqual( 5, UnpackFromProto(client.get("wireless.wl0.psk").get.kvs[0].value))
def test_basic_sets(self): # one in-flight client = SysAdminClient("127.0.0.1", 4000) resp = client.set("wireless.wl0.ssid", "cheese") self.assertEqual("wireless.wl0.ssid", resp.get.kvs[0].key) self.assertEqual("cheese", UnpackFromProto(resp.get.kvs[0].value)) resp = client.get("wireless.wl0.ssid") self.assertEqual(sysadminctl_pb2.KEY_NOT_FOUND, resp.status) client.commit() resp = client.get("wireless.wl0.ssid") self.assertEqual("cheese", UnpackFromProto(resp.get.kvs[0].value)) # multiple in-flight, one erasing the other client.set("wireless.wl0.psk", "password") client.set("wireless.wl0.channel", 40) client.set("wireless.wl0.channel", 60) client.set("wireless.wl0.enabled", True) client.set("wireless.wl0.frequencies", [1, 2, 3]) client.commit() self.assertEqual( 60, UnpackFromProto( client.get("wireless.wl0.channel").get.kvs[0].value)) self.assertEqual( "password", UnpackFromProto(client.get("wireless.wl0.psk").get.kvs[0].value)) self.assertEqual( True, UnpackFromProto( client.get("wireless.wl0.enabled").get.kvs[0].value)) self.assertEqual( [1, 2, 3], UnpackFromProto( client.get("wireless.wl0.frequencies").get.kvs[0].value))
def test_basic_rollback(self): # This client gets the regular sysadmin assigned id client = SysAdminClient("127.0.0.1", 4000) client.set("network.dhcp.startip", "192.168.1.1") client.set("network.dhcp.endip", "192.168.1.100") self.assertEqual(1, client.commit().commit.commit_id) client.set("network.dhcp.startip", "10.0.0.1") client.set("network.dhcp.endip", "10.0.0.100") self.assertEqual(2, client.commit().commit.commit_id) # Can't rollback the first commit self.assertEqual(sysadminctl_pb2.FAILED_ROLLBACK, client.rollback(1).status) self.assertEqual(sysadminctl_pb2.SUCCESS, client.rollback(2).status) self.assertEqual( "192.168.1.1", UnpackFromProto( client.get("network.dhcp.startip").get.kvs[0].value)) self.assertEqual(sysadminctl_pb2.FAILED_ROLLBACK, client.rollback(100).status)