def test_set_sets_can_associate_data_to_UNDEFINED(self): environment = Environment(environment={}) data = DataSet({'value'}, 6) environment.set(UNDEFINED, data) # Testing internal attribute which is not ideal, but I am trying to not rely on any other behavior # (e.g. without using `get`). self.assertDictEqual(environment._environment, {UNDEFINED: data})
def test_get_returns_a_dataset_with_UNDEFINED_when_the_environment_variable_is_not_set( self): environment = Environment(environment={}) name = 'variable_name' variable_names = {name} result_values, _ = environment.get(variable_names) self.assertEqual(result_values, {UNDEFINED})
def test_get_returns_a_dataset_with_UNDEFINED_when_the_environment_variable_is_not_set( self): environment = Environment(environment={}) name = 'variable_name' variable_names = DataSet({name}, len(name)) result_values, _ = environment.get(variable_names) self.assertEqual(result_values, DataSet({UNDEFINED}, UNKNOWN_SIZE))
def test_set_sets_associates_data_to_a_given_variable_name(self): environment = Environment(environment={}) data = DataSet({'value'}, 6) environment.set('variable_name', data) # Testing internal attribute which is not ideal, but I am trying to not rely on any other behavior # (e.g. without using `get`). self.assertDictEqual(environment._environment, {'variable_name': data})
def test_get_returns_the_pointer_associated_with_an_existing_environment_variable( self): name = 'variable_name' data = DataSet({self.pointer}, self.pointer_length) environment = Environment(environment={name: data}) variable_names = DataSet({name}, len(name)) result_values, _ = environment.get(variable_names) self.assertEqual(result_values, data)
def test_get_returns_the_pointer_associated_with_an_existing_environment_variable( self): name = 'variable_name' data = {claripy.BVV(self.pointer, self.pointer_length)} environment = Environment(environment={name: data}) variable_names = {name} result_values, _ = environment.get(variable_names) self.assertEqual(result_values, data)
def test_get_result_tells_if_environment_does_not_know_about_one_of_the_variable( self): name = 'variable_name' data = {claripy.BVV(b'value\x00', 6 * 8)} environment = Environment(environment={name: data}) name = 'variable_name' variable_names = {name, 'unknown_name'} _, has_unknown = environment.get(variable_names) self.assertTrue(has_unknown)
def test_get_result_tells_if_environment_knows_about_all_the_variables( self): name = 'variable_name' data = {claripy.BVV(self.pointer, self.pointer_length)} environment = Environment(environment={name: data}) name = 'variable_name' variable_names = {name} _, has_unknown = environment.get(variable_names) self.assertFalse(has_unknown)
def test_get_result_tells_if_environment_knows_about_all_the_variables( self): name = 'variable_name' data = DataSet({self.pointer}, self.pointer_length) environment = Environment(environment={name: data}) name = 'variable_name' variable_names = DataSet({name}, UNKNOWN_SIZE) _, has_unknown = environment.get(variable_names) self.assertFalse(has_unknown)
def test_get_result_tells_if_environment_does_not_know_about_one_of_the_variable( self): name = 'variable_name' data = DataSet({'value'}, 6) environment = Environment(environment={name: data}) name = 'variable_name' variable_names = DataSet({name, 'unknown_name'}, UNKNOWN_SIZE) _, has_unknown = environment.get(variable_names) self.assertTrue(has_unknown)
def test_get_returns_values_associated_with_all_possible_given_names(self): environment_data = { 'variable_name': DataSet({self.pointer}, self.pointer_length), UNDEFINED: DataSet({UNDEFINED}, UNKNOWN_SIZE) } environment = Environment(environment=environment_data) variable_names = DataSet({'variable_name', UNDEFINED}, UNKNOWN_SIZE) expected_result = DataSet({self.pointer, UNDEFINED}, UNKNOWN_SIZE) result_values, _ = environment.get(variable_names) self.assertEqual(result_values, expected_result)
def test_get_returns_values_associated_with_all_possible_given_names(self): environment_data = { 'variable_name': {claripy.BVV(self.pointer, self.pointer_length)}, UNDEFINED: {UNDEFINED}, } environment = Environment(environment=environment_data) variable_names = {'variable_name', UNDEFINED} expected_result = { claripy.BVV(self.pointer, self.pointer_length), UNDEFINED } result_values, _ = environment.get(variable_names) self.assertEqual(result_values, expected_result)
def test_merge_two_environments_merge_data_associated_with_each_variable( self): first = 'variable_name' second = 'other_variable_name' environment = Environment(environment={first: DataSet({'value1'}, 7)}) other_environment = Environment( environment={ first: DataSet({'value2'}, 7), second: DataSet({UNDEFINED}, UNKNOWN_SIZE) }) expected_environment = Environment( environment={ first: DataSet({'value1', 'value2'}, 7), second: DataSet({UNDEFINED}, UNKNOWN_SIZE) }) self.assertEqual(environment.merge(other_environment), expected_environment)
def test_merge_two_environments_merge_data_associated_with_each_variable( self): first = 'variable_name' second = 'other_variable_name' environment = Environment( environment={first: {claripy.BVV(b'value1\x00', 7 * 8)}}) other_environment = Environment(environment={ first: {claripy.BVV(b'value2\x00', 7 * 8)}, second: {UNDEFINED} }) expected_environment = Environment( environment={ first: { claripy.BVV(b'value1\x00', 7 * 8), claripy.BVV(b'value2\x00', 7 * 8) }, second: {UNDEFINED} }) self.assertEqual(environment.merge(other_environment), expected_environment)
def test_merge_fails_when_merging_with_a_non_Environment_instance(self): environment = Environment( environment={'variable_name': DataSet({'value1'}, 7)}) other_environment = 0x42 self.assertRaises(TypeError, environment.merge, other_environment)
def test_get_fails_when_called_with_wrong_type_of_name(self): environment = Environment(environment={}) variable_names = DataSet({0x42}, 8) self.assertRaises(TypeError, environment.get, variable_names)
def test_get_fails_when_called_with_wrong_type_of_name(self): environment = Environment(environment={}) variable_names = {claripy.BVV(0x42, 8 * 8)} self.assertRaises(TypeError, environment.get, variable_names)
def test_merge_fails_when_merging_with_a_non_Environment_instance(self): environment = Environment( environment={'variable_name': {claripy.BVV(b'value1\x00', 7 * 8)}}) other_environment = 0x42 self.assertRaises(TypeError, environment.merge, other_environment)
def test_set_fails_when_wvariable_name_is_of_wrong_type(self): environment = Environment(environment={}) data = DataSet({'value'}, 6) self.assertRaises(TypeError, environment.set, 0x42, data)
def test_set_fails_when_wvariable_name_is_of_wrong_type(self): environment = Environment(environment={}) data = {claripy.BVV(b'value\x00', 6 * 8)} self.assertRaises(TypeError, environment.set, 0x42, data)