def test_cluster_state(self): password_box = PasswordBox(gen_encryption_key()) expected = MySQLCluster( 'cluster1', 'cluster_user', password_box.encrypt('cluster_password'), 3, DEFAULT_TASK_CPUS, DEFAULT_TASK_MEM, DEFAULT_TASK_DISK) expected.tasks['task1'] = MySQLTask( 'cluster1', 'task1', 'slave1', 'host1', 10000) self._state_provider.dump_cluster_state(expected) actual = self._state_provider.load_cluster_state('cluster1') assert expected.user == actual.user assert isinstance(actual.num_nodes, int) assert expected.num_nodes == actual.num_nodes assert len(expected.tasks) == len(actual.tasks) assert expected.tasks['task1'].port == actual.tasks['task1'].port assert expected.encrypted_password == actual.encrypted_password assert password_box.match('cluster_password', actual.encrypted_password)
def test_scheduler_recovery_failure_before_launch(self): scheduler_key = gen_encryption_key() scheduler1 = MysosScheduler( self._state, self._state_provider, self._framework_user, "./executor.pex", "cmd.sh", self._zk_client, self._zk_url, Amount(5, Time.SECONDS), "/etc/mysos/admin_keyfile.yml", scheduler_key) scheduler1.registered(self._driver, self._framework_id, object()) _, password = scheduler1.create_cluster("cluster1", "mysql_user", 3) # Simulate restart before the task is successfully launched. scheduler2 = MysosScheduler( self._state, self._state_provider, self._framework_user, "./executor.pex", "cmd.sh", self._zk_client, self._zk_url, Amount(5, Time.SECONDS), "/etc/mysos/admin_keyfile.yml", scheduler_key) assert len(scheduler2._launchers) == 0 # No launchers are recovered. # Scheduler always receives registered() with the same FrameworkID after failover. scheduler2.registered(self._driver, self._framework_id, object()) assert len(scheduler2._launchers) == 1 assert scheduler2._launchers["cluster1"].cluster_name == "cluster1" password_box = PasswordBox(scheduler_key) assert password_box.match( password, scheduler2._launchers["cluster1"]._cluster.encrypted_password) # Now offer the resources for this task. scheduler2.resourceOffers(self._driver, [self._offer]) # One task is launched for the offer. assert len(scheduler2._launchers["cluster1"]._cluster.active_tasks) == 1 # Scheduler has recovered the cluster so it doesn't accept another of the same name. with pytest.raises(MysosScheduler.ClusterExists): scheduler2.create_cluster("cluster1", "mysql_user", 3)
def test_cluster_state(self): password_box = PasswordBox(gen_encryption_key()) expected = MySQLCluster('cluster1', 'cluster_user', password_box.encrypt('cluster_password'), 3, DEFAULT_TASK_CPUS, DEFAULT_TASK_MEM, DEFAULT_TASK_DISK) expected.tasks['task1'] = MySQLTask('cluster1', 'task1', 'slave1', 'host1', 10000) self._state_provider.dump_cluster_state(expected) actual = self._state_provider.load_cluster_state('cluster1') assert expected.user == actual.user assert isinstance(actual.num_nodes, int) assert expected.num_nodes == actual.num_nodes assert len(expected.tasks) == len(actual.tasks) assert expected.tasks['task1'].port == actual.tasks['task1'].port assert expected.encrypted_password == actual.encrypted_password assert password_box.match('cluster_password', actual.encrypted_password)