Call withStyles callback during tests

scommons · Sep 22, 2021 · 85308bb · 85308bb
1 parent 2737d7c
commit 85308bb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/src/main/resources/sc-material-ui-styles-mocks.js b/test/src/main/resources/sc-material-ui-styles-mocks.js
@@ -8,8 +8,11 @@ module.exports = {
   MuiThemeProvider: 'MuiThemeProvider',
   ThemeProvider: 'ThemeProvider',
 
-  withStyles: function () {
-    styles.withStyles(...arguments)
+  withStyles: function (stylesArg) {
+    if (typeof stylesArg == 'function') {
+      const theme = styles.createTheme({})
+      stylesArg(theme)
+    }
 
     return comp => {
       return "WithStylesMock-" + comp + "-" + (withStylesCompCounter++);