mock methods from class

by María Fernández

JavaScript

class blabla {
	print(num){
    console.log('original class', num);
  }
}
const blabla1 = new blabla();
blabla1.print(1);

// Save method and mock it
const originalMethod = blabla.prototype['print'];
blabla.prototype['print'] = () => {
	console.log('boom!')
};

const blabla2 = new blabla();
blabla2.print(2); // boom!

// Restore method
blabla.prototype['print'] = originalMethod;
const blabla3 = new blabla();
blabla3.print(3);