1 files changed, 55 insertions, 24 deletions
diff --git a/app/client/system/system.component.js b/app/client/system/system.component.js
index 7428c22..00c5395 100644
--- a/app/client/system/system.component.js
+++ b/app/client/system/system.component.js
@@ -6,6 +6,19 @@ import Group from '../common/group.component'
 import Param from '../common/param.component'
 
 import * as systemActions from './system.actions'
+import * as taskActions from '../task/task.actions'
+
+const cpu_test_task = {
+  id: 1073,
+  activity: 'train',
+  library: 'test',
+  dataset: 'test',
+  epochs: 1,
+  opt: {}
+}
+const gpu_test_task = {
+
+}
 
 class System extends Component {
   constructor(props){
@@ -20,30 +33,45 @@ class System extends Component {
         </div>
 
         <div className='row params'>
-          <Group title="Status">
-            <Param title='Server'>{server.status}</Param>
-            {server.error &&
-              <Param title='Server error'>{server.error.message}</Param>
-            }
-            <Param title='Relay'>{relay.status}</Param>
-            <Param title='RPC'>{rpc.status}</Param>
-            <Param title='CPU'>{rpc.cpu_cmd}</Param>
-            <Param title='GPU'>{rpc.gpu_cmd}</Param>
-            <Param title='Current Task'>train samplernn</Param>
-          </Group>
-          <Group title="Diagnostics">
-            <Param title='Check GPU'>
-              <button onClick={() => actions.run('nvidia-smi')}>nvidia-smi</button>
-            </Param>
-            <Param title='List processes'>
-              <button onClick={() => actions.run('ps')}>ps</button>
-            </Param>
-            <Param title='List users'>
-              <button onClick={() => actions.run('w')}>w</button>
-            </Param>
-          </Group>
+          <div className='column'>
+            <Group title="Status">
+              <Param title='Server'>{server.status}</Param>
+              {server.error &&
+                <Param title='Server error'>{server.error.message}</Param>
+              }
+              <Param title='Relay'>{relay.status}</Param>
+              <Param title='RPC'>{rpc.status}</Param>
+              <Param title='CPU'>{rpc.cpu_cmd}</Param>
+              <Param title='GPU'>{rpc.gpu_cmd}</Param>
+              <Param title='Current Task'>train samplernn</Param>
+            </Group>
+            <Group title="Diagnostics">
+              <Param title='Check GPU'>
+                <button onClick={() => actions.system.run('nvidia-smi')}>nvidia-smi</button>
+              </Param>
+              <Param title='List processes'>
+                <button onClick={() => actions.system.run('ps')}>ps</button>
+              </Param>
+              <Param title='List users'>
+                <button onClick={() => actions.system.run('w')}>w</button>
+              </Param>
+              <Param title='Disk free space'>
+                <button onClick={() => actions.system.run('df')}>df</button>
+              </Param>
+            </Group>
+            <Group title="Test">
+              <Param title='CPU Test Task'>
+                <button onClick={() => actions.task.start_task(cpu_test_task)}>Start</button>
+                <button onClick={() => actions.task.stop_task(cpu_test_task)}>Stop</button>
+              </Param>
+              <Param title='GPU Test Task'>
+                <button onClick={() => actions.task.start_task(gpu_test_task)}>Start</button>
+                <button onClick={() => actions.task.stop_task(gpu_test_task)}>Stop</button>
+              </Param>
+            </Group>
+          </div>
+          {this.renderCommandOutput()}
         </div>
-        {this.renderCommandOutput()}
       </div>
     )
   }
@@ -76,7 +104,10 @@ const mapStateToProps = state => ({
 })
 
 const mapDispatchToProps = (dispatch, ownProps) => ({
-  actions: bindActionCreators(systemActions, dispatch)
+  actions: {
+    system: bindActionCreators(systemActions, dispatch),
+    task: bindActionCreators(taskActions, dispatch),
+  },
 })
 
 export default connect(mapStateToProps, mapDispatchToProps)(System)